Enable using Operator with Gemini models

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

View File

@@ -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

View File

@@ -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,