fix: yi

labring · Dec 30, 2024 · 53d59c6 · 53d59c6
1 parent f60395c
commit 53d59c6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/service/aiproxy/relay/adaptor/lingyiwanwu/constants.go b/service/aiproxy/relay/adaptor/lingyiwanwu/constants.go
@@ -17,6 +17,7 @@ var ModelList = []*model.ModelConfig{
 		RPM:         60,
 		Config: model.NewModelConfig(
 			model.WithModelConfigMaxContextTokens(16384),
+			model.WithModelConfigToolChoice(true),
 		),
 	},
 	{
@@ -28,6 +29,8 @@ var ModelList = []*model.ModelConfig{
 		RPM:         60,
 		Config: model.NewModelConfig(
 			model.WithModelConfigMaxContextTokens(16384),
+			model.WithModelConfigVision(true),
+			model.WithModelConfigToolChoice(true),
 		),
 	},
 }