Update multi_agent.py
Browse files- 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("###
|
| 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 |
|