we can do better

Signed-off-by: Dmitrii Tikhomirov <[email protected]>

Author: treblereel

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

@@ -18,6 +18,7 @@
 import static io.serverlessworkflow.fluent.agentic.AgentWorkflowBuilder.workflow;
 import static io.serverlessworkflow.fluent.agentic.dsl.AgenticDSL.conditional;
 import static io.serverlessworkflow.fluent.agentic.dsl.AgenticDSL.doTasks;
+import static io.serverlessworkflow.fluent.agentic.dsl.AgenticDSL.fn;
 import static io.serverlessworkflow.fluent.agentic.dsl.AgenticDSL.loop;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -30,6 +31,7 @@
 import io.serverlessworkflow.api.types.func.CallTaskJava;
 import io.serverlessworkflow.api.types.func.ForTaskFunction;
 import io.serverlessworkflow.impl.WorkflowApplication;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.atomic.AtomicReference;
@@ -147,13 +149,49 @@ public void loopWorkflowWithMaxIterations() {
     assertThat(result).containsKey("story");
   }
 
+  public record EveningPlan(String movie, String meal) {}
+
   @Test
   @DisplayName("Parallel agents via DSL.parallel(...)")
   public void parallelWorkflow() {
     var foodExpert = AgentsUtils.newFoodExpert();
     var movieExpert = AgentsUtils.newMovieExpert();
 
-    Workflow wf = workflow("forkFlow").parallel("fanout", foodExpert, movieExpert).build();
+    workflow("forkFlow")
+        .tasks(
+            d ->
+                d.parallel(foodExpert, movieExpert)
+                    .callFn(
+                        fn(
+                            f -> {
+                              Map<String, List<String>> asMap = (Map<String, List<String>>) f;
+                              List<EveningPlan> result = new ArrayList<>();
+                              int max =
+                                  asMap.values().stream()
+                                      .map(List::size)
+                                      .min(Integer::compareTo)
+                                      .orElse(0);
+                              for (int i = 0; i < max; i++) {
+                                result.add(
+                                    new EveningPlan(
+                                        asMap.get("movies").get(i), asMap.get("meals").get(i)));
+                              }
+                              return result;
+                            })))
+        .build();
+
+    Workflow wf = workflow("forkFlow")
+            .tasks(d -> d
+                    .parallel("fanout", foodExpert, movieExpert)
+                    .callFn(fn((Map<String, List<String>> m) -> {
+                      var movies = m.getOrDefault("movies", List.of());
+                      var meals  = m.getOrDefault("meals",  List.of());
+                      return java.util.stream.IntStream
+                              .range(0, Math.min(movies.size(), meals.size()))
+                              .mapToObj(i -> new EveningPlan(movies.get(i), meals.get(i)))
+                              .toList();
+                    }))
+            ).build();
 
     List<TaskItem> items = wf.getDo();
     assertThat(items).hasSize(1);

experimental/fluent/agentic/src/test/java/io/serverlessworkflow/fluent/agentic/README.md

@@ -288,17 +288,17 @@ List<EveningPlan> plans = eveningPlannerAgent.plan("romantic");
 <td style="vertical-align:top;">
 <pre style="background:none; margin:0; padding:0; font-family:monospace; line-height:1.4;">
 <code class="language-java" style="background:none;white-space:pre;">
-Workflow wf = workflow("forkFlow")
-    .parallel(foodExpert, movieExpert)
-    .build();
-&nbsp;
-&nbsp;
-&nbsp;
-&nbsp;
-&nbsp;
-&nbsp;
-&nbsp;
-&nbsp;
+    Workflow wf = workflow("forkFlow")
+            .tasks(d -> d.parallel(foodExpert, movieExpert)
+                    .callFn(fn((Map&lt;String, List<String>> m) -> {
+                      var movies = m.getOrDefault("movies", List.of());
+                      var meals  = m.getOrDefault("meals",  List.of());
+                      return java.util.stream.IntStream
+                              .range(0, Math.min(movies.size(), meals.size()))
+                              .mapToObj(i -> new EveningPlan(movies.get(i), meals.get(i)))
+                              .toList();
+                    }))
+            ).build();
 &nbsp;
 &nbsp;
 &nbsp;
@@ -307,7 +307,7 @@ Workflow wf = workflow("forkFlow")
 &nbsp;
 Map&lt;String, Object> input = Map.of("mood", "I am hungry and bored");
 
-Map<String, Object> result = app.workflowDefinition(wf).instance(input).start().get().asMap().orElseThrow();
+List<EveningPlan> result = app.workflowDefinition(wf).instance(input).start().get().asMap().orElseThrow();
 
 </code>
 </pre>