From 53d59c628ca3e22e681590a39bc738fb34054b1d Mon Sep 17 00:00:00 2001
From: zijiren233 <pyh1670605849@gmail.com>
Date: Mon, 30 Dec 2024 16:48:47 +0800
Subject: [PATCH] fix: yi

---
 service/aiproxy/relay/adaptor/lingyiwanwu/constants.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/service/aiproxy/relay/adaptor/lingyiwanwu/constants.go b/service/aiproxy/relay/adaptor/lingyiwanwu/constants.go
index 0d61e1d0c978..3b95f8412cb3 100644
--- 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),
 		),
 	},
 }