Added AgenticServices helper

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

Author: treblereel

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

@@ -212,8 +212,13 @@ public interface StyleReviewLoop {
   public interface StyledWriter extends AgenticScopeAccess {
 
     @Agent
-    ResultWithAgenticScope<String> writeStoryWithStyle(
-        @V("topic") String topic, @V("style") String style);
+    String writeStoryWithStyle(@V("topic") String topic, @V("style") String style);
+  }
+
+  public interface NovelCreator {
+
+    @Agent
+    String createNovel(@V("topic") String topic, @V("audience") String audience, @V("style") String style);
   }
 
   public interface FoodExpert {
@@ -250,4 +255,11 @@ public interface EveningPlannerAgent {
     @Agent
     List<EveningPlan> plan(@V("mood") String mood);
   }
+
+  public interface HoroscopeAgent {
+
+    @Agent
+    String invoke(@V("name") String name);
+  }
+
 }

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

@@ -15,19 +15,31 @@
  */
 package io.serverlessworkflow.fluent.agentic.langchain4j;
 
+import static io.serverlessworkflow.fluent.agentic.AgentWorkflowBuilder.workflow;
+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.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;
 import static org.mockito.Mockito.verify;
 
-import dev.langchain4j.agentic.AgenticServices;
 import dev.langchain4j.agentic.UntypedAgent;
+import dev.langchain4j.agentic.scope.AgenticScope;
 import dev.langchain4j.agentic.workflow.WorkflowAgentsBuilder;
+
+import java.util.List;
 import java.util.Map;
+import java.util.function.Function;
+import java.util.function.Predicate;
+
+import io.serverlessworkflow.fluent.agentic.AgenticServices;
+import io.serverlessworkflow.fluent.agentic.AgentsUtils;
 import org.junit.jupiter.api.Test;
 
 public class WorkflowAgentsIT {
@@ -38,21 +50,21 @@ void sequential_agents_tests() {
 
     CreativeWriter creativeWriter =
         spy(
-            AgenticServices.agentBuilder(CreativeWriter.class)
+            dev.langchain4j.agentic.AgenticServices.agentBuilder(CreativeWriter.class)
                 .chatModel(BASE_MODEL)
                 .outputName("story")
                 .build());
 
     AudienceEditor audienceEditor =
         spy(
-            AgenticServices.agentBuilder(AudienceEditor.class)
+            dev.langchain4j.agentic.AgenticServices.agentBuilder(AudienceEditor.class)
                 .chatModel(BASE_MODEL)
                 .outputName("story")
                 .build());
 
     StyleEditor styleEditor =
         spy(
-            AgenticServices.agentBuilder(StyleEditor.class)
+            dev.langchain4j.agentic.AgenticServices.agentBuilder(StyleEditor.class)
                 .chatModel(BASE_MODEL)
                 .outputName("story")
                 .build());
@@ -77,4 +89,76 @@ void sequential_agents_tests() {
     verify(audienceEditor).editStory(any(), eq("young adults"));
     verify(styleEditor).editStory(any(), eq("fantasy"));
   }
+
+  @Test
+  public void sequenceHelperTest() {
+    var creativeWriter = AgentsUtils.newCreativeWriter();
+    var audienceEditor = AgentsUtils.newAudienceEditor();
+    var styleEditor = AgentsUtils.newStyleEditor();
+
+    NovelCreator novelCreator = AgenticServices.of(NovelCreator.class)
+            .flow(workflow("seqFlow")
+                    .sequence(creativeWriter, audienceEditor, styleEditor)
+            ).build();
+
+    String story = novelCreator.createNovel("dragons and wizards", "young adults", "fantasy");
+    assertNotNull(story);
+  }
+
+  @Test
+  public void parallelWorkflow() {
+    var foodExpert = AgentsUtils.newFoodExpert();
+    var movieExpert = AgentsUtils.newMovieExpert();
+
+    EveningPlannerAgent eveningPlannerAgent = AgenticServices.of(EveningPlannerAgent.class)
+            .flow(workflow("parallelFlow")
+                    .parallel(foodExpert, movieExpert)
+            ).build();
+    List<EveningPlan> result = eveningPlannerAgent.plan("romantic");
+    assertTrue(result.size() > 0);
+  }
+
+  @Test
+  public void loopTest() {
+    var creativeWriter = AgentsUtils.newCreativeWriter();
+    var scorer = AgentsUtils.newStyleScorer();
+    var editor = AgentsUtils.newStyleEditor();
+
+    Predicate<AgenticScope> until = s -> s.readState("score", 0).doubleValue() >= 0.8;
+
+
+    StyledWriter styledWriter = AgenticServices.of(StyledWriter.class)
+            .flow(workflow("loopFlow").agent(creativeWriter)
+                    .loop(until, scorer, editor)
+            ).build();
+
+    String story = styledWriter.writeStoryWithStyle("dragons and wizards", "fantasy");
+    assertNotNull(story);
+  }
+
+  @Test
+  public void humanInTheLoop() {
+    var astrologyAgent = AgentsUtils.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;
+      }
+    };
+
+    String result = AgenticServices.of(HoroscopeAgent.class)
+            .flow(workflow("humanInTheLoop")
+                    .tasks(tasks -> tasks.callFn(fn(askSign)))
+                    .agent(astrologyAgent))
+            .build()
+            .invoke("My name is Mario. What is my horoscope?");
+
+    assertNotNull(result);
+
+  }
+
 }

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

@@ -77,12 +77,12 @@ public void sequentialWorkflow() {
   @Test
   @DisplayName("Looping agents via DSL.loop(...)")
   public void loopWorkflow() {
-
+    var creativeWriter = AgentsUtils.newCreativeWriter();
     var scorer = AgentsUtils.newStyleScorer();
     var editor = AgentsUtils.newStyleEditor();
 
     Workflow wf =
-        AgentWorkflowBuilder.workflow("retryFlow")
+        AgentWorkflowBuilder.workflow("retryFlow").agent(creativeWriter)
             .loop("reviewLoop", c -> c.readState("score", 0).doubleValue() >= 0.8, scorer, editor)
             .build();