diff --git a/spring-ai-core/src/main/java/org/springframework/ai/prompt/AssistantPromptTemplate.java b/spring-ai-core/src/main/java/org/springframework/ai/prompt/AssistantPromptTemplate.java index 715763f47fe..bce567f9b61 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/prompt/AssistantPromptTemplate.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/prompt/AssistantPromptTemplate.java @@ -17,6 +17,7 @@ package org.springframework.ai.prompt; import org.springframework.ai.prompt.messages.AssistantMessage; +import org.springframework.ai.prompt.messages.Message; import java.util.Map; @@ -36,4 +37,14 @@ public Prompt create(Map model) { 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)); + } + }