TySS-Dev/ollama-ai-answers-searxng
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Changed text output method
CI Test Guard / validate-code (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Tyler
2026-05-14 22:06:02 -04:00
parent dab3f293a1
commit de6156933a
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ai_answers.py
+2 -3
View File
@@ -1146,8 +1146,6 @@ class SXNGPlugin(Plugin):
"max_tokens": self.max_tokens, "max_tokens": self.max_tokens,
"temperature": self.temperature "temperature": self.temperature
} }
if self.provider == 'ollama':
payload_dict["think"] = False
payload = json.dumps(payload_dict) payload = json.dumps(payload_dict)
headers = { headers = {
"Content-Type": "application/json", "Content-Type": "application/json",
@@ -1174,7 +1172,8 @@ class SXNGPlugin(Plugin):
return '', "No choices in API response." return '', "No choices in API response."
message = choices[0].get("message", {}) message = choices[0].get("message", {})
content = message.get("content") or "" content = message.get("content") or ""
reasoning = message.get("reasoning_content") or "" content = re.sub(r'<think>.*?</think>', '', content, flags=re.DOTALL).strip()
reasoning = message.get("reasoning") or message.get("reasoning_content") or ""
full = (f"<think>\n{reasoning}\n</think>\n\n" if reasoning else "") + content full = (f"<think>\n{reasoning}\n</think>\n\n" if reasoning else "") + content
return full, None return full, None
except Exception as e: except Exception as e: