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

PromptTemplates can create messages

Browse Source
This commit is contained in:
Mark Pollack
2023-08-18 01:23:15 -04:00
parent f3840e3651
commit 236393ef5d
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
spring-ai-core/src/main/java/org/springframework/ai/prompt/SystemPromptTemplate.java
View File

@@ -16,9 +16,15 @@
package org.springframework.ai.prompt;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.Map;
import org.springframework.ai.prompt.messages.SystemMessage;
import org.springframework.core.io.Resource;
import org.springframework.util.StreamUtils;
import org.stringtemplate.v4.ST;
public class SystemPromptTemplate extends PromptTemplate {
@@ -26,6 +32,14 @@ public class SystemPromptTemplate extends PromptTemplate {
super(template);
}
public SystemPromptTemplate(Resource resource) {
super(resource);
}
public SystemMessage createMessage(Map<String, Object> model) {
return new SystemMessage(render(model));
}
@Override
public Prompt create() {
return new Prompt(new SystemMessage(render()));