Skip to content

Commit 3a54e52

Browse files
Fix
1 parent 48cce05 commit 3a54e52

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
import com.sap.ai.sdk.prompt.registry.model.PromptTemplateSpecResponseFormat;
1111
import com.sap.ai.sdk.prompt.registry.model.ResponseFormatText;
1212
import com.sap.ai.sdk.prompt.registry.model.SingleChatTemplate;
13-
import com.sap.ai.sdk.prompt.registry.model.Template;
13+
import com.sap.ai.sdk.prompt.registry.model.PromptTemplate;
1414
import com.sap.cloud.sdk.cloudplatform.connectivity.ApacheHttpClient5Accessor;
1515
import com.sap.cloud.sdk.services.openapi.apiclient.ApiClient;
1616
import javax.annotation.Nonnull;
@@ -59,7 +59,7 @@ private static ApiClient addMixin(@Nonnull final AiCoreService service) {
5959
converter ->
6060
converter.setObjectMapper(
6161
getDefaultObjectMapper()
62-
.addMixIn(Template.class, JacksonMixin.TemplateMixIn.class)
62+
.addMixIn(PromptTemplate.class, JacksonMixin.TemplateMixIn.class)
6363
.addMixIn(
6464
PromptTemplateSpecResponseFormat.class,
6565
JacksonMixin.ResponseFormat.class)));

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import com.sap.ai.sdk.prompt.registry.model.PromptTemplateSubstitutionResponse;
44
import com.sap.ai.sdk.prompt.registry.model.SingleChatTemplate;
5-
import com.sap.ai.sdk.prompt.registry.model.Template;
5+
import com.sap.ai.sdk.prompt.registry.model.PromptTemplate;
66
import java.util.List;
77
import javax.annotation.Nonnull;
88
import lombok.val;
@@ -33,7 +33,7 @@ public static List<Message> promptTemplateToMessages(
3333
// TRANSFORM TEMPLATE TO SPRING AI MESSAGES
3434
return res.stream()
3535
.map(
36-
(Template t) -> {
36+
(PromptTemplate t) -> {
3737
final SingleChatTemplate message = (SingleChatTemplate) t;
3838
return (Message)
3939
switch (message.getRole()) {

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import com.sap.ai.sdk.prompt.registry.model.PromptTemplatePostResponse;
88
import com.sap.ai.sdk.prompt.registry.model.PromptTemplateSubstitutionResponse;
99
import com.sap.ai.sdk.prompt.registry.model.SingleChatTemplate;
10-
import com.sap.ai.sdk.prompt.registry.model.Template;
10+
import com.sap.ai.sdk.prompt.registry.model.PromptTemplate;
1111
import java.io.IOException;
1212
import java.util.List;
1313
import org.junit.jupiter.api.Test;
@@ -34,7 +34,7 @@ void createDeleteTemplate() {
3434

3535
// use template
3636
PromptTemplateSubstitutionResponse template = controller.useTemplate();
37-
List<Template> prompt = template.getParsedPrompt();
37+
List<PromptTemplate> prompt = template.getParsedPrompt();
3838
assertThat(prompt).hasSize(2);
3939
SingleChatTemplate userMessage = (SingleChatTemplate) prompt.get(1);
4040
assertThat(userMessage.getRole()).isEqualTo("user");

0 commit comments

Comments
 (0)