parallelWorkflow() works

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

Author: treblereel

experimental/fluent/agentic-langchain4j/src/test/java/io/serverlessworkflow/fluent/agentic/langchain4j/WorkflowAgentsIT.java

@@ -24,15 +24,13 @@
 import static io.serverlessworkflow.fluent.agentic.AgentsUtils.newStyleEditor;
 import static io.serverlessworkflow.fluent.agentic.AgentsUtils.newStyleScorer;
 import static io.serverlessworkflow.fluent.agentic.AgentsUtils.newSummaryStory;
-import static io.serverlessworkflow.fluent.agentic.dsl.AgenticDSL.fn;
 import static io.serverlessworkflow.fluent.agentic.langchain4j.Agents.*;
 import static io.serverlessworkflow.fluent.agentic.langchain4j.Agents.AudienceEditor;
 import static io.serverlessworkflow.fluent.agentic.langchain4j.Agents.CreativeWriter;
 import static io.serverlessworkflow.fluent.agentic.langchain4j.Agents.StyleEditor;
 import static io.serverlessworkflow.fluent.agentic.langchain4j.Models.BASE_MODEL;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
-import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.spy;
@@ -48,7 +46,6 @@
 import java.util.function.Function;
 import java.util.function.Predicate;
 import java.util.stream.IntStream;
-
 import org.junit.jupiter.api.Test;
 
 public class WorkflowAgentsIT {
@@ -106,9 +103,9 @@ public void sequenceHelperTest() {
     var styleEditor = newStyleEditor();
 
     AgentsUtils.NovelCreator novelCreator =
-            io.serverlessworkflow.fluent.agentic.AgenticServices.of(AgentsUtils.NovelCreator.class)
-                    .flow(workflow("seqFlow").sequence(creativeWriter, audienceEditor, styleEditor))
-                    .build();
+        io.serverlessworkflow.fluent.agentic.AgenticServices.of(AgentsUtils.NovelCreator.class)
+            .flow(workflow("seqFlow").sequence(creativeWriter, audienceEditor, styleEditor))
+            .build();
 
     String story = novelCreator.createNovel("dragons and wizards", "young adults", "fantasy");
     assertNotNull(story);
@@ -121,9 +118,9 @@ public void agentAndSequenceHelperTest() {
     var styleEditor = newStyleEditor();
 
     AgentsUtils.NovelCreator novelCreator =
-            io.serverlessworkflow.fluent.agentic.AgenticServices.of(AgentsUtils.NovelCreator.class)
-                    .flow(workflow("seqFlow").agent(creativeWriter).sequence(audienceEditor, styleEditor))
-                    .build();
+        io.serverlessworkflow.fluent.agentic.AgenticServices.of(AgentsUtils.NovelCreator.class)
+            .flow(workflow("seqFlow").agent(creativeWriter).sequence(audienceEditor, styleEditor))
+            .build();
 
     String story = novelCreator.createNovel("dragons and wizards", "young adults", "fantasy");
     assertNotNull(story);
@@ -137,13 +134,13 @@ public void agentAndSequenceAndAgentHelperTest() {
     var summaryStory = newSummaryStory();
 
     AgentsUtils.NovelCreator novelCreator =
-            io.serverlessworkflow.fluent.agentic.AgenticServices.of(AgentsUtils.NovelCreator.class)
-                    .flow(
-                            workflow("seqFlow")
-                                    .agent(creativeWriter)
-                                    .sequence(audienceEditor, styleEditor)
-                                    .agent(summaryStory))
-                    .build();
+        io.serverlessworkflow.fluent.agentic.AgenticServices.of(AgentsUtils.NovelCreator.class)
+            .flow(
+                workflow("seqFlow")
+                    .agent(creativeWriter)
+                    .sequence(audienceEditor, styleEditor)
+                    .agent(summaryStory))
+            .build();
 
     String story = novelCreator.createNovel("dragons and wizards", "young adults", "fantasy");
     assertNotNull(story);
@@ -155,20 +152,20 @@ public void parallelWorkflow() {
     var movieExpert = newMovieExpert();
 
     Function<Map<String, List<String>>, List<EveningPlan>> planEvening =
-            input -> {
-              List<String> movies = input.get("findMovie");
-              List<String> meals = input.get("findMeal");
+        input -> {
+          List<String> movies = input.get("findMovie");
+          List<String> meals = input.get("findMeal");
 
-              int max = Math.min(movies.size(), meals.size());
-              return IntStream.range(0, max)
-                      .mapToObj(i -> new EveningPlan(movies.get(i), meals.get(i)))
-                      .toList();
-            };
+          int max = Math.min(movies.size(), meals.size());
+          return IntStream.range(0, max)
+              .mapToObj(i -> new EveningPlan(movies.get(i), meals.get(i)))
+              .toList();
+        };
 
     EveningPlannerAgent eveningPlannerAgent =
-            AgenticServices.of(EveningPlannerAgent.class)
-                    .flow(workflow("parallelFlow").parallel(foodExpert, movieExpert).outputAs(planEvening))
-                    .build();
+        AgenticServices.of(EveningPlannerAgent.class)
+            .flow(workflow("parallelFlow").parallel(foodExpert, movieExpert).outputAs(planEvening))
+            .build();
     List<EveningPlan> result = eveningPlannerAgent.plan("romantic");
     assertEquals(3, result.size());
   }
@@ -182,9 +179,9 @@ public void loopTest() {
     Predicate<AgenticScope> until = s -> s.readState("score", 0.0) >= 0.8;
 
     StyledWriter styledWriter =
-            AgenticServices.of(StyledWriter.class)
-                    .flow(workflow("loopFlow").agent(creativeWriter).loop(until, scorer, editor))
-                    .build();
+        AgenticServices.of(StyledWriter.class)
+            .flow(workflow("loopFlow").agent(creativeWriter).loop(until, scorer, editor))
+            .build();
 
     String story = styledWriter.writeStoryWithStyle("dragons and wizards", "fantasy");
     assertNotNull(story);
@@ -195,25 +192,25 @@ public void humanInTheLoop() {
     var astrologyAgent = newAstrologyAgent();
 
     var askSign =
-            new Function<Map<String, Object>, Map<String, Object>>() {
-              @Override
-              public Map<String, Object> apply(Map<String, Object> holder) {
-                System.out.println("What's your star sign?");
-                // var sign = System.console().readLine();
-                holder.put("sign", "piscis");
-                return holder;
-              }
-            };
+        new Function<Map<String, Object>, Map<String, Object>>() {
+          @Override
+          public Map<String, Object> apply(Map<String, Object> holder) {
+            System.out.println("What's your star sign?");
+            // var sign = System.console().readLine();
+            holder.put("sign", "piscis");
+            return holder;
+          }
+        };
 
     String result =
-            AgenticServices.of(Agents.HoroscopeAgent.class)
-                    .flow(
-                            workflow("humanInTheLoop")
-                                    .inputFrom(askSign)
-                                    // .tasks(tasks -> tasks.callFn(fn(askSign))) // TODO should work too
-                                    .agent(astrologyAgent))
-                    .build()
-                    .invoke("My name is Mario. What is my horoscope?");
+        AgenticServices.of(Agents.HoroscopeAgent.class)
+            .flow(
+                workflow("humanInTheLoop")
+                    .inputFrom(askSign)
+                    // .tasks(tasks -> tasks.callFn(fn(askSign))) // TODO should work too
+                    .agent(astrologyAgent))
+            .build()
+            .invoke("My name is Mario. What is my horoscope?");
 
     assertNotNull(result);
   }