Update client.go (#30)

sunshineplan · Mar 30, 2024 · 7e58417 · 7e58417
1 parent e1c7243
commit 7e58417
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/client/client.go b/client/client.go
@@ -16,6 +16,7 @@ func New(cfg ai.ClientConfig) (client ai.AI, err error) {
 		ai.WithAPIKey(cfg.APIKey),
 		ai.WithEndpoint(cfg.Endpoint),
 		ai.WithProxy(cfg.Proxy),
+		ai.WithModel(cfg.Model),
 		ai.WithModelConfig(cfg.ModelConfig),
 	}
 	if cfg.Limit != nil {