Skip to content

Commit c1d3034

Browse files
FormatTodaymarkpollack
authored andcommitted
add AssistantPromptTemplate override method createMessage()
Signed-off-by: ZhangQin <dev@zhangqin56.cn>
1 parent a786784 commit c1d3034

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/AssistantPromptTemplate.java

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,9 @@
1616

1717
package org.springframework.ai.chat.prompt;
1818

19+
import org.springframework.ai.chat.messages.Message;
1920
import org.springframework.ai.chat.messages.AssistantMessage;
21+
import org.springframework.core.io.Resource;
2022

2123
import java.util.Map;
2224

@@ -26,6 +28,10 @@ public AssistantPromptTemplate(String template) {
2628
super(template);
2729
}
2830

31+
public AssistantPromptTemplate(Resource resource) {
32+
super(resource);
33+
}
34+
2935
@Override
3036
public Prompt create() {
3137
return new Prompt(new AssistantMessage(render()));
@@ -36,4 +42,14 @@ public Prompt create(Map<String, Object> model) {
3642
return new Prompt(new AssistantMessage(render(model)));
3743
}
3844

45+
@Override
46+
public Message createMessage() {
47+
return new AssistantMessage(render());
48+
}
49+
50+
@Override
51+
public Message createMessage(Map<String, Object> model) {
52+
return new AssistantMessage(render(model));
53+
}
54+
3955
}

0 commit comments

Comments
 (0)