bstraehle commited on
Commit
40b8162
·
verified ·
1 Parent(s): bb9aede

Update multi_agent.py

Browse files
Files changed (1) hide show
  1. multi_agent.py +5 -3
multi_agent.py CHANGED
@@ -47,7 +47,7 @@ def create_agent(llm: ChatOpenAI, tools: list, system_prompt: str):
47
 
48
  def agent_node(state, agent, name):
49
  try:
50
- print("### node=" + name)
51
  result = agent.invoke(state)
52
  return {"messages": [HumanMessage(content=result["output"], name=name)]}
53
  except Exception as e:
@@ -196,9 +196,11 @@ def create_graph():
196
  return graph.compile()
197
 
198
  def should_continue(state):
199
- print("#### should_continue")
200
  global move_num, num_moves, legal_moves
201
-
 
 
202
  if move_num == num_moves:
203
  return END # max moves reached
204
 
 
47
 
48
  def agent_node(state, agent, name):
49
  try:
50
+ print("### node=" + name + " ###")
51
  result = agent.invoke(state)
52
  return {"messages": [HumanMessage(content=result["output"], name=name)]}
53
  except Exception as e:
 
196
  return graph.compile()
197
 
198
  def should_continue(state):
199
+ print("### should_continue")
200
  global move_num, num_moves, legal_moves
201
+ print("### move_num=" + str(move_num))
202
+ print("### num_moves=" + str(num_moves))
203
+
204
  if move_num == num_moves:
205
  return END # max moves reached
206