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

Enable using Operator with Gemini models

Browse Source
This commit is contained in:
Debanjum
2025-05-08 11:11:28 -06:00
parent 0d8fb667ec
commit b279060e2c
2 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/khoj/processor/conversation/google/gemini_chat.py
View File

@@ -166,6 +166,7 @@ async def converse_gemini(
user_query, user_query,
online_results: Optional[Dict[str, Dict]] = None, online_results: Optional[Dict[str, Dict]] = None,
code_results: Optional[Dict[str, Dict]] = None, code_results: Optional[Dict[str, Dict]] = None,
operator_results: Optional[Dict[str, Dict]] = None,
conversation_log={}, conversation_log={},
model: Optional[str] = "gemini-2.0-flash", model: Optional[str] = "gemini-2.0-flash",
api_key: Optional[str] = None, api_key: Optional[str] = None,
@@ -238,6 +239,10 @@ async def converse_gemini(
context_message += ( context_message += (
f"{prompts.code_executed_context.format(code_results=truncate_code_context(code_results))}\n\n" 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() context_message = context_message.strip()
# Setup Prompt with Primer or Conversation History # Setup Prompt with Primer or Conversation History

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

@@ -1507,9 +1507,10 @@ async def agenerate_chat_response(
chat_response_generator = converse_gemini( chat_response_generator = converse_gemini(
compiled_references, compiled_references,
query_to_run, query_to_run,
online_results, online_results=online_results,
code_results, code_results=code_results,
meta_log, operator_results=operator_results,
conversation_log=meta_log,
model=chat_model.name, model=chat_model.name,
api_key=api_key, api_key=api_key,
api_base_url=api_base_url, api_base_url=api_base_url,