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

Fix: Removed model params except max_token for deepseek r1 in volcengine (#13446)

Browse Source
This commit is contained in:
Fei He 2025-02-10 10:26:26 +08:00 committed by GitHub
parent 75113c26c6
commit 7796984444
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 63 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
api/core/model_runtime/model_providers/volcengine_maas/llm/llm.py
View File

@ -328,6 +328,19 @@ class VolcengineMaaSLargeLanguageModel(LargeLanguageModel):
"""
model_config = get_model_config(credentials)
if model.startswith("DeepSeek-R1"):
rules = [
ParameterRule(
name="max_tokens",
type=ParameterType.INT,
use_template="max_tokens",
min=1,
max=model_config.properties.max_tokens,
default=512,
label=I18nObject(zh_Hans="最大生成长度", en_US="Max Tokens"),
),
]
else:
rules = [
ParameterRule(
name="temperature",
@ -342,7 +355,11 @@ class VolcengineMaaSLargeLanguageModel(LargeLanguageModel):
label=I18nObject(zh_Hans="Top P", en_US="Top P"),
),
ParameterRule(
name="top_k", type=ParameterType.INT, min=1, default=1, label=I18nObject(zh_Hans="Top K", en_US="Top K")
name="top_k",
type=ParameterType.INT,
min=1,
default=1,
label=I18nObject(zh_Hans="Top K", en_US="Top K"),
),
ParameterRule(
name="presence_penalty",