Fix outputName; Minor TODOs;

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

Author: ricardozanini

fluent/agentic-langchain4j/src/main/java/io/serverlessworkflow/fluent/agentic/langchain4j/AbstractAgentService.java

@@ -23,7 +23,9 @@
 
 public abstract class AbstractAgentService<T, S> implements WorkflowDefinitionBuilder {
 
+  // Workflow OutputAs
   private static final Function<Cognisphere, Object> DEFAULT_OUTPUT_FUNCTION = cognisphere -> null;
+  // Workflow InputFrom
   private static final Consumer<Cognisphere> DEFAULT_INIT_FUNCTION = cognisphere -> {};
 
   protected final AgentWorkflowBuilder workflowBuilder;
@@ -52,8 +54,8 @@ public S beforeCall(Consumer<Cognisphere> beforeCall) {
 
   @SuppressWarnings("unchecked")
   public S outputName(String outputName) {
-    // TODO: This can be solved using a combination with our output object
-    this.workflowBuilder.document(d -> d.name(outputName));
+    Function<Cognisphere, Object> outputFunction = cog -> cog.readState(outputName);
+    this.workflowBuilder.outputAs(outputFunction);
     return (S) this;
   }
 

fluent/agentic-langchain4j/src/main/java/io/serverlessworkflow/fluent/agentic/langchain4j/ParallelAgentServiceImpl.java

@@ -23,6 +23,8 @@
 public class ParallelAgentServiceImpl<T> extends AbstractAgentService<T, ParallelAgentService<T>>
     implements ParallelAgentService<T> {
 
+  private ExecutorService executor;
+
   private ParallelAgentServiceImpl(Class<T> agentServiceClass) {
     super(agentServiceClass);
   }
@@ -33,8 +35,8 @@ public static <T> ParallelAgentService<T> builder(Class<T> agentServiceClass) {
 
   @Override
   public ParallelAgentService<T> executorService(ExecutorService executorService) {
-    // TODO: add to the internal WorkflowApplication the executorService to run agents in parallel
-    return null;
+    this.executor = executorService;
+    return this;
   }
 
   @Override

fluent/agentic-langchain4j/src/test/java/io/serverlessworkflow/fluent/agentic/langchain4j/SequentialAgentServiceImplTest.java

@@ -23,6 +23,7 @@
 import dev.langchain4j.agentic.cognisphere.Cognisphere;
 import io.serverlessworkflow.api.types.Workflow;
 import io.serverlessworkflow.api.types.func.CallJava;
+import io.serverlessworkflow.api.types.func.OutputAsFunction;
 import io.serverlessworkflow.fluent.agentic.AgentsUtils;
 import java.util.function.Function;
 import org.junit.jupiter.api.Test;
@@ -70,8 +71,8 @@ void shouldSetOutputName_inDocumentMetadata() {
 
     // then
     assertNotNull(wf.getDocument().getName(), "Workflow name should not be null");
-    assertEquals(
-        outputName, wf.getDocument().getName(), "Workflow name should match the outputName");
+    assertNotNull(wf.getOutput().getAs(), "Workflow outputAs should not be null");
+    assertInstanceOf(OutputAsFunction.class, wf.getOutput().getAs());
   }
 
   @Test