Hugging Face's logo

Spaces:
luping85
/
Agent_Course_Final_Assignment
Runtime error

App Files Community
luping85 commited on
Commit
cf4a854
·
verified ·
1 Parent(s): 6d444b4

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 tavily_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.5-flash-preview-04-17", google_api_key=GOOGLE_API_KEY)
17
  tools = [tavily_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.0-flash", google_api_key=GOOGLE_API_KEY)
17
  tools = [tavily_search_tool, repl_tool]
18
  chat_model_with_tools = chat_model.bind_tools(tools)
19