Implementation to be able to use Prompt Registry for SpringAi

Author: n-o-u-r-h-a-n

orchestration/pom.xml

@@ -154,6 +154,10 @@
       <artifactId>javaparser-core</artifactId>
       <scope>test</scope>
     </dependency>
+      <dependency>
+          <groupId>com.sap.ai.sdk</groupId>
+          <artifactId>prompt-registry</artifactId>
+      </dependency>
   </dependencies>
 
   <profiles>

orchestration/src/main/java/com/sap/ai/sdk/orchestration/spring/OrchestrationSpringUtil.java

@@ -0,0 +1,54 @@
+package com.sap.ai.sdk.orchestration.spring;
+
+import com.sap.ai.sdk.prompt.registry.PromptClient;
+import com.sap.ai.sdk.prompt.registry.model.PromptTemplateSubstitutionRequest;
+import com.sap.ai.sdk.prompt.registry.model.SingleChatTemplate;
+import com.sap.ai.sdk.prompt.registry.model.Template;
+import org.springframework.ai.chat.messages.AssistantMessage;
+import org.springframework.ai.chat.messages.Message;
+import org.springframework.ai.chat.messages.SystemMessage;
+import org.springframework.ai.chat.messages.UserMessage;
+import org.springframework.ai.chat.prompt.Prompt;
+
+import java.util.Map;
+import java.util.UUID;
+
+public class OrchestrationSpringUtil {
+  private OrchestrationSpringUtil() {
+    // Utility class, no instantiation allowed
+  }
+
+  public static Prompt getPromptTemplate(String templateName, Map<String, Object> inputParams) {
+    var templateMessages =
+        new PromptClient()
+            .parsePromptTemplateByNameVersion(
+                "categorization",
+                "1.0.0",
+                "PROMPT-CHART",
+                "default",
+                false,
+                PromptTemplateSubstitutionRequest.create()
+                    .inputParams(Map.of("current_timestamp", System.currentTimeMillis())))
+            .getParsedPrompt();
+
+    // TRANSFORM TEMPLATE TO SPRING AI MESSAGES
+    var messages =
+        templateMessages.stream()
+            .map(
+                (Template t) -> {
+                  SingleChatTemplate message = (SingleChatTemplate) t;
+                  return (Message)
+                      switch (message.getRole()) {
+                        case "system" -> new SystemMessage(message.getContent());
+                        case "user" -> new UserMessage(message.getContent());
+                        case "assistant" -> new AssistantMessage(message.getContent());
+                        default ->
+                            throw new IllegalArgumentException(
+                                "Unknown role: " + message.getRole());
+                      };
+                })
+            .toList();
+
+    return new Prompt(messages);
+  }
+}

sample-code/spring-app/src/main/java/com/sap/ai/sdk/app/services/SpringAiOrchestrationService.java

@@ -19,6 +19,8 @@
 import java.util.Objects;
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
+
+import com.sap.ai.sdk.orchestration.spring.OrchestrationSpringUtil;
 import lombok.val;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
@@ -220,4 +222,19 @@ public Translation responseFormat() {
 
     return cl.prompt(prompt).call().entity(Translation.class);
   }
+
+  @Nullable
+  public ChatResponse getPromptTemplate() {
+    ChatModel client = new OrchestrationChatModel();
+    var memory = new InMemoryChatMemory();
+    var advisor = new MessageChatMemoryAdvisor(memory);
+    var cl = ChatClient.builder(client).defaultAdvisors(advisor).build();
+
+    val prompt = OrchestrationSpringUtil.getPromptTemplate(
+        "PROMPT-CHART",
+        Map.of("current_timestamp", System.currentTimeMillis()));
+
+    var response = cl.prompt(prompt).call();
+    return response.chatResponse();
+  }
 }

sample-code/spring-app/src/test/java/com/sap/ai/sdk/app/controllers/SpringAiOrchestrationTest.java

@@ -166,4 +166,14 @@ void testResponseFormat() {
     assertThat(translation.translation()).isNotEmpty();
     assertThat(translation.language()).containsIgnoringCase("dutch");
   }
+
+  @Test
+  void getPromptTemplate() {
+    var ChatResponse = service.getPromptTemplate();
+    assertThat(ChatResponse).isNotNull();
+    assertThat(ChatResponse.getResult().getOutput().getText()).isNotEmpty();
+    assertThat(ChatResponse.getResult().getOutput().getText())
+        .contains("The current timestamp is");
+  }
+
 }