Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add AssistantPromptTemplate override method createMessage()

Browse Source 
Signed-off-by: ZhangQin <dev@zhangqin56.cn>
This commit is contained in:
FormatToday
2024-01-10 11:14:04 +08:00
committed by Mark Pollack
parent a786784982
commit c1d3034569
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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<String, Object> model) {
return new AssistantMessage(render(model));
}
}