Procházet zdrojové kódy

feat: add deepseek models for volcengine provider (#13283)

Co-authored-by: zhaoqingyu.1075 <zhaoqingyu.1075@bytedance.com>
tags/0.15.3
zhu-an před 8 měsíci
rodič
revize
16865d43a8
Žádný účet není propojen s e-mailovou adresou tvůrce revize

+ 16
- 0
api/core/model_runtime/model_providers/volcengine_maas/llm/models.py Zobrazit soubor

@@ -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],

+ 24
- 0
api/core/model_runtime/model_providers/volcengine_maas/volcengine_maas.yaml Zobrazit soubor

@@ -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

Načítá se…
Zrušit
Uložit