From 16865d43a8195053a5f1fe45d414e5332a568c1d Mon Sep 17 00:00:00 2001 From: zhu-an <70234959+xhdd123321@users.noreply.github.com> Date: Thu, 6 Feb 2025 18:20:03 +0800 Subject: [PATCH] feat: add deepseek models for volcengine provider (#13283) Co-authored-by: zhaoqingyu.1075 --- .../volcengine_maas/llm/models.py | 16 +++++++++++++ .../volcengine_maas/volcengine_maas.yaml | 24 +++++++++++++++++++ 2 files changed, 40 insertions(+) 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