KPatelis commited on
Commit
8df4ab3
·
1 Parent(s): 5f8bfc7

fix langgraph workflow

Browse files
Files changed (1) hide show
  1. agent.py +10 -10
agent.py CHANGED
@@ -62,20 +62,20 @@ def processor_node(state: MessagesState):
62
  return {"messages": response}
63
 
64
  def agent_graph():
65
- builder = StateGraph(MessagesState)
66
 
67
  ## Add nodes
68
- builder.add_node("retriever_node", retriever_node)
69
- builder.add_node("processor_node", processor_node)
70
- builder.add_node("tools", ToolNode(tools))
71
 
72
  ## Add edges
73
- builder.add_edge(START, "retriever_node")
74
- builder.add_edge("retriever_node", "processor_node")
75
- builder.add_conditional_edges("processor_node", tools_condition)
76
- builder.add_edge("tools", "processor_node")
77
 
78
  # Compile graph
79
- builder.compile()
80
 
81
- return builder
 
62
  return {"messages": response}
63
 
64
  def agent_graph():
65
+ workflow = StateGraph(MessagesState)
66
 
67
  ## Add nodes
68
+ workflow.add_node("retriever_node", retriever_node)
69
+ workflow.add_node("processor_node", processor_node)
70
+ workflow.add_node("tools", ToolNode(tools))
71
 
72
  ## Add edges
73
+ workflow.add_edge(START, "retriever_node")
74
+ workflow.add_edge("retriever_node", "processor_node")
75
+ workflow.add_conditional_edges("processor_node", tools_condition)
76
+ workflow.add_edge("tools", "processor_node")
77
 
78
  # Compile graph
79
+ graph = workflow.compile()
80
 
81
+ return graph