Skip to content

feat(graph): refactor agent api #2234

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Aug 24, 2025
Merged

feat(graph): refactor agent api #2234

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
82fe9a2
feat(graph): refactor agent package structure and add new flow graph …
Aias00 Aug 24, 2025
d536eba
feat(graph): rename AgentEnum to FlowAgentEnum and update references
Aias00 Aug 24, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
131 changes: 0 additions & 131 deletions ...libaba-graph-core/src/main/java/com/alibaba/cloud/ai/graph/agent/flow/BuilderExample.java
View file
Open in desktop

This file was deleted.

34 changes: 20 additions & 14 deletions .../cloud/ai/graph/agent/flow/FlowAgent.java → .../ai/graph/agent/flow/agent/FlowAgent.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alibaba.cloud.ai.graph.agent.flow;
package com.alibaba.cloud.ai.graph.agent.flow.agent;

import java.util.List;

Expand All @@ -24,9 +24,9 @@
import com.alibaba.cloud.ai.graph.action.AsyncNodeAction;
import com.alibaba.cloud.ai.graph.agent.BaseAgent;
import com.alibaba.cloud.ai.graph.agent.ReactAgent;
import com.alibaba.cloud.ai.graph.agent.flow.builder.FlowGraphBuilder;
import com.alibaba.cloud.ai.graph.exception.GraphStateException;

import static com.alibaba.cloud.ai.graph.StateGraph.START;
import static com.alibaba.cloud.ai.graph.action.AsyncNodeAction.node_async;

public abstract class FlowAgent extends BaseAgent {
Expand Down Expand Up @@ -54,20 +54,26 @@ protected FlowAgent(String name, String description, String outputKey, String in
}

protected StateGraph initGraph() throws GraphStateException {
StateGraph graph = new StateGraph(this.name(), keyStrategyFactory);

// add root agent
graph.addNode(this.name(), node_async(new TransparentNode(this.outputKey, this.inputKey)));

// add starting edge
graph.addEdge(START, this.name());
// Use recursive method to add all sub-agents
processSubAgents(graph, this, this.subAgents());

return graph;
// Use FlowGraphBuilder to construct the graph
FlowGraphBuilder.FlowGraphConfig config = FlowGraphBuilder.FlowGraphConfig.builder()
.name(this.name())
.keyStrategyFactory(keyStrategyFactory)
.rootAgent(this)
.subAgents(this.subAgents());

// Delegate to specific graph builder based on agent type
return buildSpecificGraph(config);
}

protected abstract void processSubAgents(StateGraph graph, BaseAgent parentAgent, List<BaseAgent> subAgents)
/**
* Abstract method for subclasses to specify their graph building strategy. This
* method should be implemented by concrete FlowAgent subclasses to define how their
* specific graph structure should be built.
* @param config the graph configuration
* @return the constructed StateGraph
* @throws GraphStateException if graph construction fails
*/
protected abstract StateGraph buildSpecificGraph(FlowGraphBuilder.FlowGraphConfig config)
throws GraphStateException;

@Override
Expand Down
62 changes: 9 additions & 53 deletions .../ai/graph/agent/flow/LlmRoutingAgent.java → ...aph/agent/flow/agent/LlmRoutingAgent.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,23 +13,20 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alibaba.cloud.ai.graph.agent.flow;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
package com.alibaba.cloud.ai.graph.agent.flow.agent;

import com.alibaba.cloud.ai.graph.CompiledGraph;
import com.alibaba.cloud.ai.graph.OverAllState;
import com.alibaba.cloud.ai.graph.StateGraph;
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.FlowAgentEnum;
import com.alibaba.cloud.ai.graph.exception.GraphRunnerException;
import com.alibaba.cloud.ai.graph.exception.GraphStateException;

import org.springframework.ai.chat.model.ChatModel;

import static com.alibaba.cloud.ai.graph.StateGraph.END;
import java.util.Map;
import java.util.Optional;

public class LlmRoutingAgent extends FlowAgent {

Expand All @@ -48,51 +45,10 @@ public Optional<OverAllState> invoke(Map<String, Object> input) throws GraphStat
return compiledGraph.invoke(input);
}

// protected StateGraph initGraph() throws GraphStateException {
// StateGraph graph = new StateGraph(this.name(), keyStrategyFactory);
//
// // add root agent
// graph.addNode(this.name(), node_async(LlmNode.builder().(this.outputKey,
// this.inputKey)));
//
// // add starting edge
// graph.addEdge(START, this.name());
// // Use recursive method to add all sub-agents
// processSubAgents(graph, this, this.subAgents());
//
// return graph;
// }

/**
* Recursively adds sub-agents and their nested sub-agents to the graph
* @param graph the StateGraph to add nodes and edges to
* @param parentAgent the name of the parent node
* @param subAgents the list of sub-agents to process
*/
@Override
protected void processSubAgents(StateGraph graph, BaseAgent parentAgent, List<BaseAgent> subAgents)
throws GraphStateException {
Map<String, String> edgeRoutingMap = new HashMap<>();
for (BaseAgent subAgent : subAgents) {
// Add the current sub-agent as a node
graph.addNode(subAgent.name(), subAgent.asAsyncNodeAction(parentAgent.outputKey(), subAgent.outputKey()));
// graph.addEdge(parentAgent.name(), subAgent.name());
edgeRoutingMap.put(subAgent.name(), subAgent.name());

// Recursively process this sub-agent's sub-agents if they exist
if (subAgent instanceof FlowAgent subFlowAgent) {
if (subFlowAgent.subAgents() == null || subFlowAgent.subAgents().isEmpty()) {
graph.addEdge(subAgent.name(), END);
}
}
else {
graph.addEdge(subAgent.name(), END);
}
}

// Connect parent to this sub-agent
graph.addConditionalEdges(parentAgent.name(), new RoutingEdgeAction(chatModel, this, subAgents),
edgeRoutingMap);
protected StateGraph buildSpecificGraph(FlowGraphBuilder.FlowGraphConfig config) throws GraphStateException {
config.setChatModel(this.chatModel);
return FlowGraphBuilder.buildGraph(FlowAgentEnum.ROUTING.getType(), config);
}

public static LlmRoutingAgentBuilder builder() {
Expand Down
Loading
Loading