more

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

Author: treblereel

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

@@ -140,4 +140,31 @@ interface SceneAgent extends AgentInstance {
     @Agent("Generates the opening scene of the story from a plot outline.")
     String invoke(@V("plot") String plot);
   }
-}
+
+  interface MeetingInvitationDraft extends AgentInstance {
+
+    @UserMessage(
+        """
+                  You are a professional meeting invitation writer. Draft a concise and clear meeting invitation email based on the following details:
+                  Subject: {{subject}}
+                  Date: {{date}}
+                  Time: {{time}}
+                  Location: {{location}}
+                  Agenda: {{agenda}}
+                """)
+    @Agent("Drafts a professional meeting invitation email.")
+    String invoke(@V("subject") String subject, @V("date") String date, @V("time") String time, @V("location") String location, @V("agenda") String agenda);
+  }
+
+  interface MeetingInvitationStyle extends AgentInstance {
+
+    @UserMessage(
+        """
+                  You are a professional meeting invitation writer. Rewrite the following meeting invitation email to better fit the {{style}} style:
+                  Original Invitation: {{invitation}}
+                """)
+    @Agent("Edits a meeting invitation email to better fit a given style.")
+    String invoke(@V("invitation") String invitation, @V("style") String style);
+  }
+
+ }

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

@@ -23,13 +23,16 @@
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
+import dev.langchain4j.agentic.AgentServices;
+import dev.langchain4j.agentic.workflow.HumanInTheLoop;
 import io.serverlessworkflow.api.types.Workflow;
 import io.serverlessworkflow.impl.WorkflowApplication;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
 import java.util.concurrent.ExecutionException;
+import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Function;
 import java.util.stream.Collectors;
 import org.junit.jupiter.api.Test;
@@ -220,4 +223,53 @@ public void testSeqAndThenParallel() throws ExecutionException, InterruptedExcep
       assertEquals(technologyTraits, result.get("branch-1-cultureAndTechnology"));
     }
   }
+
+  @Test
+  public void humanInTheLoop() throws ExecutionException, InterruptedException {
+    final MeetingInvitationDraft meetingInvitationDraft = mock(MeetingInvitationDraft.class);
+    when(meetingInvitationDraft.invoke(eq("Meeting with John Doe"),
+            eq("2023-10-01"), eq("08:00AM"),
+            eq("London"),
+            eq("Discuss project updates")))
+        .thenReturn("Drafted meeting invitation for John Doe");
+    when(meetingInvitationDraft.outputName()).thenReturn("draft");
+
+
+    final MeetingInvitationStyle meetingInvitationStyle = mock(MeetingInvitationStyle.class);
+    when(meetingInvitationStyle.invoke(eq("Drafted meeting invitation for John Doe"), eq("formal")))
+        .thenReturn("Styled meeting invitation for John Doe");
+    when(meetingInvitationStyle.outputName()).thenReturn("styled");
+
+    AtomicReference<String> request = new AtomicReference<>();
+
+    HumanInTheLoop humanInTheLoop = AgentServices.humanInTheLoopBuilder()
+            .description("What level of formality would you like? (please reply with “formal”, “casual”, or “friendly”)")
+            .inputName("style")
+            .outputName("style")
+            .requestWriter(q -> request.set("What level of formality would you like? (please reply with “formal”, “casual”, or “friendly”)"))
+            .responseReader(() -> "formal")
+            .build();
+
+    Workflow workflow =
+        AgentWorkflowBuilder.workflow("meetingInvitationFlow")
+            .tasks(
+                d ->
+                    d.sequence("draft", meetingInvitationDraft, humanInTheLoop, meetingInvitationStyle)
+            ).build();
+    Map<String, String> initialValues = new HashMap<>();
+    initialValues.put("title", "Meeting with John Doe");
+    initialValues.put("date", "2023-10-01");
+    initialValues.put("time", "08:00AM");
+    initialValues.put("location", "London");
+    initialValues.put("agenda", "Discuss project updates");
+
+    try (WorkflowApplication app = WorkflowApplication.builder().build()) {
+      String result =
+              app.workflowDefinition(workflow).instance(initialValues).start().get().asText().orElseThrow();
+
+        assertEquals("Styled meeting invitation for John Doe", result);
+    }
+
+
+  }
 }