Add sequence

Signed-off-by: Ricardo Zanini <[email protected]>

Author: ricardozanini

fluent/agentic/pom.xml

@@ -58,6 +58,10 @@
             <artifactId>mockito-core</artifactId>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.assertj</groupId>
+            <artifactId>assertj-core</artifactId>
+        </dependency>
         <dependency>
             <groupId>dev.langchain4j</groupId>
             <artifactId>langchain4j-ollama</artifactId>

fluent/agentic/src/main/java/io/serverlessworkflow/fluent/agentic/AgentDoTaskFluent.java

@@ -15,7 +15,19 @@
  */
 package io.serverlessworkflow.fluent.agentic;
 
+import java.util.UUID;
+
 public interface AgentDoTaskFluent<SELF extends AgentDoTaskFluent<SELF>> {
 
   SELF agent(String name, Object agent);
+
+  default SELF agent(Object agent) {
+    return agent(UUID.randomUUID().toString(), agent);
+  }
+
+  SELF sequence(String name, Object... agents);
+
+  default SELF sequence(Object... agents) {
+    return sequence("seq-" + UUID.randomUUID(), agents);
+  }
 }

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

@@ -42,6 +42,14 @@ public AgentTaskItemListBuilder agent(String name, Object agent) {
     return this;
   }
 
+  @Override
+  public AgentTaskItemListBuilder sequence(String name, Object... agents) {
+    for (int i = 0; i < agents.length; i++) {
+      agent(name + "-" + i, agents[i]);
+    }
+    return self();
+  }
+
   @Override
   public AgentTaskItemListBuilder self() {
     return this;

fluent/agentic/src/main/java/io/serverlessworkflow/fluent/agentic/DelegatingAgentDoTaskFluent.java

@@ -27,8 +27,16 @@ private AgentDoTaskFluent<SELF> d() {
   }
 
   @SuppressWarnings("unchecked")
+  @Override
   default SELF agent(String name, Object agent) {
     d().agent(name, agent);
     return (SELF) this;
   }
+
+  @SuppressWarnings("unchecked")
+  @Override
+  default SELF sequence(String name, Object... agents) {
+    d().sequence(name, agents);
+    return (SELF) this;
+  }
 }

fluent/agentic/src/test/java/io/serverlessworkflow/fluent/agentic/AgenticWorkflowBuilderTest.java

@@ -16,6 +16,7 @@
 package io.serverlessworkflow.fluent.agentic;
 
 import static io.serverlessworkflow.fluent.agentic.Models.BASE_MODEL;
+import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertInstanceOf;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
@@ -24,6 +25,7 @@
 import dev.langchain4j.agentic.AgentServices;
 import io.serverlessworkflow.api.types.Workflow;
 import io.serverlessworkflow.api.types.func.CallJava;
+import org.junit.jupiter.api.DisplayName;
 import org.junit.jupiter.api.Test;
 
 public class AgenticWorkflowBuilderTest {
@@ -46,4 +48,49 @@ public void verifyAgentCall() {
     assertEquals(1, workflow.getDo().size());
     assertInstanceOf(CallJava.class, workflow.getDo().get(0).getTask().getCallTask().get());
   }
+
+  @Test
+  void sequenceAgents() {
+    Agents.MovieExpert movieExpert = newMovieExpert();
+    Workflow wf =
+        AgenticWorkflowBuilder.workflow("seqFlow")
+            .tasks(d -> d.sequence("lineup", movieExpert, movieExpert))
+            .build();
+
+    assertThat(wf.getDo()).hasSize(2);
+    assertThat(wf.getDo().get(0).getName()).isEqualTo("lineup-0");
+    assertThat(wf.getDo().get(1).getName()).isEqualTo("lineup-1");
+    wf.getDo()
+        .forEach(
+            ti -> {
+              assertThat(ti.getTask().getCallTask()).isNotNull();
+              assertThat(ti.getTask().getCallTask().get()).isNotNull();
+            });
+  }
+
+  @Test
+  @DisplayName("Mix spec verbs with agent()")
+  void mixSpecAndAgent() {
+    Workflow wf =
+        AgenticWorkflowBuilder.workflow("mixFlow")
+            .tasks(
+                d ->
+                    d.set("init", s -> s.expr("$.mood = 'comedy'"))
+                        .agent("pickMovies", newMovieExpert())
+                        .set("done", "$.done = true"))
+            .build();
+
+    assertThat(wf.getDo()).hasSize(3);
+    assertThat(wf.getDo().get(0).getTask().getSetTask()).isNotNull();
+    assertThat(wf.getDo().get(1).getTask().getCallTask().get()).isNotNull();
+    assertThat(wf.getDo().get(2).getTask().getSetTask()).isNotNull();
+  }
+
+  private Agents.MovieExpert newMovieExpert() {
+    return spy(
+        AgentServices.agentBuilder(Agents.MovieExpert.class)
+            .outputName("movies")
+            .chatModel(BASE_MODEL)
+            .build());
+  }
 }