AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 06:29:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat:add deepseek r1 think display for ollama provider (#13272)

Browse Source
This commit is contained in:
呆萌闷油瓶 2025-02-06 15:32:10 +08:00 committed by GitHub
parent 49b4144ffd
commit 0d13aee15c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -314,6 +314,7 @@ class OllamaLargeLanguageModel(LargeLanguageModel):
""" """
full_text = "" full_text = ""
chunk_index = 0 chunk_index = 0
is_reasoning_started = False
def create_final_llm_result_chunk( def create_final_llm_result_chunk(
index: int, message: AssistantPromptMessage, finish_reason: str index: int, message: AssistantPromptMessage, finish_reason: str
@ -367,6 +368,14 @@ class OllamaLargeLanguageModel(LargeLanguageModel):
# transform assistant message to prompt message # transform assistant message to prompt message
text = chunk_json["response"] text = chunk_json["response"]
if "<think>" in text:
is_reasoning_started = True
text = text.replace("<think>", "> 💭 ")
elif "</think>" in text:
is_reasoning_started = False
text = text.replace("</think>", "") + "\n\n"
elif is_reasoning_started:
text = text.replace("\n", "\n> ")
assistant_prompt_message = AssistantPromptMessage(content=text) assistant_prompt_message = AssistantPromptMessage(content=text)