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

fix(model_runtime): ensure compatibility with O1 models by adjusting token parameters (#10537)

Browse Source
This commit is contained in:
-LAN- 2024-11-11 16:06:53 +08:00 committed by GitHub
parent 9018ef30fe
commit 867bf70f1a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/model_runtime/model_providers/openai/llm/llm.py
View File

@ -617,6 +617,10 @@ class OpenAILargeLanguageModel(_CommonOpenAI, LargeLanguageModel):
# o1 compatibility # o1 compatibility
block_as_stream = False block_as_stream = False
if model.startswith("o1"): if model.startswith("o1"):
if "max_tokens" in model_parameters:
model_parameters["max_completion_tokens"] = model_parameters["max_tokens"]
del model_parameters["max_tokens"]
if stream: if stream:
block_as_stream = True block_as_stream = True
stream = False stream = False