Initial version. No unit test

Author: rpanackal

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

@@ -2,13 +2,16 @@
 
 import static com.sap.ai.sdk.core.JacksonConfiguration.getDefaultObjectMapper;
 
+import com.fasterxml.jackson.annotation.JsonSubTypes;
 import com.fasterxml.jackson.annotation.JsonTypeInfo;
 import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
 import com.google.common.collect.Iterables;
 import com.sap.ai.sdk.core.AiCoreService;
 import com.sap.ai.sdk.prompt.registry.client.PromptTemplatesApi;
 import com.sap.ai.sdk.prompt.registry.model.PromptTemplate;
 import com.sap.ai.sdk.prompt.registry.model.PromptTemplateSpecResponseFormat;
+import com.sap.ai.sdk.prompt.registry.model.ResponseFormatJsonObject;
+import com.sap.ai.sdk.prompt.registry.model.ResponseFormatJsonSchema;
 import com.sap.ai.sdk.prompt.registry.model.ResponseFormatText;
 import com.sap.ai.sdk.prompt.registry.model.SingleChatTemplate;
 import com.sap.cloud.sdk.cloudplatform.connectivity.ApacheHttpClient5Accessor;
@@ -75,8 +78,16 @@ private static class JacksonMixin {
     @JsonDeserialize(as = SingleChatTemplate.class)
     interface TemplateMixIn {}
 
-    @JsonTypeInfo(use = JsonTypeInfo.Id.NONE)
-    @JsonDeserialize(as = ResponseFormatText.class)
+    @JsonTypeInfo(
+        use = JsonTypeInfo.Id.NAME,
+        include = JsonTypeInfo.As.EXISTING_PROPERTY,
+        property = "type",
+        visible = true)
+    @JsonSubTypes({
+      @JsonSubTypes.Type(value = ResponseFormatJsonSchema.class, name = "json_schema"),
+      @JsonSubTypes.Type(value = ResponseFormatJsonObject.class, name = "json_object"),
+      @JsonSubTypes.Type(value = ResponseFormatText.class, name = "text")
+    })
     interface ResponseFormat {}
   }
 }

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

@@ -36,4 +36,6 @@ void testPipelines() {
     assertThat(template.isIsVersionHead()).isEqualTo(true);
     assertThat(template.getSpec()).isNull();
   }
+  
+  
 }

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

@@ -52,7 +52,7 @@ void testPromptRegistryToSpringAi() {
 
   @Test
   void testInvalidRoleThrowsException() {
-    var client = new PromptClient(SERVICE);
+    final var client = new PromptClient(SERVICE);
     val errorPrompt =
         client.parsePromptTemplateByNameVersion(
             "categorization",

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

@@ -2,11 +2,16 @@
 
 import static org.assertj.core.api.Assertions.assertThat;
 
+import com.sap.ai.sdk.prompt.registry.PromptClient;
 import com.sap.ai.sdk.prompt.registry.model.PromptTemplate;
 import com.sap.ai.sdk.prompt.registry.model.PromptTemplateDeleteResponse;
+import com.sap.ai.sdk.prompt.registry.model.PromptTemplateGetResponse;
 import com.sap.ai.sdk.prompt.registry.model.PromptTemplateListResponse;
+import com.sap.ai.sdk.prompt.registry.model.PromptTemplatePostRequest;
 import com.sap.ai.sdk.prompt.registry.model.PromptTemplatePostResponse;
+import com.sap.ai.sdk.prompt.registry.model.PromptTemplateSpec;
 import com.sap.ai.sdk.prompt.registry.model.PromptTemplateSubstitutionResponse;
+import com.sap.ai.sdk.prompt.registry.model.ResponseFormatJsonObject;
 import com.sap.ai.sdk.prompt.registry.model.SingleChatTemplate;
 import java.io.IOException;
 import java.util.List;
@@ -93,4 +98,34 @@ void promptRegistryToSpringAi() {
     assertThat(ChatResponse).isNotNull();
     assertThat(ChatResponse.getOutput().getText()).contains("Sports");
   }
+
+  @Test
+  void testPromptCreateAndGet() {
+    final PromptClient client = new PromptClient();
+    // cleanup
+    final var responseFormat =
+        ResponseFormatJsonObject.create().type(ResponseFormatJsonObject.TypeEnum.JSON_OBJECT);
+    final var template = SingleChatTemplate.create().role("system").content("Test content");
+    final PromptTemplateSpec spec =
+        PromptTemplateSpec.create().template(List.of(template)).responseFormat(responseFormat);
+    final PromptTemplatePostRequest promptTemplateRequest =
+        PromptTemplatePostRequest.create()
+            .name("test")
+            .version("0.0.1")
+            .scenario("test-retrival")
+            .spec(spec);
+
+    final PromptTemplatePostResponse postResponse =
+        client.createUpdatePromptTemplate(promptTemplateRequest);
+    assertThat(postResponse.getName()).isEqualTo("test");
+
+    final PromptTemplateGetResponse getResponse =
+        client.getPromptTemplateByUuid(postResponse.getId());
+    // This assertion fails
+    assertThat(getResponse.getSpec().getResponseFormat())
+        .isEqualTo(
+            ResponseFormatJsonObject.create().type(ResponseFormatJsonObject.TypeEnum.JSON_OBJECT));
+
+    client.deletePromptTemplate(postResponse.getId());
+  }
 }