feat(graph): rename AgentEnum to FlowAgentEnum and update references

Aias00 · Aias00 · commit d536eba875b2 · 2025-08-24T19:57:31.000+08:00
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/agent/LlmRoutingAgent.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/agent/LlmRoutingAgent.java
@@ -20,7 +20,7 @@
 import com.alibaba.cloud.ai.graph.StateGraph;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowAgentBuilder;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowGraphBuilder;
-import com.alibaba.cloud.ai.graph.agent.flow.enums.AgentEnum;
+import com.alibaba.cloud.ai.graph.agent.flow.enums.FlowAgentEnum;
 import com.alibaba.cloud.ai.graph.exception.GraphRunnerException;
 import com.alibaba.cloud.ai.graph.exception.GraphStateException;
 import org.springframework.ai.chat.model.ChatModel;
@@ -48,7 +48,7 @@ public Optional<OverAllState> invoke(Map<String, Object> input) throws GraphStat
 	@Override
 	protected StateGraph buildSpecificGraph(FlowGraphBuilder.FlowGraphConfig config) throws GraphStateException {
 		config.setChatModel(this.chatModel);
-		return FlowGraphBuilder.buildGraph(AgentEnum.ROUTING.getType(), config);
+		return FlowGraphBuilder.buildGraph(FlowAgentEnum.ROUTING.getType(), config);
 	}
 
 	public static LlmRoutingAgentBuilder builder() {
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/agent/ParallelAgent.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/agent/ParallelAgent.java
@@ -21,7 +21,7 @@
 import com.alibaba.cloud.ai.graph.agent.BaseAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowAgentBuilder;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowGraphBuilder;
-import com.alibaba.cloud.ai.graph.agent.flow.enums.AgentEnum;
+import com.alibaba.cloud.ai.graph.agent.flow.enums.FlowAgentEnum;
 import com.alibaba.cloud.ai.graph.exception.GraphRunnerException;
 import com.alibaba.cloud.ai.graph.exception.GraphStateException;
 import org.slf4j.Logger;
@@ -79,7 +79,7 @@ protected StateGraph buildSpecificGraph(FlowGraphBuilder.FlowGraphConfig config)
 		config.customProperty("mergeStrategy", this.mergeStrategy);
 		config.customProperty("maxConcurrency", this.maxConcurrency);
 
-		return FlowGraphBuilder.buildGraph(AgentEnum.PARALLEL.getType(), config);
+		return FlowGraphBuilder.buildGraph(FlowAgentEnum.PARALLEL.getType(), config);
 	}
 
 	/**
@@ -163,9 +163,13 @@ protected ParallelAgentBuilder self() {
 		 */
 		@Override
 		protected void validate() {
-			super.validate();
+			// Validate name first (from parent)
+			if (name == null || name.trim().isEmpty()) {
+				throw new IllegalArgumentException("Name must be provided");
+			}
 
-			// Validate minimum sub-agent count
+			// Validate minimum sub-agent count for ParallelAgent (skip parent subAgents
+			// check)
 			if (subAgents == null || subAgents.size() < 2) {
 				throw new IllegalArgumentException(
 						"ParallelAgent requires at least 2 sub-agents for parallel execution, but got: "
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/agent/SequentialAgent.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/agent/SequentialAgent.java
@@ -20,7 +20,7 @@
 import com.alibaba.cloud.ai.graph.StateGraph;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowAgentBuilder;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowGraphBuilder;
-import com.alibaba.cloud.ai.graph.agent.flow.enums.AgentEnum;
+import com.alibaba.cloud.ai.graph.agent.flow.enums.FlowAgentEnum;
 import com.alibaba.cloud.ai.graph.exception.GraphRunnerException;
 import com.alibaba.cloud.ai.graph.exception.GraphStateException;
 
@@ -43,7 +43,7 @@ public Optional<OverAllState> invoke(Map<String, Object> input) throws GraphStat
 
 	@Override
 	protected StateGraph buildSpecificGraph(FlowGraphBuilder.FlowGraphConfig config) throws GraphStateException {
-		return FlowGraphBuilder.buildGraph(AgentEnum.SEQUENTIAL.getType(), config);
+		return FlowGraphBuilder.buildGraph(FlowAgentEnum.SEQUENTIAL.getType(), config);
 	}
 
 	public static SequentialAgentBuilder builder() {
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/builder/FlowGraphBuilder.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/builder/FlowGraphBuilder.java
@@ -18,7 +18,7 @@
 import com.alibaba.cloud.ai.graph.KeyStrategyFactory;
 import com.alibaba.cloud.ai.graph.StateGraph;
 import com.alibaba.cloud.ai.graph.agent.BaseAgent;
-import com.alibaba.cloud.ai.graph.agent.flow.registry.FlowGraphBuildingStrategyRegistry;
+import com.alibaba.cloud.ai.graph.agent.flow.strategy.FlowGraphBuildingStrategyRegistry;
 import com.alibaba.cloud.ai.graph.agent.flow.strategy.FlowGraphBuildingStrategy;
 import com.alibaba.cloud.ai.graph.exception.GraphStateException;
 import org.springframework.ai.chat.model.ChatModel;
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/enums/FlowAgentEnum.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/enums/FlowAgentEnum.java
@@ -15,13 +15,13 @@
  */
 package com.alibaba.cloud.ai.graph.agent.flow.enums;
 
-public enum AgentEnum {
+public enum FlowAgentEnum {
 
 	CONDITIONAL("CONDITIONAL"), SEQUENTIAL("SEQUENTIAL"), ROUTING("ROUTING"), PARALLEL("PARALLEL");
 
 	private final String type;
 
-	AgentEnum(final String type) {
+	FlowAgentEnum(final String type) {
 		this.type = type;
 	}
 
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/ConditionalGraphBuildingStrategy.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/ConditionalGraphBuildingStrategy.java
@@ -22,7 +22,7 @@
 import com.alibaba.cloud.ai.graph.agent.BaseAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.agent.FlowAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowGraphBuilder;
-import com.alibaba.cloud.ai.graph.agent.flow.enums.AgentEnum;
+import com.alibaba.cloud.ai.graph.agent.flow.enums.FlowAgentEnum;
 import com.alibaba.cloud.ai.graph.agent.flow.node.ConditionEvaluator;
 import com.alibaba.cloud.ai.graph.agent.flow.node.ConditionEvaluatorAction;
 import com.alibaba.cloud.ai.graph.agent.flow.node.TransparentNode;
@@ -84,7 +84,7 @@ public StateGraph buildGraph(FlowGraphBuilder.FlowGraphConfig config) throws Gra
 
 	@Override
 	public String getStrategyType() {
-		return AgentEnum.CONDITIONAL.getType();
+		return FlowAgentEnum.CONDITIONAL.getType();
 	}
 
 	@Override
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/FlowGraphBuildingStrategyRegistry.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/FlowGraphBuildingStrategyRegistry.java
@@ -13,18 +13,12 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package com.alibaba.cloud.ai.graph.agent.flow.registry;
+package com.alibaba.cloud.ai.graph.agent.flow.strategy;
 
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.Map;
 import java.util.Set;
 
-import com.alibaba.cloud.ai.graph.agent.flow.strategy.ConditionalGraphBuildingStrategy;
-import com.alibaba.cloud.ai.graph.agent.flow.strategy.FlowGraphBuildingStrategy;
-import com.alibaba.cloud.ai.graph.agent.flow.strategy.ParallelGraphBuildingStrategy;
-import com.alibaba.cloud.ai.graph.agent.flow.strategy.RoutingGraphBuildingStrategy;
-import com.alibaba.cloud.ai.graph.agent.flow.strategy.SequentialGraphBuildingStrategy;
-
 /**
  * Registry for FlowGraphStrategy implementations. This allows for dynamic registration
  * and lookup of graph building strategies, enabling extensibility without modifying core
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/ParallelGraphBuildingStrategy.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/ParallelGraphBuildingStrategy.java
@@ -19,7 +19,7 @@
 import com.alibaba.cloud.ai.graph.agent.BaseAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.agent.FlowAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowGraphBuilder;
-import com.alibaba.cloud.ai.graph.agent.flow.enums.AgentEnum;
+import com.alibaba.cloud.ai.graph.agent.flow.enums.FlowAgentEnum;
 import com.alibaba.cloud.ai.graph.agent.flow.node.EnhancedParallelResultAggregator;
 import com.alibaba.cloud.ai.graph.agent.flow.node.ParallelResultAggregator;
 import com.alibaba.cloud.ai.graph.agent.flow.node.TransparentNode;
@@ -86,7 +86,7 @@ public StateGraph buildGraph(FlowGraphBuilder.FlowGraphConfig config) throws Gra
 
 	@Override
 	public String getStrategyType() {
-		return AgentEnum.PARALLEL.getType();
+		return FlowAgentEnum.PARALLEL.getType();
 	}
 
 	@Override
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/RoutingGraphBuildingStrategy.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/RoutingGraphBuildingStrategy.java
@@ -22,7 +22,7 @@
 import com.alibaba.cloud.ai.graph.agent.BaseAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.agent.FlowAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowGraphBuilder;
-import com.alibaba.cloud.ai.graph.agent.flow.enums.AgentEnum;
+import com.alibaba.cloud.ai.graph.agent.flow.enums.FlowAgentEnum;
 import com.alibaba.cloud.ai.graph.agent.flow.node.RoutingEdgeAction;
 import com.alibaba.cloud.ai.graph.agent.flow.node.TransparentNode;
 import com.alibaba.cloud.ai.graph.exception.GraphStateException;
@@ -81,7 +81,7 @@ public StateGraph buildGraph(FlowGraphBuilder.FlowGraphConfig config) throws Gra
 
 	@Override
 	public String getStrategyType() {
-		return AgentEnum.ROUTING.getType();
+		return FlowAgentEnum.ROUTING.getType();
 	}
 
 	@Override
diff --git a/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/SequentialGraphBuildingStrategy.java b/spring-ai-alibaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/strategy/SequentialGraphBuildingStrategy.java
@@ -19,7 +19,7 @@
 import com.alibaba.cloud.ai.graph.agent.BaseAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.agent.FlowAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowGraphBuilder;
-import com.alibaba.cloud.ai.graph.agent.flow.enums.AgentEnum;
+import com.alibaba.cloud.ai.graph.agent.flow.enums.FlowAgentEnum;
 import com.alibaba.cloud.ai.graph.agent.flow.node.TransparentNode;
 import com.alibaba.cloud.ai.graph.exception.GraphStateException;
 
@@ -66,7 +66,7 @@ public StateGraph buildGraph(FlowGraphBuilder.FlowGraphConfig config) throws Gra
 
 	@Override
 	public String getStrategyType() {
-		return AgentEnum.SEQUENTIAL.getType();
+		return FlowAgentEnum.SEQUENTIAL.getType();
 	}
 
 	@Override
diff --git a/spring-ai-alibaba-graph-core/src/test/java/com/alibaba/cloud/ai/graph/agent/flow/FlowAgentArchitectureTest.java b/spring-ai-alibaba-graph-core/src/test/java/com/alibaba/cloud/ai/graph/agent/flow/FlowAgentArchitectureTest.java
@@ -22,8 +22,8 @@
 import com.alibaba.cloud.ai.graph.agent.flow.agent.SequentialAgent;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowAgentBuilder;
 import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowGraphBuilder;
-import com.alibaba.cloud.ai.graph.agent.flow.enums.AgentEnum;
-import com.alibaba.cloud.ai.graph.agent.flow.registry.FlowGraphBuildingStrategyRegistry;
+import com.alibaba.cloud.ai.graph.agent.flow.enums.FlowAgentEnum;
+import com.alibaba.cloud.ai.graph.agent.flow.strategy.FlowGraphBuildingStrategyRegistry;
 import com.alibaba.cloud.ai.graph.agent.flow.strategy.FlowGraphBuildingStrategy;
 import com.alibaba.cloud.ai.graph.agent.flow.strategy.SequentialGraphBuildingStrategy;
 import com.alibaba.cloud.ai.graph.state.strategy.AppendStrategy;
@@ -180,13 +180,13 @@ void testStrategyRegistryExtensibility() {
 		FlowGraphBuildingStrategyRegistry registry = FlowGraphBuildingStrategyRegistry.getInstance();
 
 		// Verify default strategies are registered
-		assertTrue(registry.hasStrategy(AgentEnum.SEQUENTIAL.getType()));
-		assertTrue(registry.hasStrategy(AgentEnum.ROUTING.getType()));
-		assertTrue(registry.hasStrategy(AgentEnum.PARALLEL.getType()));
-		assertTrue(registry.hasStrategy(AgentEnum.CONDITIONAL.getType()));
+		assertTrue(registry.hasStrategy(FlowAgentEnum.SEQUENTIAL.getType()));
+		assertTrue(registry.hasStrategy(FlowAgentEnum.ROUTING.getType()));
+		assertTrue(registry.hasStrategy(FlowAgentEnum.PARALLEL.getType()));
+		assertTrue(registry.hasStrategy(FlowAgentEnum.CONDITIONAL.getType()));
 
 		// Test getting strategies
-		FlowGraphBuildingStrategy sequentialStrategy = registry.getStrategy(AgentEnum.SEQUENTIAL.getType());
+		FlowGraphBuildingStrategy sequentialStrategy = registry.getStrategy(FlowAgentEnum.SEQUENTIAL.getType());
 		assertNotNull(sequentialStrategy);
 		assertTrue(sequentialStrategy instanceof SequentialGraphBuildingStrategy);
 
