Improved handling of WorkflowModel and Cognisphere

Signed-off-by: Dmitrii Tikhomirov <chani.liet@gmail.com>

Author: treblereel

fluent/agentic/src/main/java/io/serverlessworkflow/fluent/agentic/AgentAdapters.java

@@ -15,26 +15,45 @@
  */
 package io.serverlessworkflow.fluent.agentic;
 
-import static dev.langchain4j.agentic.internal.AgentExecutor.agentsToExecutors;
+import static dev.langchain4j.agentic.internal.AgentUtil.agentsToExecutors;
 
-import dev.langchain4j.agentic.Cognisphere;
+import dev.langchain4j.agentic.cognisphere.Cognisphere;
 import dev.langchain4j.agentic.internal.AgentExecutor;
 import dev.langchain4j.agentic.internal.AgentInstance;
 import io.serverlessworkflow.impl.expressions.LoopPredicateIndex;
 import java.util.List;
+import java.util.Map;
 import java.util.function.Function;
 import java.util.function.Predicate;
 import java.util.stream.Stream;
 
 public final class AgentAdapters {
+
+  private static final Cognisphere cognisphere =
+      Cognisphere.registry().createEphemeralCognisphere();
+
   private AgentAdapters() {}
 
   public static List<AgentExecutor> toExecutors(Object... agents) {
-    return agentsToExecutors(Stream.of(agents).map(AgentInstance.class::cast).toList());
+    return agentsToExecutors(Stream.of(agents).map(AgentInstance.class::cast).toArray());
   }
 
-  public static Function<Cognisphere, Object> toFunction(AgentExecutor exec) {
-    return exec::invoke;
+  public static Function<Object, Object> toFunction(AgentExecutor exec) {
+    return o -> {
+      if (!(o instanceof Map)) {
+        throw new IllegalArgumentException(
+            "Expected input to be a Map, but got: " + o.getClass().getCanonicalName());
+      }
+
+      Map<String, Object> input = (Map<String, Object>) o;
+      String outputName = exec.agentSpecification().outputName();
+      cognisphere.writeStates(input);
+
+      Object result = exec.invoke(cognisphere);
+      input.put(outputName, result);
+
+      return input;
+    };
   }
 
   public static LoopPredicateIndex<Object, Object> toWhile(Predicate<Cognisphere> exit) {

fluent/agentic/src/main/java/io/serverlessworkflow/fluent/agentic/LoopAgentsBuilder.java

@@ -15,7 +15,7 @@
  */
 package io.serverlessworkflow.fluent.agentic;
 
-import dev.langchain4j.agentic.Cognisphere;
+import dev.langchain4j.agentic.cognisphere.Cognisphere;
 import dev.langchain4j.agentic.internal.AgentExecutor;
 import io.serverlessworkflow.api.types.ForTaskConfiguration;
 import io.serverlessworkflow.api.types.func.ForTaskFunction;

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

@@ -24,7 +24,7 @@
 import static org.mockito.Mockito.spy;
 
 import dev.langchain4j.agentic.AgentServices;
-import dev.langchain4j.agentic.Cognisphere;
+import dev.langchain4j.agentic.cognisphere.Cognisphere;
 import io.serverlessworkflow.api.types.ForkTask;
 import io.serverlessworkflow.api.types.Task;
 import io.serverlessworkflow.api.types.TaskItem;