BocchiMing commited on
Commit
d9fbcde
·
verified ·
1 Parent(s): e7300af

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
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)