Adding agentic workflow patterns to the DSL

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

Author: ricardozanini

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

@@ -15,9 +15,11 @@
  */
 package io.serverlessworkflow.fluent.agentic;
 
+import dev.langchain4j.agentic.scope.AgenticScope;
 import io.serverlessworkflow.fluent.func.spi.FuncTransformations;
 import io.serverlessworkflow.fluent.spec.BaseWorkflowBuilder;
 import java.util.UUID;
+import java.util.function.Predicate;
 
 public class AgentWorkflowBuilder
     extends BaseWorkflowBuilder<AgentWorkflowBuilder, AgentDoTaskBuilder, AgentTaskItemListBuilder>
@@ -40,6 +42,40 @@ public static AgentWorkflowBuilder workflow(String name, String ns) {
     return new AgentWorkflowBuilder(name, ns, DEFAULT_VERSION);
   }
 
+  public AgentWorkflowBuilder sequence(Object... agents) {
+    return sequence(UUID.randomUUID().toString(), agents);
+  }
+
+  public AgentWorkflowBuilder sequence(String name, Object... agents) {
+    final AgentDoTaskBuilder doTaskBuilder = this.newDo();
+    doTaskBuilder.sequence(name, agents);
+    this.workflow.setDo(doTaskBuilder.build().getDo());
+    return this;
+  }
+
+  public AgentWorkflowBuilder parallel(Object... agents) {
+    return this.parallel(UUID.randomUUID().toString(), agents);
+  }
+
+  public AgentWorkflowBuilder parallel(String name, Object... agents) {
+    final AgentDoTaskBuilder doTaskBuilder = this.newDo();
+    doTaskBuilder.parallel(name, agents);
+    this.workflow.setDo(doTaskBuilder.build().getDo());
+    return this;
+  }
+
+  public AgentWorkflowBuilder loop(Predicate<AgenticScope> exitCondition, Object... agents) {
+    return this.loop(UUID.randomUUID().toString(), exitCondition, agents);
+  }
+
+  public AgentWorkflowBuilder loop(
+      String name, Predicate<AgenticScope> exitCondition, Object... agents) {
+    final AgentDoTaskBuilder doTaskBuilder = this.newDo();
+    doTaskBuilder.loop(name, loop -> loop.subAgents(agents).exitCondition(exitCondition));
+    this.workflow.setDo(doTaskBuilder.build().getDo());
+    return this;
+  }
+
   @Override
   protected AgentDoTaskBuilder newDo() {
     return new AgentDoTaskBuilder();

experimental/fluent/agentic/src/main/java/io/serverlessworkflow/fluent/agentic/dsl/AgenticDSL.java

@@ -15,6 +15,7 @@
  */
 package io.serverlessworkflow.fluent.agentic.dsl;
 
+import dev.langchain4j.agentic.scope.AgenticScope;
 import io.cloudevents.CloudEventData;
 import io.serverlessworkflow.api.types.FlowDirectiveEnum;
 import io.serverlessworkflow.fluent.agentic.AgentDoTaskBuilder;
@@ -107,6 +108,19 @@ public static <T> Consumer<FuncEmitTaskBuilder> event(
     return event -> event.event(e -> e.type(type).data(function, clazz));
   }
 
+  // -------- Agentic Workflow Patterns -------- //
+  public static AgentTaskConfigurer sequence(Object... agents) {
+    return list -> list.sequence(agents);
+  }
+
+  public static AgentTaskConfigurer loop(Predicate<AgenticScope> exitCondition, Object... agents) {
+    return list -> list.loop(l -> l.subAgents(agents).exitCondition(exitCondition));
+  }
+
+  public static AgentTaskConfigurer parallel(Object... agents) {
+    return list -> list.parallel(agents);
+  }
+
   // --------- Tasks ------ //
   public static Consumer<AgentDoTaskBuilder> doTasks(AgentTaskConfigurer... steps) {
     Objects.requireNonNull(steps, "Steps in a tasks are required");

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

@@ -15,6 +15,7 @@
  */
 package io.serverlessworkflow.fluent.agentic;
 
+import static io.serverlessworkflow.fluent.agentic.AgentWorkflowBuilder.workflow;
 import static org.assertj.core.api.Assertions.assertThat;
 
 import io.serverlessworkflow.api.types.TaskItem;
@@ -34,11 +35,24 @@ void dslSequentialAgents() {
     var a2 = AgentsUtils.newMovieExpert();
     var a3 = AgentsUtils.newMovieExpert();
 
-    Workflow wf =
-        AgentWorkflowBuilder.workflow("seqFlow")
-            .tasks(tasks -> tasks.sequence("process", a1, a2, a3))
-            .build();
+    Workflow wf = workflow("seqFlow").tasks(tasks -> tasks.sequence("process", a1, a2, a3)).build();
 
+    this.assertSequentialAgents(wf);
+  }
+
+  @Test
+  @DisplayName("Sequential agents via DSL.sequence(...)")
+  void dslSequentialAgentsShortcut() {
+    var a1 = AgentsUtils.newMovieExpert();
+    var a2 = AgentsUtils.newMovieExpert();
+    var a3 = AgentsUtils.newMovieExpert();
+
+    Workflow wf = workflow("seqFlow").sequence("process", a1, a2, a3).build();
+
+    this.assertSequentialAgents(wf);
+  }
+
+  private void assertSequentialAgents(Workflow wf) {
     List<TaskItem> items = wf.getDo();
     assertThat(items).hasSize(3);
     // names should be process-0, process-1, process-2
@@ -53,7 +67,7 @@ void dslSequentialAgents() {
   @DisplayName("Bare Java‑bean call via DSL.callFn(...)")
   void dslCallFnBare() {
     Workflow wf =
-        AgentWorkflowBuilder.workflow("beanCall")
+        workflow("beanCall")
             .tasks(tasks -> tasks.callFn("plainCall", fn -> fn.function(ctx -> "pong")))
             .build();
 
@@ -71,14 +85,7 @@ void dslLoopAgents() {
 
     Workflow wf =
         AgentWorkflowBuilder.workflow("retryFlow")
-            .tasks(
-                tasks ->
-                    tasks.loop(
-                        "reviewLoop",
-                        loop ->
-                            loop.maxIterations(5)
-                                .exitCondition(c -> c.readState("score", 0).doubleValue() > 0.75)
-                                .subAgents("reviewer", scorer, editor)))
+            .loop("reviewLoop", c -> c.readState("score", 0).doubleValue() > 0.75, scorer, editor)
             .build();
 
     List<TaskItem> items = wf.getDo();
@@ -96,10 +103,7 @@ void dslParallelAgents() {
     var a1 = AgentsUtils.newMovieExpert();
     var a2 = AgentsUtils.newMovieExpert();
 
-    Workflow wf =
-        AgentWorkflowBuilder.workflow("forkFlow")
-            .tasks(tasks -> tasks.parallel("fanout", a1, a2))
-            .build();
+    Workflow wf = workflow("forkFlow").parallel("fanout", a1, a2).build();
 
     List<TaskItem> items = wf.getDo();
     assertThat(items).hasSize(1);
@@ -117,7 +121,7 @@ void dslMixSpecAndAgent() {
     var agent = AgentsUtils.newMovieExpert();
 
     Workflow wf =
-        AgentWorkflowBuilder.workflow("mixedFlow")
+        workflow("mixedFlow")
             .tasks(
                 tasks ->
                     tasks

fluent/spec/src/main/java/io/serverlessworkflow/fluent/spec/BaseWorkflowBuilder.java

@@ -34,7 +34,7 @@ public abstract class BaseWorkflowBuilder<
   public static final String DEFAULT_VERSION = "0.0.1";
   public static final String DEFAULT_NAMESPACE = "org.acme";
 
-  private final Workflow workflow;
+  protected final Workflow workflow;
   private final Document document;
 
   protected BaseWorkflowBuilder(final String name, final String namespace, final String version) {
@@ -87,7 +87,11 @@ public SELF use(Consumer<UseBuilder> useBuilderConsumer) {
   public SELF tasks(Consumer<DBuilder> doTaskConsumer) {
     final DBuilder doTaskBuilder = newDo();
     doTaskConsumer.accept(doTaskBuilder);
-    this.workflow.setDo(doTaskBuilder.build().getDo());
+    if (this.workflow.getDo() == null) {
+      this.workflow.setDo(doTaskBuilder.build().getDo());
+    } else {
+      this.workflow.getDo().addAll(doTaskBuilder.build().getDo());
+    }
     return self();
   }