@@ -140,8 +140,10 @@ class Agent():
|
|||||||
Remove the reasoning block of reasoning model like deepseek.
|
Remove the reasoning block of reasoning model like deepseek.
|
||||||
"""
|
"""
|
||||||
end_tag = "</think>"
|
end_tag = "</think>"
|
||||||
end_idx = text.rfind(end_tag)+8
|
end_idx = text.rfind(end_tag)
|
||||||
return text[end_idx:]
|
if end_idx == -1:
|
||||||
|
return text
|
||||||
|
return text[end_idx+8:]
|
||||||
|
|
||||||
def extract_reasoning_text(self, text: str) -> None:
|
def extract_reasoning_text(self, text: str) -> None:
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user