Update agent.py
Browse files
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.
|
| 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
|