Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -214,9 +214,9 @@ def build_graph(provider: str = "google"):
|
|
214 |
#builder.add_edge("tools", "assistant")
|
215 |
|
216 |
builder = StateGraph(MessagesState)
|
217 |
-
|
218 |
builder.add_node("assistant", assistant)
|
219 |
-
|
220 |
# Retriever ist Start und Endpunkt
|
221 |
builder.set_entry_point("retriever")
|
222 |
builder.set_finish_point("retriever")
|
|
|
214 |
#builder.add_edge("tools", "assistant")
|
215 |
|
216 |
builder = StateGraph(MessagesState)
|
217 |
+
builder.add_node("retriever", retriever)
|
218 |
builder.add_node("assistant", assistant)
|
219 |
+
|
220 |
# Retriever ist Start und Endpunkt
|
221 |
builder.set_entry_point("retriever")
|
222 |
builder.set_finish_point("retriever")
|