From c56f45bb86aa0774bb96ed2b7f79b46aca9ed203 Mon Sep 17 00:00:00 2001 From: Steve Androulakis Date: Fri, 24 Jan 2025 15:59:14 -0800 Subject: [PATCH] google gemini 2.0 exp by default --- activities/tool_activities.py | 2 +- workflows/tool_workflow.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/activities/tool_activities.py b/activities/tool_activities.py index a78cc7d..7acda91 100644 --- a/activities/tool_activities.py +++ b/activities/tool_activities.py @@ -105,7 +105,7 @@ class ToolActivities: genai.configure(api_key=api_key) model = genai.GenerativeModel( - "models/gemini-1.5-flash", + "models/gemini-2.0-flash-exp", system_instruction=input.context_instructions, ) response = model.generate_content(input.prompt) diff --git a/workflows/tool_workflow.py b/workflows/tool_workflow.py index f80ee7a..5f61149 100644 --- a/workflows/tool_workflow.py +++ b/workflows/tool_workflow.py @@ -60,6 +60,7 @@ class ToolWorkflow: self.prompt_queue.append( f"### The '{current_tool}' tool completed successfully with {dynamic_result}. " "INSTRUCTIONS: Parse this tool result as plain text, and use the system prompt containing the list of tools in sequence and the conversation history (and previous tool_results) to figure out next steps, if any. " + "You will need to use the tool_results to auto-fill arguments for subsequent tools and also to figure out if all tools have been run." '{"next": "", "tool": "", "args": {"": "", "": "}, "response": ""}' "ONLY return those json keys (next, tool, args, response), nothing else." 'Next should only be "done" if all tools have been run (use the system prompt to figure that out).'