Spring AI samples

Author: donald-pinckney

core/build.gradle

@@ -7,6 +7,9 @@ dependencies {
     // Environment configuration
     implementation "io.temporal:temporal-envconfig:$javaSDKVersion"
 
+    // Needed for SSL sample (AdvancedTlsX509KeyManager)
+    implementation "io.grpc:grpc-util"
+
     // Needed for SDK related functionality
     implementation(platform("com.fasterxml.jackson:jackson-bom:2.17.2"))
     implementation "com.fasterxml.jackson.core:jackson-databind"

core/src/main/java/io/temporal/samples/ssl/Starter.java

@@ -50,12 +50,14 @@ public static void main(String[] args) throws Exception {
     if (refreshPeriod > 0) {
       AdvancedTlsX509KeyManager clientKeyManager = new AdvancedTlsX509KeyManager();
       // Reload credentials every minute
-      clientKeyManager.updateIdentityCredentialsFromFile(
-          clientKeyFile,
-          clientCertFile,
-          refreshPeriod,
-          TimeUnit.MINUTES,
-          Executors.newScheduledThreadPool(1));
+      @SuppressWarnings("InlineMeInliner")
+      var unused =
+          clientKeyManager.updateIdentityCredentialsFromFile(
+              clientKeyFile,
+              clientCertFile,
+              refreshPeriod,
+              TimeUnit.MINUTES,
+              Executors.newScheduledThreadPool(1));
       sslContext =
           GrpcSslContexts.configure(SslContextBuilder.forClient().keyManager(clientKeyManager))
               .build();

gradle/springai.gradle

@@ -0,0 +1,52 @@
+// Shared configuration for all Spring AI sample modules.
+// Applied via: apply from: "$rootDir/gradle/springai.gradle"
+
+apply plugin: 'org.springframework.boot'
+apply plugin: 'io.spring.dependency-management'
+
+ext {
+    springBootVersionForSpringAi = '3.5.3'
+    springAiVersion = '1.1.0'
+}
+
+java {
+    sourceCompatibility = JavaVersion.VERSION_17
+    targetCompatibility = JavaVersion.VERSION_17
+}
+
+dependencyManagement {
+    imports {
+        mavenBom "org.springframework.boot:spring-boot-dependencies:$springBootVersionForSpringAi"
+        mavenBom "org.springframework.ai:spring-ai-bom:$springAiVersion"
+    }
+}
+
+dependencies {
+    // Temporal
+    implementation "io.temporal:temporal-spring-boot-starter:$javaSDKVersion"
+    implementation "io.temporal:temporal-spring-ai:$javaSDKVersion"
+
+    // Spring Boot
+    implementation 'org.springframework.boot:spring-boot-starter'
+
+    // Required at runtime: the SpringAiPlugin introspects for VectorStore/MCP support
+    runtimeOnly 'org.springframework.ai:spring-ai-rag'
+    runtimeOnly 'org.springframework.ai:spring-ai-mcp'
+
+    dependencies {
+        errorproneJavac('com.google.errorprone:javac:9+181-r4173-1')
+        errorprone('com.google.errorprone:error_prone_core:2.28.0')
+    }
+}
+
+bootJar {
+    enabled = false
+}
+
+jar {
+    enabled = true
+}
+
+bootRun {
+    standardInput = System.in
+}

settings.gradle

@@ -1,4 +1,24 @@
 rootProject.name = 'temporal-java-samples'
 include 'core'
+include 'springai'
+include 'springai-mcp'
+include 'springai-multimodel'
+include 'springai-rag'
+include 'springai-sandboxing'
 include 'springboot'
 include 'springboot-basic'
+
+// Include local sdk-java build for temporal-spring-ai (until published to Maven Central).
+// temporal-spring-ai requires the plugin API (SimplePlugin) which is not yet in a released SDK,
+// so we substitute all SDK modules from the local build.
+includeBuild('../sdk-java') {
+    dependencySubstitution {
+        substitute module('io.temporal:temporal-spring-ai') using project(':temporal-spring-ai')
+        substitute module('io.temporal:temporal-sdk') using project(':temporal-sdk')
+        substitute module('io.temporal:temporal-serviceclient') using project(':temporal-serviceclient')
+        substitute module('io.temporal:temporal-spring-boot-autoconfigure') using project(':temporal-spring-boot-autoconfigure')
+        substitute module('io.temporal:temporal-spring-boot-starter') using project(':temporal-spring-boot-starter')
+        substitute module('io.temporal:temporal-testing') using project(':temporal-testing')
+        substitute module('io.temporal:temporal-opentracing') using project(':temporal-opentracing')
+    }
+}

springai-mcp/build.gradle

@@ -0,0 +1,8 @@
+apply from: "$rootDir/gradle/springai.gradle"
+
+dependencies {
+    implementation 'org.springframework.ai:spring-ai-starter-model-openai'
+    implementation 'org.springframework.ai:spring-ai-starter-mcp-client'
+    implementation 'org.springframework.ai:spring-ai-rag'
+    implementation 'org.springframework.boot:spring-boot-starter-webflux'
+}

springai-mcp/build/classes/java/main/io/temporal/samples/springai/mcp/McpApplication.class

@@ -0,0 +1,136 @@
+package io.temporal.samples.springai.mcp;
+
+import io.temporal.client.WorkflowClient;
+import io.temporal.client.WorkflowOptions;
+import java.util.Scanner;
+import java.util.UUID;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.context.event.ApplicationReadyEvent;
+import org.springframework.context.event.EventListener;
+
+/**
+ * Example application demonstrating MCP (Model Context Protocol) integration.
+ *
+ * <p>This application shows how to use tools from MCP servers within Temporal workflows. It
+ * connects to a filesystem MCP server and provides an AI assistant that can read and write files.
+ *
+ * <h2>Usage</h2>
+ *
+ * <pre>
+ * Commands:
+ *   tools                - List available MCP tools
+ *   &lt;any message&gt;       - Chat with the AI (it can use file tools)
+ *   quit                 - End the chat
+ * </pre>
+ *
+ * <h2>Example Interactions</h2>
+ *
+ * <pre>
+ * &gt; List files in the current directory
+ * [AI uses list_directory tool and returns results]
+ *
+ * &gt; Create a file called hello.txt with "Hello from MCP!"
+ * [AI uses write_file tool]
+ *
+ * &gt; Read the contents of hello.txt
+ * [AI uses read_file tool]
+ * </pre>
+ *
+ * <h2>Prerequisites</h2>
+ *
+ * <ol>
+ *   <li>Start a Temporal dev server: {@code temporal server start-dev}
+ *   <li>Set OPENAI_API_KEY environment variable
+ *   <li>Ensure Node.js/npx is available (for MCP server)
+ *   <li>Optionally set MCP_ALLOWED_PATH (defaults to /tmp/mcp-example)
+ *   <li>Run: {@code ./gradlew :example-mcp:bootRun}
+ * </ol>
+ */
+@SpringBootApplication
+public class McpApplication {
+
+  private static final String TASK_QUEUE = "mcp-example-queue";
+
+  @Autowired private WorkflowClient workflowClient;
+
+  public static void main(String[] args) {
+    SpringApplication.run(McpApplication.class, args);
+  }
+
+  /** Runs after workers are started (ApplicationReadyEvent fires after CommandLineRunner). */
+  @EventListener(ApplicationReadyEvent.class)
+  public void onReady() throws Exception {
+    // Start a new workflow
+    String workflowId = "mcp-example-" + UUID.randomUUID().toString().substring(0, 8);
+    McpWorkflow workflow =
+        workflowClient.newWorkflowStub(
+            McpWorkflow.class,
+            WorkflowOptions.newBuilder()
+                .setTaskQueue(TASK_QUEUE)
+                .setWorkflowId(workflowId)
+                .build());
+
+    // Start the workflow asynchronously
+    WorkflowClient.start(workflow::run);
+
+    // Give the workflow time to initialize (first workflow task must complete)
+    Thread.sleep(1000);
+
+    System.out.println("\n=== MCP Tools Demo ===");
+    System.out.println("Workflow ID: " + workflowId);
+    System.out.println("\nThis demo uses the filesystem MCP server.");
+    System.out.println("The AI can read, write, and list files in the allowed directory.");
+    System.out.println("\nCommands:");
+    System.out.println("  tools    - List available MCP tools");
+    System.out.println("  <text>   - Chat with the AI");
+    System.out.println("  quit     - End the chat");
+    System.out.println();
+
+    // Get a workflow stub for sending signals/queries
+    McpWorkflow workflowStub = workflowClient.newWorkflowStub(McpWorkflow.class, workflowId);
+
+    // Note: tools command may take a moment to work while workflow initializes
+    System.out.println("Type 'tools' to list available MCP tools.\n");
+
+    Scanner scanner = new Scanner(System.in, java.nio.charset.StandardCharsets.UTF_8);
+    while (true) {
+      System.out.print("> ");
+      String input = scanner.nextLine().trim();
+
+      if (input.equalsIgnoreCase("quit")) {
+        workflowStub.end();
+        System.out.println("Chat ended. Goodbye!");
+        break;
+      }
+
+      if (input.equalsIgnoreCase("tools")) {
+        System.out.println(workflowStub.listTools());
+        continue;
+      }
+
+      if (input.isEmpty()) {
+        continue;
+      }
+
+      System.out.println("[Processing...]");
+
+      // Capture current response BEFORE sending, so we can detect when it changes
+      String previousResponse = workflowStub.getLastResponse();
+
+      // Send the message via signal
+      workflowStub.chat(input);
+
+      // Poll until the response changes (workflow has processed our message)
+      for (int i = 0; i < 600; i++) { // Wait up to 60 seconds (MCP tools can be slow)
+        String response = workflowStub.getLastResponse();
+        if (!response.equals(previousResponse)) {
+          System.out.println("\n[AI]: " + response + "\n");
+          break;
+        }
+        Thread.sleep(100);
+      }
+    }
+  }
+}