fix: fix StateGraph with overAllStateFactory (alibaba#1054)

zhouyou9505 · yuluo-yx · Aias00 · web-flow · commit c75a9e72fdda · 2025-06-02T16:39:37.000+08:00
fix StateGraph with overAllStateFactory

Co-authored-by: shown &lt;yuluo08290126@gmail.com&gt;
Co-authored-by: aias00 &lt;liuhongyu@apache.org&gt;
Co-authored-by: Ken Liu &lt;ken.lj.hz@gmail.com&gt;
diff --git a/spring-ai-alibaba-graph/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/ReactAgent.java b/spring-ai-alibaba-graph/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/ReactAgent.java
@@ -165,7 +165,7 @@ private StateGraph initGraph() throws GraphStateException {
 			};
 		}
 
-		return new StateGraph().addNode("agent", node_async(this.llmNode))
+		return new StateGraph(this.overAllStateFactory).addNode("agent", node_async(this.llmNode))
 			.addNode("tool", node_async(this.toolNode))
 			.addEdge(START, "agent")
 			.addConditionalEdges("agent", edge_async(this::think), Map.of("continue", "tool", "end", END))

Original file line number	Diff line number	Diff line change
`@@ -165,7 +165,7 @@ private StateGraph initGraph() throws GraphStateException {`
`165`	`165`	`};`
`166`	`166`	`}`
`167`	`167`
`168`		`- return new StateGraph().addNode("agent", node_async(this.llmNode))`
	`168`	`+ return new StateGraph(this.overAllStateFactory).addNode("agent", node_async(this.llmNode))`
`169`	`169`	`.addNode("tool", node_async(this.toolNode))`
`170`	`170`	`.addEdge(START, "agent")`
`171`	`171`	`.addConditionalEdges("agent", edge_async(this::think), Map.of("continue", "tool", "end", END))`