klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 21:29:13 +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. Format the AI response to send in automation email to user.
""" """
name = get_user_name(user) name = get_user_name(user)
if name: username = prompts.user_name.format(name=name) if name else ""
username = prompts.user_name.format(name=name)
automation_format_prompt = prompts.automation_format_prompt.format( automation_format_prompt = prompts.automation_format_prompt.format(
original_query=scheduling_request, original_query=scheduling_request,
@@ -1704,7 +1703,7 @@ def should_notify(original_query: str, executed_query: str, ai_response: str, us
try: try:
# TODO Replace with async call so we don't have to maintain a sync version # 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") 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" should_notify_result = response["decision"] == "Yes"
reason = response.get("reason", "unknown") reason = response.get("reason", "unknown")
logger.info( logger.info(