HCho commited on
Commit
59976dd
·
verified ·
1 Parent(s): dacffdf

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -105,7 +105,7 @@ tools = [
105
  def build_graph(provider: str = "google"):
106
  if provider == "google":
107
  # Google Gemini
108
- llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-preview-05-20", temperature=0,api_key=google_api_key)
109
  # Bind tools to LLM
110
  llm_with_tools = llm.bind_tools(tools)
111
 
 
105
  def build_graph(provider: str = "google"):
106
  if provider == "google":
107
  # Google Gemini
108
+ llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0,api_key=google_api_key)
109
  # Bind tools to LLM
110
  llm_with_tools = llm.bind_tools(tools)
111