From af5f74f2ad76ad8b180e6ec0d5c121c599169be3 Mon Sep 17 00:00:00 2001 From: Quanzheng Long Date: Fri, 13 Mar 2026 13:27:32 -0700 Subject: [PATCH] more --- langgraph-go/go.mod | 1 - .../tests/advanced-graph/test_sub_agents.py | 16 ++++++---------- 2 files changed, 6 insertions(+), 11 deletions(-) diff --git a/langgraph-go/go.mod b/langgraph-go/go.mod index d297fbf70..58b57b372 100644 --- a/langgraph-go/go.mod +++ b/langgraph-go/go.mod @@ -1,4 +1,3 @@ module github.com/langchain-ai/langgraph/langgraph-go go 1.25 - diff --git a/libs/langgraph/tests/advanced-graph/test_sub_agents.py b/libs/langgraph/tests/advanced-graph/test_sub_agents.py index 388d49b44..bb8f343b8 100644 --- a/libs/langgraph/tests/advanced-graph/test_sub_agents.py +++ b/libs/langgraph/tests/advanced-graph/test_sub_agents.py @@ -78,10 +78,7 @@ def build_main_agent(planner: MockLLM, sub_agent: Any) -> Any: return Command( goto=Send( order_food_node, - { - "state": state, - "complete": decision.complete or "order flow completed", - }, + decision.complete or "order flow completed", ) ) if decision.type == "sub_agent" and decision.sub_agent: @@ -89,7 +86,7 @@ def build_main_agent(planner: MockLLM, sub_agent: Any) -> Any: if decision.type == "tool" and decision.tool: sends.append(Send("tool_node", decision.tool)) # Keep the main loop responsive: wait for one inbound message and continue. - sends.append(Send("wait_node", state)) + sends.append(Send("wait_node", None)) return Command(goto=sends) async def wait_node(ctx: Context, state: MainAgentState) -> Command: @@ -112,11 +109,11 @@ def build_main_agent(planner: MockLLM, sub_agent: Any) -> Any: elif channel == "user_input_channel": state["output"].append(f"user_input: {payload}") # State changed -> ask planner what to do next. - return Command(goto=Send("llm_node", state)) + return Command(goto=Send("llm_node", None)) else: state["output"].append("timer: no updates yet") # No meaningful state change -> keep waiting without calling planner. - return Command(goto=Send("wait_node", state)) + return Command(goto=Send("wait_node", None)) async def tool_node(ctx: Context, tool_input: str) -> None: await asyncio.sleep(0.1) @@ -138,9 +135,8 @@ def build_main_agent(planner: MockLLM, sub_agent: Any) -> Any: sub_agent_output["output"], ) - async def order_food_node(payload: dict[str, Any]) -> dict[str, Any]: - state = payload["state"] - complete_message = payload["complete"] + async def order_food_node(input: str, state: MainAgentState) -> dict[str, Any]: + complete_message = input return { "done": complete_message, "output": [*state["output"], f"order_food: {complete_message}"],