Doing Unit Test and moving method to Prompt Registry.

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

core-services/prompt-registry/pom.xml

@@ -64,6 +64,11 @@
       <groupId>org.springframework</groupId>
       <artifactId>spring-web</artifactId>
     </dependency>
+    <dependency>
+      <groupId>org.springframework.ai</groupId>
+      <artifactId>spring-ai-model</artifactId>
+      <optional>true</optional>
+    </dependency>
     <dependency>
       <groupId>com.sap.cloud.sdk.cloudplatform</groupId>
       <artifactId>cloudplatform-connectivity</artifactId>

core-services/prompt-registry/src/main/java/com/sap/ai/sdk/prompt/registry/model/PromptTemplateSubstitutionResponse.java

@@ -40,7 +40,7 @@ public class PromptTemplateSubstitutionResponse
   private final Map<String, Object> cloudSdkCustomFields = new LinkedHashMap<>();
 
   /** Default constructor for PromptTemplateSubstitutionResponse. */
-  protected PromptTemplateSubstitutionResponse() {}
+  public PromptTemplateSubstitutionResponse() {}
 
   /**
    * Set the parsedPrompt of this {@link PromptTemplateSubstitutionResponse} instance and return the

core-services/prompt-registry/src/test/java/com/sap/ai/sdk/prompt/registry/spring/SpringUtilTest.java

@@ -0,0 +1,51 @@
+package com.sap.ai.sdk.prompt.registry.spring;
+
+import com.github.tomakehurst.wiremock.junit5.WireMockExtension;
+import com.sap.ai.sdk.core.AiCoreService;
+import com.sap.ai.sdk.prompt.registry.PromptClient;
+import com.sap.ai.sdk.prompt.registry.model.PromptTemplateSubstitutionRequest;
+import com.sap.cloud.sdk.cloudplatform.connectivity.DefaultHttpDestination;
+import com.sap.cloud.sdk.cloudplatform.connectivity.HttpDestination;
+import lombok.val;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.RegisterExtension;
+import org.springframework.ai.chat.messages.Message;
+import org.springframework.ai.chat.messages.SystemMessage;
+import org.springframework.ai.chat.messages.UserMessage;
+
+import java.util.List;
+import java.util.Map;
+
+import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.wireMockConfig;
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class SpringUtilTest {
+  @RegisterExtension
+  private static final WireMockExtension WM =
+      WireMockExtension.newInstance().options(wireMockConfig().dynamicPort()).build();
+
+  private final HttpDestination DESTINATION = DefaultHttpDestination.builder(WM.baseUrl()).build();
+  private final AiCoreService SERVICE = new AiCoreService().withBaseDestination(DESTINATION);
+
+  @Test
+  void testPipelines() {
+    var client = new PromptClient(SERVICE);
+    val promptResponse =
+        client.parsePromptTemplateByNameVersion(
+            "categorization",
+            "0.0.1",
+            "java-e2e-test",
+            "default",
+            false,
+            PromptTemplateSubstitutionRequest.create()
+                .inputParams(Map.of("inputExample", "I love football")));
+
+    List<Message> messages = SpringUtil.promptRegistryToSpringAi(promptResponse);
+    assertThat(messages)
+        .isEqualTo(
+            List.of(
+                new SystemMessage(
+                    "You classify input text into the two following categories: Finance, Tech, Sports, Politics"),
+                new UserMessage("I love football")));
+  }
+}

core-services/prompt-registry/src/test/resources/mappings/templatesInputParams.json

@@ -0,0 +1,26 @@
+{
+  "request": {
+    "method": "POST",
+    "url": "/v2/lm/scenarios/categorization/promptTemplates/java-e2e-test/versions/0.0.1/substitution?metadata=false"
+  },
+  "response": {
+    "status": 200,
+    "headers": {
+      "Content-Type": "application/json"
+    },
+    "jsonBody": {
+      "parsedPrompt": [
+        {
+          "role": "system",
+          "content": "You classify input text into the two following categories: Finance, Tech, Sports, Politics"
+        },
+        {
+          "role": "user",
+          "content": "I love football"
+        }
+      ]
+    }
+  }
+}
+
+

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

@@ -1,5 +1,8 @@
 package com.sap.ai.sdk.app.controllers;
 
+import com.sap.ai.sdk.foundationmodels.openai.OpenAiClient;
+import com.sap.ai.sdk.foundationmodels.openai.OpenAiModel;
+import com.sap.ai.sdk.foundationmodels.openai.spring.OpenAiChatModel;
 import com.sap.ai.sdk.prompt.registry.PromptClient;
 import com.sap.ai.sdk.prompt.registry.model.PromptTemplateDeleteResponse;
 import com.sap.ai.sdk.prompt.registry.model.PromptTemplateListResponse;
@@ -13,12 +16,27 @@
 import java.io.IOException;
 import java.util.List;
 import java.util.Map;
+
+import com.sap.ai.sdk.prompt.registry.spring.SpringUtil;
+import lombok.val;
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
+import org.springframework.ai.chat.memory.InMemoryChatMemoryRepository;
+import org.springframework.ai.chat.memory.MessageWindowChatMemory;
+import org.springframework.ai.chat.messages.Message;
+import org.springframework.ai.chat.model.ChatModel;
+import org.springframework.ai.chat.model.ChatResponse;
+import org.springframework.ai.chat.model.Generation;
+import org.springframework.ai.chat.prompt.Prompt;
 import org.springframework.core.io.ClassPathResource;
 import org.springframework.core.io.Resource;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
+import javax.annotation.Nullable;
+
 /** Endpoint for Prompt Registry operations */
 @SuppressWarnings("unused") // debug class that doesn't need to be tested
 @RestController
