Skip to content

Commit b8071a8

Browse files
committed
fix: use the default model provider when listing models
Signed-off-by: Donnie Adams <[email protected]>
1 parent 2bbe7bc commit b8071a8

File tree

3 files changed

+9
-1
lines changed

3 files changed

+9
-1
lines changed

pkg/cli/gptscript.go

+3
Original file line numberDiff line numberDiff line change
@@ -406,6 +406,9 @@ func (r *GPTScript) Run(cmd *cobra.Command, args []string) (retErr error) {
406406
defer gptScript.Close(true)
407407

408408
if r.ListModels {
409+
if r.DefaultModelProvider != "" {
410+
args = append(args, r.DefaultModelProvider)
411+
}
409412
return r.listModels(ctx, gptScript, args)
410413
}
411414

pkg/sdkserver/routes.go

+4
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,10 @@ func (s *server) listModels(w http.ResponseWriter, r *http.Request) {
127127
providers = reqObject.Providers
128128
}
129129

130+
if s.gptscriptOpts.DefaultModelProvider != "" {
131+
providers = append(providers, s.gptscriptOpts.DefaultModelProvider)
132+
}
133+
130134
out, err := s.client.ListModels(r.Context(), providers...)
131135
if err != nil {
132136
writeError(logger, w, http.StatusInternalServerError, fmt.Errorf("failed to list models: %w", err))

pkg/server/server.go

+2-1
Original file line numberDiff line numberDiff line change
@@ -24,5 +24,6 @@ func ContextWithNewRunID(ctx context.Context) context.Context {
2424
}
2525

2626
func RunIDFromContext(ctx context.Context) string {
27-
return ctx.Value(execKey{}).(string)
27+
runID, _ := ctx.Value(execKey{}).(string)
28+
return runID
2829
}

0 commit comments

Comments
 (0)