AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-17 11:35:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(model_providers/ollama): Fix OllamaLargeLanguageModel to correctly set the stop option (#5217)

Browse Source
This commit is contained in:
-LAN- 2024-06-14 18:26:14 +08:00 committed by GitHub
parent b7ff765d8d
commit 7f44e88eda
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/model_runtime/model_providers/ollama/llm/llm.py
View File

@ -195,7 +195,7 @@ class OllamaLargeLanguageModel(LargeLanguageModel):
data["options"] = model_parameters or {} data["options"] = model_parameters or {}
if stop: if stop:
data["stop"] = "\n".join(stop) data["options"]["stop"] = stop
completion_type = LLMMode.value_of(credentials["mode"]) completion_type = LLMMode.value_of(credentials["mode"])