Introduce Agentic DSL

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

Author: ricardozanini

experimental/lambda/src/test/java/io/serverless/workflow/impl/FluentDSLCallTest.java

@@ -34,7 +34,7 @@ void testJavaFunction() throws InterruptedException, ExecutionException {
     try (WorkflowApplication app = WorkflowApplication.builder().build()) {
       final Workflow workflow =
           FuncWorkflowBuilder.workflow("testJavaCall")
-              .tasks(tasks -> tasks.callFn(f -> f.fn(JavaFunctions::getName)))
+              .tasks(tasks -> tasks.callFn(f -> f.function(JavaFunctions::getName)))
               .build();
       assertThat(
               app.workflowDefinition(workflow)
@@ -85,7 +85,7 @@ void testSwitch() throws InterruptedException, ExecutionException {
                                   switchOdd.items(
                                       item ->
                                           item.when(CallTest::isOdd).then(FlowDirectiveEnum.END)))
-                          .callFn(callJava -> callJava.fn(CallTest::zero)))
+                          .callFn(callJava -> callJava.function(CallTest::zero)))
               .build();
 
       WorkflowDefinition definition = app.workflowDefinition(workflow);

fluent/agentic/pom.xml

@@ -0,0 +1,69 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+    <parent>
+        <groupId>io.serverlessworkflow</groupId>
+        <artifactId>serverlessworkflow-fluent</artifactId>
+        <version>8.0.0-SNAPSHOT</version>
+    </parent>
+
+    <name>Serverless Workflow :: Fluent :: Agentic</name>
+    <artifactId>serverlessworkflow-fluent-agentic</artifactId>
+
+    <properties>
+        <maven.compiler.source>17</maven.compiler.source>
+        <maven.compiler.target>17</maven.compiler.target>
+        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+
+        <version.dev.langchain4j>1.2.0-beta8-SNAPSHOT</version.dev.langchain4j>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>io.serverlessworkflow</groupId>
+            <artifactId>serverlessworkflow-experimental-types</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>io.serverlessworkflow</groupId>
+            <artifactId>serverlessworkflow-types</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>io.serverlessworkflow</groupId>
+            <artifactId>serverlessworkflow-fluent-spec</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>io.serverlessworkflow</groupId>
+            <artifactId>serverlessworkflow-fluent-func</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j-agentic</artifactId>
+            <version>${version.dev.langchain4j}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.slf4j</groupId>
+            <artifactId>slf4j-simple</artifactId>
+            <scope>test</scope>
+        </dependency>
+
+        <dependency>
+            <groupId>org.junit.jupiter</groupId>
+            <artifactId>junit-jupiter-api</artifactId>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.mockito</groupId>
+            <artifactId>mockito-core</artifactId>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j-ollama</artifactId>
+            <scope>test</scope>
+            <version>1.2.0-SNAPSHOT</version>
+        </dependency>
+    </dependencies>
+
+</project>

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

@@ -0,0 +1,37 @@
+/*
+ * 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.fluent.agentic;
+
+import static dev.langchain4j.agentic.internal.AgentExecutor.agentsToExecutors;
+
+import dev.langchain4j.agentic.Cognisphere;
+import dev.langchain4j.agentic.internal.AgentExecutor;
+import dev.langchain4j.agentic.internal.AgentInstance;
+import java.util.List;
+import java.util.function.Function;
+import java.util.stream.Stream;
+
+public final class AgentAdapters {
+  private AgentAdapters() {}
+
+  public static List<AgentExecutor> toExecutors(Object... agents) {
+    return agentsToExecutors(Stream.of(agents).map(AgentInstance.class::cast).toList());
+  }
+
+  public static Function<Cognisphere, Object> toFn(AgentExecutor exec) {
+    return exec::invoke;
+  }
+}

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

@@ -0,0 +1,46 @@
+/*
+ * 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.fluent.agentic;
+
+import io.serverlessworkflow.fluent.func.FuncDoTaskBuilder;
+import io.serverlessworkflow.fluent.func.FuncTaskItemListBuilder;
+
+public class AgentDoTaskBuilder extends FuncDoTaskBuilder
+    implements DelegatingAgentDoTaskFluent<AgentDoTaskBuilder> {
+
+  private AgentDoTaskBuilder(AgentTaskItemListBuilder agentTaskItemListBuilder) {
+    super(agentTaskItemListBuilder);
+  }
+
+  static AgentDoTaskBuilder wrap(FuncDoTaskBuilder base) {
+    FuncTaskItemListBuilder funcList = base.internalDelegate();
+    AgentTaskItemListBuilder agentList =
+        (funcList instanceof AgentTaskItemListBuilder al)
+            ? al
+            : new AgentTaskItemListBuilder(funcList.getInternalList());
+    return new AgentDoTaskBuilder(agentList);
+  }
+
+  @Override
+  public AgentDoTaskFluent<?> agentInternalDelegate() {
+    return (AgentDoTaskFluent<?>) super.internalDelegate();
+  }
+
+  @Override
+  public AgentDoTaskBuilder self() {
+    return this;
+  }
+}

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

@@ -0,0 +1,21 @@
+/*
+ * 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.fluent.agentic;
+
+public interface AgentDoTaskFluent<SELF extends AgentDoTaskFluent<SELF>> {
+
+  SELF agent(String name, Object agent);
+}

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

@@ -0,0 +1,36 @@
+/*
+ * 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.fluent.agentic;
+
+import dev.langchain4j.agentic.internal.AgentExecutor;
+import io.serverlessworkflow.api.types.TaskItem;
+import io.serverlessworkflow.fluent.func.FuncTaskItemListBuilder;
+import java.util.List;
+
+public class AgentTaskItemListBuilder extends FuncTaskItemListBuilder
+    implements AgentDoTaskFluent<AgentTaskItemListBuilder> {
+
+  AgentTaskItemListBuilder(final List<TaskItem> list) {
+    super(list);
+  }
+
+  @Override
+  public AgentTaskItemListBuilder agent(String name, Object agent) {
+    AgentExecutor exec = AgentAdapters.toExecutors(agent).get(0);
+    this.callFn(name, fn -> fn.function(AgentAdapters.toFn(exec)));
+    return this;
+  }
+}

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

@@ -0,0 +1,78 @@
+/*
+ * 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.fluent.agentic;
+
+import io.serverlessworkflow.api.types.Workflow;
+import io.serverlessworkflow.fluent.func.FuncWorkflowBuilder;
+import io.serverlessworkflow.fluent.spec.DocumentBuilder;
+import io.serverlessworkflow.fluent.spec.InputBuilder;
+import io.serverlessworkflow.fluent.spec.OutputBuilder;
+import io.serverlessworkflow.fluent.spec.UseBuilder;
+import java.util.function.Consumer;
+
+public final class AgenticWorkflowBuilder {
+
+  private final FuncWorkflowBuilder delegate;
+
+  AgenticWorkflowBuilder(final FuncWorkflowBuilder delegate) {
+    this.delegate = delegate;
+  }
+
+  public static AgenticWorkflowBuilder workflow() {
+    return new AgenticWorkflowBuilder(FuncWorkflowBuilder.workflow());
+  }
+
+  public static AgenticWorkflowBuilder workflow(String name) {
+    return new AgenticWorkflowBuilder(FuncWorkflowBuilder.workflow(name));
+  }
+
+  public static AgenticWorkflowBuilder workflow(String name, String ns) {
+    return new AgenticWorkflowBuilder(FuncWorkflowBuilder.workflow(name, ns));
+  }
+
+  public AgenticWorkflowBuilder document(Consumer<DocumentBuilder> c) {
+    delegate.document(c);
+    return this;
+  }
+
+  public AgenticWorkflowBuilder input(Consumer<InputBuilder> c) {
+    delegate.input(c);
+    return this;
+  }
+
+  public AgenticWorkflowBuilder output(Consumer<OutputBuilder> c) {
+    delegate.output(c);
+    return this;
+  }
+
+  public AgenticWorkflowBuilder use(Consumer<UseBuilder> c) {
+    delegate.use(c);
+    return this;
+  }
+
+  public AgenticWorkflowBuilder tasks(Consumer<AgentDoTaskBuilder> c) {
+    delegate.tasks(
+        funcDo -> {
+          AgentDoTaskBuilder agentDoTaskBuilder = AgentDoTaskBuilder.wrap(funcDo);
+          c.accept(agentDoTaskBuilder);
+        });
+    return this;
+  }
+
+  public Workflow build() {
+    return delegate.build();
+  }
+}

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

@@ -0,0 +1,33 @@
+/*
+ * 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.fluent.agentic;
+
+public interface DelegatingAgentDoTaskFluent<SELF extends DelegatingAgentDoTaskFluent<SELF>>
+    extends AgentDoTaskFluent<SELF> {
+
+  /** Return the underlying functional ops delegate. */
+  AgentDoTaskFluent<?> agentInternalDelegate();
+
+  @SuppressWarnings("unchecked")
+  default SELF self() {
+    return (SELF) this;
+  }
+
+  default SELF agent(String name, Object agent) {
+    agentInternalDelegate().agent(name, agent);
+    return (SELF) this;
+  }
+}

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

@@ -0,0 +1,47 @@
+/*
+ * 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.fluent.agentic;
+
+import static io.serverlessworkflow.fluent.agentic.Models.BASE_MODEL;
+import static org.junit.jupiter.api.Assertions.assertInstanceOf;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.mockito.Mockito.spy;
+
+import dev.langchain4j.agentic.AgentServices;
+import io.serverlessworkflow.api.types.Workflow;
+import io.serverlessworkflow.api.types.func.CallJava;
+import org.junit.jupiter.api.Test;
+
+public class AgenticWorkflowBuilderTest {
+
+  @Test
+  public void verifyAgentCall() {
+    Agents.MovieExpert movieExpert =
+        spy(
+            AgentServices.agentBuilder(Agents.MovieExpert.class)
+                .outputName("movies")
+                .chatModel(BASE_MODEL)
+                .build());
+
+    Workflow workflow =
+        AgenticWorkflowBuilder.workflow()
+            .tasks(tasks -> tasks.agent("myAgent", movieExpert))
+            .build();
+
+    assertNotNull(workflow);
+    assertInstanceOf(CallJava.class, workflow.getDo().get(0).getTask().getCallTask().get());
+  }
+}