From ee7d5e7206bf3d87035822b062ad52954401370a Mon Sep 17 00:00:00 2001
From: sino <sino2322@gmail.com>
Date: Tue, 27 Aug 2024 14:43:37 +0800
Subject: [PATCH] feat: support Moonshot and GLM models tool call for volc ark
 provider (#7666)

---
 .../model_providers/volcengine_maas/llm/models.py    | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/api/core/model_runtime/model_providers/volcengine_maas/llm/models.py b/api/core/model_runtime/model_providers/volcengine_maas/llm/models.py
index 4e2c66a066..a882f68a36 100644
--- a/api/core/model_runtime/model_providers/volcengine_maas/llm/models.py
+++ b/api/core/model_runtime/model_providers/volcengine_maas/llm/models.py
@@ -38,7 +38,7 @@ configs: dict[str, ModelConfig] = {
     ),
     'Doubao-lite-128k': ModelConfig(
         properties=ModelProperties(context_size=131072, max_tokens=4096, mode=LLMMode.CHAT),
-        features=[ModelFeature.TOOL_CALL]
+        features=[]
     ),
     'Skylark2-pro-4k': ModelConfig(
         properties=ModelProperties(context_size=4096, max_tokens=4096, mode=LLMMode.CHAT),
@@ -54,23 +54,23 @@ configs: dict[str, ModelConfig] = {
     ),
     'Moonshot-v1-8k': ModelConfig(
         properties=ModelProperties(context_size=8192, max_tokens=4096, mode=LLMMode.CHAT),
-        features=[]
+        features=[ModelFeature.TOOL_CALL]
     ),
     'Moonshot-v1-32k': ModelConfig(
         properties=ModelProperties(context_size=32768, max_tokens=16384, mode=LLMMode.CHAT),
-        features=[]
+        features=[ModelFeature.TOOL_CALL]
     ),
     'Moonshot-v1-128k': ModelConfig(
         properties=ModelProperties(context_size=131072, max_tokens=65536, mode=LLMMode.CHAT),
-        features=[]
+        features=[ModelFeature.TOOL_CALL]
     ),
     'GLM3-130B': ModelConfig(
         properties=ModelProperties(context_size=8192, max_tokens=4096, mode=LLMMode.CHAT),
-        features=[]
+        features=[ModelFeature.TOOL_CALL]
     ),
     'GLM3-130B-Fin': ModelConfig(
         properties=ModelProperties(context_size=8192, max_tokens=4096, mode=LLMMode.CHAT),
-        features=[]
+        features=[ModelFeature.TOOL_CALL]
     ),
     'Mistral-7B': ModelConfig(
         properties=ModelProperties(context_size=8192, max_tokens=2048, mode=LLMMode.CHAT),