PromptTemplates can create messages

spring-projects · Aug 18, 2023 · 236393e · 236393e
1 parent f3840e3
commit 236393e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/spring-ai-core/src/main/java/org/springframework/ai/prompt/SystemPromptTemplate.java b/spring-ai-core/src/main/java/org/springframework/ai/prompt/SystemPromptTemplate.java
@@ -16,16 +16,30 @@
 
 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 {
 
 	public SystemPromptTemplate(String template) {
 		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()));