diff --git a/src/khoj/processor/conversation/google/gemini_chat.py b/src/khoj/processor/conversation/google/gemini_chat.py index 5f45f69e..621d3eb5 100644 --- a/src/khoj/processor/conversation/google/gemini_chat.py +++ b/src/khoj/processor/conversation/google/gemini_chat.py @@ -166,6 +166,7 @@ async def converse_gemini( user_query, online_results: Optional[Dict[str, Dict]] = None, code_results: Optional[Dict[str, Dict]] = None, + operator_results: Optional[Dict[str, Dict]] = None, conversation_log={}, model: Optional[str] = "gemini-2.0-flash", api_key: Optional[str] = None, @@ -238,6 +239,10 @@ async def converse_gemini( context_message += ( f"{prompts.code_executed_context.format(code_results=truncate_code_context(code_results))}\n\n" ) + if ConversationCommand.Operator in conversation_commands and not is_none_or_empty(operator_results): + context_message += ( + f"{prompts.operator_execution_context.format(operator_results=yaml_dump(operator_results))}\n\n" + ) context_message = context_message.strip() # Setup Prompt with Primer or Conversation History diff --git a/src/khoj/routers/helpers.py b/src/khoj/routers/helpers.py index bde6aafd..4836d962 100644 --- a/src/khoj/routers/helpers.py +++ b/src/khoj/routers/helpers.py @@ -1507,9 +1507,10 @@ async def agenerate_chat_response( chat_response_generator = converse_gemini( compiled_references, query_to_run, - online_results, - code_results, - meta_log, + online_results=online_results, + code_results=code_results, + operator_results=operator_results, + conversation_log=meta_log, model=chat_model.name, api_key=api_key, api_base_url=api_base_url,