From 398bcd2199167a1fcfb45e12109acef82e10f5d2 Mon Sep 17 00:00:00 2001
From: FormatToday <dev@zhangqin56.cn>
Date: Wed, 10 Jan 2024 11:14:04 +0800
Subject: [PATCH] add AssistantPromptTemplate override method createMessage()

Signed-off-by: ZhangQin <dev@zhangqin56.cn>
---
 .../ai/prompt/AssistantPromptTemplate.java            | 11 +++++++++++
 1 file changed, 11 insertions(+)

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<String, Object> model) {
 		return new Prompt(new AssistantMessage(render(model)));
 	}
 
+	@Override
+	public Message createMessage() {
+		return new AssistantMessage(render());
+	}
+
+	@Override
+	public Message createMessage(Map<String, Object> model) {
+		return new AssistantMessage(render(model));
+	}
+
 }