From c1d30345698cad439a4789e3255e22cedc07c062 Mon Sep 17 00:00:00 2001 From: FormatToday Date: Wed, 10 Jan 2024 11:14:04 +0800 Subject: [PATCH] add AssistantPromptTemplate override method createMessage() Signed-off-by: ZhangQin --- .../ai/chat/prompt/AssistantPromptTemplate.java | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) 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)); + } + }