hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-11 20:06:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support Moonshot and GLM models tool call for volc ark provider (#7666)

Browse Source
This commit is contained in:
sino
2024-08-27 14:43:37 +08:00
committed by GitHub
parent 2726fb3d5d
commit ee7d5e7206
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/core/model_runtime/model_providers/volcengine_maas/llm/models.py
View File

@@ -38,7 +38,7 @@ configs: dict[str, ModelConfig] = {
), ),
'Doubao-lite-128k': ModelConfig( 'Doubao-lite-128k': ModelConfig(
properties=ModelProperties(context_size=131072, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=131072, max_tokens=4096, mode=LLMMode.CHAT),
features=[ModelFeature.TOOL_CALL] features=[]
), ),
'Skylark2-pro-4k': ModelConfig( 'Skylark2-pro-4k': ModelConfig(
properties=ModelProperties(context_size=4096, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=4096, max_tokens=4096, mode=LLMMode.CHAT),
@@ -54,23 +54,23 @@ configs: dict[str, ModelConfig] = {
), ),
'Moonshot-v1-8k': ModelConfig( 'Moonshot-v1-8k': ModelConfig(
properties=ModelProperties(context_size=8192, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=8192, max_tokens=4096, mode=LLMMode.CHAT),
features=[] features=[ModelFeature.TOOL_CALL]
), ),
'Moonshot-v1-32k': ModelConfig( 'Moonshot-v1-32k': ModelConfig(
properties=ModelProperties(context_size=32768, max_tokens=16384, mode=LLMMode.CHAT), properties=ModelProperties(context_size=32768, max_tokens=16384, mode=LLMMode.CHAT),
features=[] features=[ModelFeature.TOOL_CALL]
), ),
'Moonshot-v1-128k': ModelConfig( 'Moonshot-v1-128k': ModelConfig(
properties=ModelProperties(context_size=131072, max_tokens=65536, mode=LLMMode.CHAT), properties=ModelProperties(context_size=131072, max_tokens=65536, mode=LLMMode.CHAT),
features=[] features=[ModelFeature.TOOL_CALL]
), ),
'GLM3-130B': ModelConfig( 'GLM3-130B': ModelConfig(
properties=ModelProperties(context_size=8192, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=8192, max_tokens=4096, mode=LLMMode.CHAT),
features=[] features=[ModelFeature.TOOL_CALL]
), ),
'GLM3-130B-Fin': ModelConfig( 'GLM3-130B-Fin': ModelConfig(
properties=ModelProperties(context_size=8192, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=8192, max_tokens=4096, mode=LLMMode.CHAT),
features=[] features=[ModelFeature.TOOL_CALL]
), ),
'Mistral-7B': ModelConfig( 'Mistral-7B': ModelConfig(
properties=ModelProperties(context_size=8192, max_tokens=2048, mode=LLMMode.CHAT), properties=ModelProperties(context_size=8192, max_tokens=2048, mode=LLMMode.CHAT),