rsobieski commited on
Commit
ef38dc6
·
verified ·
1 Parent(s): 1488b1e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -18,7 +18,7 @@ def build_graph():
18
 
19
  # Initialize the LLM (e.g., Gemini Flash, zero temperature)
20
  # llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0)
21
- llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", temperature=0)
22
  llm_with_tools = llm.bind_tools(TOOLS)
23
 
24
  # Step 1: Retriever node
 
18
 
19
  # Initialize the LLM (e.g., Gemini Flash, zero temperature)
20
  # llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0)
21
+ llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
22
  llm_with_tools = llm.bind_tools(TOOLS)
23
 
24
  # Step 1: Retriever node