AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 06:25:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(ark): support doubao 1.5 series of models (#12935)

Browse Source
This commit is contained in:
sino 2025-01-22 15:25:57 +08:00 committed by GitHub
parent 71fa14f791
commit d167d5b1be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 63 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -18,72 +18,93 @@ class ModelConfig(BaseModel):
configs: dict[str, ModelConfig] = { configs: dict[str, ModelConfig] = {
"Doubao-1.5-vision-pro-32k": ModelConfig(
properties=ModelProperties(context_size=32768, max_tokens=12288, mode=LLMMode.CHAT),
features=[ModelFeature.AGENT_THOUGHT, ModelFeature.VISION],
),
"Doubao-1.5-pro-32k": ModelConfig(
properties=ModelProperties(context_size=32768, max_tokens=12288, mode=LLMMode.CHAT),
features=[ModelFeature.AGENT_THOUGHT],
),
"Doubao-1.5-lite-32k": ModelConfig(
properties=ModelProperties(context_size=32768, max_tokens=12288, mode=LLMMode.CHAT),
features=[ModelFeature.AGENT_THOUGHT],
),
"Doubao-1.5-pro-256k": ModelConfig(
properties=ModelProperties(context_size=262144, max_tokens=12288, mode=LLMMode.CHAT),
features=[ModelFeature.AGENT_THOUGHT],
),
"Doubao-vision-pro-32k": ModelConfig( "Doubao-vision-pro-32k": ModelConfig(
properties=ModelProperties(context_size=32768, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=32768, max_tokens=4096, mode=LLMMode.CHAT),
features=[ModelFeature.VISION], features=[ModelFeature.AGENT_THOUGHT, ModelFeature.VISION],
), ),
"Doubao-vision-lite-32k": ModelConfig( "Doubao-vision-lite-32k": ModelConfig(
properties=ModelProperties(context_size=32768, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=32768, max_tokens=4096, mode=LLMMode.CHAT),
features=[ModelFeature.VISION], features=[ModelFeature.AGENT_THOUGHT, ModelFeature.VISION],
), ),
"Doubao-pro-4k": ModelConfig( "Doubao-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),
features=[ModelFeature.TOOL_CALL], features=[ModelFeature.AGENT_THOUGHT, ModelFeature.TOOL_CALL],
), ),
"Doubao-lite-4k": ModelConfig( "Doubao-lite-4k": ModelConfig(
properties=ModelProperties(context_size=4096, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=4096, max_tokens=4096, mode=LLMMode.CHAT),
features=[ModelFeature.TOOL_CALL], features=[ModelFeature.AGENT_THOUGHT, ModelFeature.TOOL_CALL],
), ),
"Doubao-pro-32k": ModelConfig( "Doubao-pro-32k": ModelConfig(
properties=ModelProperties(context_size=32768, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=32768, max_tokens=4096, mode=LLMMode.CHAT),
features=[ModelFeature.TOOL_CALL], features=[ModelFeature.AGENT_THOUGHT, ModelFeature.TOOL_CALL],
), ),
"Doubao-lite-32k": ModelConfig( "Doubao-lite-32k": ModelConfig(
properties=ModelProperties(context_size=32768, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=32768, max_tokens=4096, mode=LLMMode.CHAT),
features=[ModelFeature.TOOL_CALL], features=[ModelFeature.AGENT_THOUGHT, ModelFeature.TOOL_CALL],
), ),
"Doubao-pro-256k": ModelConfig( "Doubao-pro-256k": ModelConfig(
properties=ModelProperties(context_size=262144, max_tokens=4096, mode=LLMMode.CHAT), properties=ModelProperties(context_size=262144, max_tokens=4096, mode=LLMMode.CHAT),
features=[], features=[ModelFeature.AGENT_THOUGHT],
), ),
"Doubao-pro-128k": ModelConfig( "Doubao-pro-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=[ModelFeature.AGENT_THOUGHT, ModelFeature.TOOL_CALL],
), ),
"Doubao-lite-128k": ModelConfig( "Doubao-lite-128k": ModelConfig(
properties=ModelProperties(context_size=131072, max_tokens=4096, mode=LLMMode.CHAT), features=[] properties=ModelProperties(context_size=131072, max_tokens=4096, mode=LLMMode.CHAT),
features=[ModelFeature.AGENT_THOUGHT],
), ),
"Skylark2-pro-4k": ModelConfig( "Skylark2-pro-4k": ModelConfig(
properties=ModelProperties(context_size=4096, max_tokens=4096, mode=LLMMode.CHAT), features=[] properties=ModelProperties(context_size=4096, max_tokens=4096, mode=LLMMode.CHAT),
features=[ModelFeature.AGENT_THOUGHT],
), ),
"Llama3-8B": ModelConfig( "Llama3-8B": ModelConfig(
properties=ModelProperties(context_size=8192, max_tokens=8192, mode=LLMMode.CHAT), features=[] properties=ModelProperties(context_size=8192, max_tokens=8192, mode=LLMMode.CHAT),
features=[ModelFeature.AGENT_THOUGHT],
), ),
"Llama3-70B": ModelConfig( "Llama3-70B": ModelConfig(
properties=ModelProperties(context_size=8192, max_tokens=8192, mode=LLMMode.CHAT), features=[] properties=ModelProperties(context_size=8192, max_tokens=8192, mode=LLMMode.CHAT),
features=[ModelFeature.AGENT_THOUGHT],
), ),
"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=[ModelFeature.TOOL_CALL], features=[ModelFeature.AGENT_THOUGHT, 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=[ModelFeature.TOOL_CALL], features=[ModelFeature.AGENT_THOUGHT, 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=[ModelFeature.TOOL_CALL], features=[ModelFeature.AGENT_THOUGHT, 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=[ModelFeature.TOOL_CALL], features=[ModelFeature.AGENT_THOUGHT, 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=[ModelFeature.TOOL_CALL], features=[ModelFeature.AGENT_THOUGHT, ModelFeature.TOOL_CALL],
), ),
"Mistral-7B": ModelConfig( "Mistral-7B": ModelConfig(
properties=ModelProperties(context_size=8192, max_tokens=2048, mode=LLMMode.CHAT), features=[] properties=ModelProperties(context_size=8192, max_tokens=2048, mode=LLMMode.CHAT),
features=[ModelFeature.AGENT_THOUGHT],
), ),
} }

24
api/core/model_runtime/model_providers/volcengine_maas/volcengine_maas.yaml
View File

@ -118,6 +118,30 @@ model_credential_schema:
type: select type: select
required: true required: true
options: options:
- label:
en_US: Doubao-1.5-vision-pro-32k
value: Doubao-1.5-vision-pro-32k
show_on:
- variable: __model_type
value: llm
- label:
en_US: Doubao-1.5-pro-32k
value: Doubao-1.5-pro-32k
show_on:
- variable: __model_type
value: llm
- label:
en_US: Doubao-1.5-lite-32k
value: Doubao-1.5-lite-32k
show_on:
- variable: __model_type
value: llm
- label:
en_US: Doubao-1.5-pro-256k
value: Doubao-1.5-pro-256k
show_on:
- variable: __model_type
value: llm
- label: - label:
en_US: Doubao-vision-pro-32k en_US: Doubao-vision-pro-32k
value: Doubao-vision-pro-32k value: Doubao-vision-pro-32k