Spaces:
Sleeping
Sleeping
disable retriver
Browse files- 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,
|