From 84729c36244b37df40a2f52eec30d2b1f80c8cd6 Mon Sep 17 00:00:00 2001 From: MNE2142 Date: Tue, 22 Apr 2025 12:31:33 +0300 Subject: [PATCH] Update agent.py changed model to gemini so it's the default choice as per the docs --- utils/agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/agent.py b/utils/agent.py index 5ae9b61..57b5ff7 100644 --- a/utils/agent.py +++ b/utils/agent.py @@ -13,7 +13,7 @@ def create_clevrr_agent(model, prompt): df_locals = {} df_locals["pg"] = pg tools = [PythonAstREPLTool(locals=df_locals), get_screen_info] - model = MODELS["openai"] + model = MODELS["gemini"] agent = create_react_agent(model, tools, prompt) agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, handle_parsing_errors=True, return_intermediate_steps=True) - return agent_executor \ No newline at end of file + return agent_executor