diff --git a/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/AssistantPromptTemplate.java b/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/AssistantPromptTemplate.java index 8a167f005..19da5511c 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/AssistantPromptTemplate.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/AssistantPromptTemplate.java @@ -16,7 +16,9 @@ package org.springframework.ai.chat.prompt; +import org.springframework.ai.chat.messages.Message; import org.springframework.ai.chat.messages.AssistantMessage; +import org.springframework.core.io.Resource; import java.util.Map; @@ -26,6 +28,10 @@ public class AssistantPromptTemplate extends PromptTemplate { super(template); } + public AssistantPromptTemplate(Resource resource) { + super(resource); + } + @Override public Prompt create() { return new Prompt(new AssistantMessage(render())); @@ -36,4 +42,14 @@ public class AssistantPromptTemplate extends PromptTemplate { return new Prompt(new AssistantMessage(render(model))); } + @Override + public Message createMessage() { + return new AssistantMessage(render()); + } + + @Override + public Message createMessage(Map model) { + return new AssistantMessage(render(model)); + } + }