vindruid commited on
Commit
1b048a6
·
1 Parent(s): 4e1b432

disable retriver

Browse files
Files changed (1) hide show
  1. gaia_agent.py +2 -1
gaia_agent.py CHANGED
@@ -181,8 +181,9 @@ def build_graph(provider: str = "groq"):
181
  # builder.add_node("retriever", retriever)
182
  builder.add_node("assistant", assistant)
183
  builder.add_node("tools", ToolNode(tools))
184
- builder.add_edge(START, "retriever")
185
  # builder.add_edge("retriever", "assistant")
 
186
  builder.add_conditional_edges(
187
  "assistant",
188
  tools_condition,
 
181
  # builder.add_node("retriever", retriever)
182
  builder.add_node("assistant", assistant)
183
  builder.add_node("tools", ToolNode(tools))
184
+ # builder.add_edge(START, "retriever")
185
  # builder.add_edge("retriever", "assistant")
186
+ builder.add_edge(START, "assistant") ## if disable retriever
187
  builder.add_conditional_edges(
188
  "assistant",
189
  tools_condition,