@@ -99,4 +117,29 @@ List<PromptTemplateDeleteResponse> deleteTemplate() {
         .map(template -> client.deletePromptTemplate(template.getId()))
         .toList();
   }
+
+  @GetMapping("/promptRegistryToSpringAi")
+  Generation promptRegistryToSpringAi() {
+    ChatModel openAiClient = new OpenAiChatModel(OpenAiClient.forModel(OpenAiModel.GPT_4O_MINI));
+    val repository = new InMemoryChatMemoryRepository();
+    val memory = MessageWindowChatMemory.builder().chatMemoryRepository(repository).build();
+    val advisor = MessageChatMemoryAdvisor.builder(memory).build();
+    val cl = ChatClient.builder(openAiClient).defaultAdvisors(advisor).build();
+
+    val promptResponse =
+        new PromptClient()
+            .parsePromptTemplateByNameVersion(
+                "categorization",
+                "0.0.1",
+                "java-e2e-test",
+                "default",
+                false,
+                PromptTemplateSubstitutionRequest.create()
+                    .inputParams(Map.of("inputExample", "I love football")));
+
+    List<Message> messages = SpringUtil.promptRegistryToSpringAi(promptResponse);
+    val prompt = new Prompt(messages);
+    val response = cl.prompt(prompt).call().chatResponse();
+    return response != null ? response.getResult() : null;
+  }
 }

sample-code/spring-app/src/main/java/com/sap/ai/sdk/app/controllers/PromptRegistryController.java

@@ -275,32 +275,4 @@ public Translation responseFormat() {
 
     return cl.prompt(prompt).call().entity(Translation.class);
   }
-
-  /**
-   * Get a prompt template by name and input parameters.
-   *
-   * @return the chat response containing the prompt template
-   */
-  @Nullable
-  public ChatResponse promptRegistryToSpringAi() {
-    val repository = new InMemoryChatMemoryRepository();
-    val memory = MessageWindowChatMemory.builder().chatMemoryRepository(repository).build();
-    val advisor = MessageChatMemoryAdvisor.builder(memory).build();
-    val cl = ChatClient.builder(client).defaultAdvisors(advisor).build();
-
-    val promptResponse =
-        new PromptClient()
-            .parsePromptTemplateByNameVersion(
-                "categorization",
-                "0.0.1",
-                "java-e2e-test",
-                "default",
-                false,
-                PromptTemplateSubstitutionRequest.create()
-                    .inputParams(Map.of("inputExample", "I love football")));
-
-    List<Message> messages = SpringUtil.promptRegistryToSpringAi(promptResponse);
-    val prompt = new Prompt(messages, defaultOptions);
-    return cl.prompt(prompt).call().chatResponse();
-  }
 }

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

@@ -1051,6 +1051,18 @@ <h2>📚 Prompt Registry</h2>
                                     </div>
                                 </div>
                             </li>
+                            <li class="list-group-item">
+                                <div class="info-tooltip">
+                                    <button type="submit"
+                                            formaction="/prompt-registry/promptRegistryToSpringAi"
+                                            class="link-offset-2-hover link-underline link-underline-opacity-0 link-underline-opacity-75-hover endpoint">
+                                        <code>/prompt-registry/promptRegistryToSpringAi</code>
+                                    </button>
+                                    <div class="tooltip-content">
+                                        Get a SpringAI list of messages from a Prompt Registry Response.
+                                    </div>
+                                </div>
+                            </li>
                         </ul>
                     </div>
                 </div>

sample-code/spring-app/src/main/resources/static/index.html

@@ -85,4 +85,12 @@ void history() {
     assertThat(deletedTemplate).hasSize(1);
     assertThat(deletedTemplate.get(0).getMessage()).contains("successful");
   }
+
+  @Test
+  void promptRegistryToSpringAi() {
+    var controller = new PromptRegistryController();
+    var ChatResponse = controller.promptRegistryToSpringAi();
+    assertThat(ChatResponse).isNotNull();
+    assertThat(ChatResponse.getOutput().getText()).contains("Sports");
+  }
 }

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

@@ -166,11 +166,4 @@ void testResponseFormat() {
     assertThat(translation.translation()).isNotEmpty();
     assertThat(translation.language()).containsIgnoringCase("dutch");
   }
-
-  @Test
-  void promptRegistryToSpringAi() {
-    var ChatResponse = service.promptRegistryToSpringAi();
-    assertThat(ChatResponse).isNotNull();
-    assertThat(ChatResponse.getResult().getOutput().getText()).contains("Sports");
-  }
 }