Skip to content

Commit 4e02ea4

Browse files
authored
use agent name as branch name (#841)
Signed-off-by: Dmitrii Tikhomirov <[email protected]>
1 parent cc4fdea commit 4e02ea4

File tree

3 files changed

+15
-7
lines changed

3 files changed

+15
-7
lines changed

experimental/fluent/agentic/src/main/java/io/serverlessworkflow/fluent/agentic/AgentTaskItemListBuilder.java

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -92,10 +92,9 @@ public AgentTaskItemListBuilder parallel(String name, Object... agents) {
9292
List<AgentExecutor> execs = AgentAdapters.toExecutors(agents);
9393
for (int i = 0; i < execs.size(); i++) {
9494
AgentExecutor ex = execs.get(i);
95-
fork.branch(
96-
"branch-" + i + "-" + name,
97-
AgentAdapters.toFunction(ex),
98-
DefaultAgenticScope.class);
95+
String agentName = ex.agentName() != null ? ex.agentName() : "branch-" + i + "-" + name;
96+
97+
fork.branch(agentName, AgentAdapters.toFunction(ex), DefaultAgenticScope.class);
9998
}
10099
});
101100
return self();

experimental/fluent/agentic/src/test/java/io/serverlessworkflow/fluent/agentic/AgentDslWorkflowTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ void dslLoopAgents() {
101101
@Test
102102
void dslParallelAgents() {
103103
var a1 = AgentsUtils.newMovieExpert();
104-
var a2 = AgentsUtils.newMovieExpert();
104+
var a2 = AgentsUtils.newMovieExpert2();
105105

106106
Workflow wf = workflow("forkFlow").parallel("fanout", a1, a2).build();
107107

@@ -112,8 +112,8 @@ void dslParallelAgents() {
112112
// two branches created
113113
assertThat(fork.getFork().getBranches()).hasSize(2);
114114
// branch names follow "branch-{index}-{name}"
115-
assertThat(fork.getFork().getBranches().get(0).getName()).isEqualTo("branch-0-fanout");
116-
assertThat(fork.getFork().getBranches().get(1).getName()).isEqualTo("branch-1-fanout");
115+
assertThat(fork.getFork().getBranches().get(0).getName()).isEqualTo("findMovie");
116+
assertThat(fork.getFork().getBranches().get(1).getName()).isEqualTo("findMovie2");
117117
}
118118

119119
@Test

experimental/fluent/agentic/src/test/java/io/serverlessworkflow/fluent/agentic/AgentsUtils.java

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,15 @@ public static Agents.MovieExpert newMovieExpert() {
3232
.build());
3333
}
3434

35+
public static Agents.MovieExpert newMovieExpert2() {
36+
return spy(
37+
AgenticServices.agentBuilder(Agents.MovieExpert.class)
38+
.outputName("movies")
39+
.name("findMovie2")
40+
.chatModel(BASE_MODEL)
41+
.build());
42+
}
43+
3544
public static Agents.CreativeWriter newCreativeWriter() {
3645
return spy(
3746
AgenticServices.agentBuilder(Agents.CreativeWriter.class)

0 commit comments

Comments
 (0)