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

Extend the PromptTemplate to support Message media content (#492)

Browse Source
This commit is contained in:
Christian Tzolov
2024-03-21 22:25:15 +01:00
committed by GitHub
parent 834d2d0487
commit 5a44f854e8
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplate.java
View File

@@ -18,6 +18,7 @@ package org.springframework.ai.chat.prompt;
import org.antlr.runtime.Token;
import org.antlr.runtime.TokenStream;
import org.springframework.ai.parser.OutputParser;
import org.springframework.ai.chat.messages.Media;
import org.springframework.ai.chat.messages.Message;
import org.springframework.ai.chat.messages.UserMessage;
import org.springframework.core.io.Resource;
@@ -173,6 +174,11 @@ public class PromptTemplate implements PromptTemplateActions, PromptTemplateMess
return new UserMessage(render());
}
@Override
public Message createMessage(List<Media> mediaList) {
return new UserMessage(render(), mediaList);
}
@Override
public Message createMessage(Map<String, Object> model) {
return new UserMessage(render(model));

4
spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplateMessageActions.java
View File

@@ -15,14 +15,18 @@
*/
package org.springframework.ai.chat.prompt;
import org.springframework.ai.chat.messages.Media;
import org.springframework.ai.chat.messages.Message;
import java.util.List;
import java.util.Map;
public interface PromptTemplateMessageActions {
Message createMessage();
Message createMessage(List<Media> mediaList);
Message createMessage(Map<String, Object> model);
}