Update agent.py
Browse files
agent.py
CHANGED
|
@@ -181,7 +181,7 @@ tools = [
|
|
| 181 |
arvix_search,
|
| 182 |
create_retriever_tool
|
| 183 |
]
|
| 184 |
-
def build_graph()
|
| 185 |
"""Build the graph"""
|
| 186 |
llm = init_chat_model("google_genai:gemini-2.0-flash",google_api_key=os.environ["GOOGLE_API_KEY"])
|
| 187 |
llm_with_tools = llm.bind_tools(tools)
|
|
|
|
| 181 |
arvix_search,
|
| 182 |
create_retriever_tool
|
| 183 |
]
|
| 184 |
+
def build_graph():
|
| 185 |
"""Build the graph"""
|
| 186 |
llm = init_chat_model("google_genai:gemini-2.0-flash",google_api_key=os.environ["GOOGLE_API_KEY"])
|
| 187 |
llm_with_tools = llm.bind_tools(tools)
|