Skip to content

Commit 258774b

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

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
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))

0 commit comments

Comments
 (0)