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 94315cd026..6a2b0e5087 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 @@ -18,6 +18,22 @@ class ModelConfig(BaseModel): configs: dict[str, ModelConfig] = { + "DeepSeek-R1-Distill-Qwen-32B": ModelConfig( + properties=ModelProperties(context_size=64000, max_tokens=8192, mode=LLMMode.CHAT), + features=[ModelFeature.AGENT_THOUGHT], + ), + "DeepSeek-R1-Distill-Qwen-7B": ModelConfig( + properties=ModelProperties(context_size=64000, max_tokens=8192, mode=LLMMode.CHAT), + features=[ModelFeature.AGENT_THOUGHT], + ), + "DeepSeek-R1": ModelConfig( + properties=ModelProperties(context_size=64000, max_tokens=8192, mode=LLMMode.CHAT), + features=[ModelFeature.AGENT_THOUGHT], + ), + "DeepSeek-V3": ModelConfig( + properties=ModelProperties(context_size=64000, max_tokens=8192, mode=LLMMode.CHAT), + features=[ModelFeature.AGENT_THOUGHT, ModelFeature.TOOL_CALL, ModelFeature.STREAM_TOOL_CALL], + ), "Doubao-1.5-vision-pro-32k": ModelConfig( properties=ModelProperties(context_size=32768, max_tokens=12288, mode=LLMMode.CHAT), features=[ModelFeature.AGENT_THOUGHT, ModelFeature.VISION], diff --git a/api/core/model_runtime/model_providers/volcengine_maas/volcengine_maas.yaml b/api/core/model_runtime/model_providers/volcengine_maas/volcengine_maas.yaml index 976cd26ac9..210ac1cdf1 100644 --- a/api/core/model_runtime/model_providers/volcengine_maas/volcengine_maas.yaml +++ b/api/core/model_runtime/model_providers/volcengine_maas/volcengine_maas.yaml @@ -118,6 +118,30 @@ model_credential_schema: type: select required: true options: + - label: + en_US: DeepSeek-R1-Distill-Qwen-32B + value: DeepSeek-R1-Distill-Qwen-32B + show_on: + - variable: __model_type + value: llm + - label: + en_US: DeepSeek-R1-Distill-Qwen-7B + value: DeepSeek-R1-Distill-Qwen-7B + show_on: + - variable: __model_type + value: llm + - label: + en_US: DeepSeek-R1 + value: DeepSeek-R1 + show_on: + - variable: __model_type + value: llm + - label: + en_US: DeepSeek-V3 + value: DeepSeek-V3 + show_on: + - variable: __model_type + value: llm - label: en_US: Doubao-1.5-vision-pro-32k value: Doubao-1.5-vision-pro-32k