luping85 commited on
Commit
044c5b5
·
verified ·
1 Parent(s): c2f2d62

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -13,7 +13,7 @@ from tools import web_search_tool, repl_tool
13
  GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
14
 
15
  # chat_model = ChatOpenAI(model='gpt-4o-mini', temperature=0, api_key=OPENAI_API_KEY)
16
- chat_model = ChatGoogleGenerativeAI(model="gemini-2.0-flash", google_api_key=GOOGLE_API_KEY)
17
  tools = [web_search_tool, repl_tool]
18
  chat_model_with_tools = chat_model.bind_tools(tools)
19
 
 
13
  GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
14
 
15
  # chat_model = ChatOpenAI(model='gpt-4o-mini', temperature=0, api_key=OPENAI_API_KEY)
16
+ chat_model = ChatGoogleGenerativeAI(model="gemini-2.5-flash-preview-04-17", google_api_key=GOOGLE_API_KEY)
17
  tools = [web_search_tool, repl_tool]
18
  chat_model_with_tools = chat_model.bind_tools(tools)
19