wt002 commited on
Commit
3b31689
·
verified ·
1 Parent(s): 13117d0

Update agent.py

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