use langchain4g agentic api for agents

Author: treblereel

experimental/ai/impl/pom.xml

@@ -19,10 +19,6 @@
             <groupId>io.serverlessworkflow</groupId>
             <artifactId>serverlessworkflow-impl-core</artifactId>
         </dependency>
-        <dependency>
-            <groupId>io.serverlessworkflow</groupId>
-            <artifactId>serverlessworkflow-experimental-types</artifactId>
-        </dependency>
         <dependency>
             <groupId>io.serverlessworkflow</groupId>
             <artifactId>serverlessworkflow-experimental-ai-types</artifactId>

experimental/ai/impl/src/main/java/io/serverlessworkflow/impl/executors/ai/AIChatModelCallExecutor.java

@@ -16,10 +16,8 @@
 
 package io.serverlessworkflow.impl.executors.ai;
 
-import io.serverlessworkflow.ai.api.types.CallAILangChainChatModel;
+import io.serverlessworkflow.ai.api.types.CallAgentAI;
 import io.serverlessworkflow.api.types.TaskBase;
-import io.serverlessworkflow.api.types.ai.AbstractCallAIChatModelTask;
-import io.serverlessworkflow.api.types.ai.CallAIChatModel;
 import io.serverlessworkflow.impl.TaskContext;
 import io.serverlessworkflow.impl.WorkflowApplication;
 import io.serverlessworkflow.impl.WorkflowContext;
@@ -29,33 +27,26 @@
 import io.serverlessworkflow.impl.resources.ResourceLoader;
 import java.util.concurrent.CompletableFuture;
 
-public class AIChatModelCallExecutor implements CallableTask<AbstractCallAIChatModelTask> {
+public class AIChatModelCallExecutor implements CallableTask<CallAgentAI> {
 
   @Override
-  public void init(
-      AbstractCallAIChatModelTask task, WorkflowApplication application, ResourceLoader loader) {}
+  public void init(CallAgentAI task, WorkflowApplication application, ResourceLoader loader) {}
 
   @Override
   public CompletableFuture<WorkflowModel> apply(
       WorkflowContext workflowContext, TaskContext taskContext, WorkflowModel input) {
     WorkflowModelFactory modelFactory = workflowContext.definition().application().modelFactory();
-    if (taskContext.task() instanceof CallAILangChainChatModel callAILangChainChatModel) {
+    if (taskContext.task() instanceof CallAgentAI agenticAI) {
       return CompletableFuture.completedFuture(
-          modelFactory.fromAny(
-              new CallAILangChainChatModelExecutor()
-                  .apply(callAILangChainChatModel, input.asJavaObject())));
-    } else if (taskContext.task() instanceof CallAIChatModel callAIChatModel) {
-      return CompletableFuture.completedFuture(
-          modelFactory.fromAny(
-              new CallAIChatModelExecutor().apply(callAIChatModel, input.asJavaObject())));
+          modelFactory.fromAny(new CallAgentAIExecutor().execute(agenticAI, input.asJavaObject())));
     }
     throw new IllegalArgumentException(
-        "AIChatModelCallExecutor can only process CallAIChatModel tasks, but received: "
+        "AIChatModelCallExecutor can only process CallAgentAI tasks, but received: "
             + taskContext.task().getClass().getName());
   }
 
   @Override
   public boolean accept(Class<? extends TaskBase> clazz) {
-    return AbstractCallAIChatModelTask.class.isAssignableFrom(clazz);
+    return CallAgentAI.class.isAssignableFrom(clazz);
   }
 }

experimental/ai/impl/src/main/java/io/serverlessworkflow/impl/executors/ai/AbstractCallAIChatModelExecutor.java

@@ -0,0 +1,41 @@
+/*
+ * Copyright 2020-Present The Serverless Workflow Specification Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.serverlessworkflow.impl.executors.ai;
+
+import dev.langchain4j.agentic.Cognisphere;
+import dev.langchain4j.agentic.internal.AgentExecutor;
+import dev.langchain4j.agentic.internal.AgentUtil;
+import io.serverlessworkflow.ai.api.types.CallAgentAI;
+import java.util.Map;
+
+public class CallAgentAIExecutor {
+
+  private static final Cognisphere cognisphere = new Cognisphere();
+
+  public Object execute(CallAgentAI callAgentAI, Object input) {
+    AgentExecutor agentExecutor = AgentUtil.agentToExecutor(callAgentAI.getAgentInstance());
+
+    Map<String, Object> output = (Map<String, Object>) input;
+
+    cognisphere.writeStates(output);
+
+    Object result = agentExecutor.invoke(cognisphere);
+
+    output.put(callAgentAI.getAgentInstance().outputName(), result);
+    return output;
+  }
+}