klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make automation should_notify check robust to non json mode chat models

Browse Source 
Use clean_json to handle automation should_notify check For gemini and
other chat models where enforcing json mode is problematic, not supported
This commit is contained in:
Debanjum
2025-01-06 18:52:54 +07:00
parent 9a5e3583cf
commit 266d274e21
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/khoj/routers/helpers.py
View File

@@ -1672,8 +1672,7 @@ def format_automation_response(scheduling_request: str, executed_query: str, ai_
Format the AI response to send in automation email to user.
"""
name = get_user_name(user)
if name:
username = prompts.user_name.format(name=name)
username = prompts.user_name.format(name=name) if name else ""
automation_format_prompt = prompts.automation_format_prompt.format(
original_query=scheduling_request,
@@ -1704,7 +1703,7 @@ def should_notify(original_query: str, executed_query: str, ai_response: str, us
try:
# TODO Replace with async call so we don't have to maintain a sync version
raw_response = send_message_to_model_wrapper_sync(to_notify_or_not, user=user, response_type="json_object")
response = json.loads(raw_response)
response = json.loads(clean_json(raw_response))
should_notify_result = response["decision"] == "Yes"
reason = response.get("reason", "unknown")
logger.info(