diff --git a/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/ChatPromptTemplate.java b/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/ChatPromptTemplate.java index ef6d08ce5e1..4db4aee5746 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/ChatPromptTemplate.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/ChatPromptTemplate.java @@ -75,10 +75,22 @@ public Prompt create() { return new Prompt(messages); } + @Override + public Prompt create(ChatOptions modelOptions) { + List messages = createMessages(); + return new Prompt(messages, modelOptions); + } + @Override public Prompt create(Map model) { List messages = createMessages(model); return new Prompt(messages); } + @Override + public Prompt create(Map model, ChatOptions modelOptions) { + List messages = createMessages(model); + return new Prompt(messages, modelOptions); + } + } diff --git a/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplate.java b/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplate.java index 1fe901350b8..162b0d59ffa 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplate.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplate.java @@ -172,11 +172,21 @@ public Prompt create() { return new Prompt(render(new HashMap<>())); } + @Override + public Prompt create(ChatOptions modelOptions) { + return new Prompt(render(new HashMap<>()), modelOptions); + } + @Override public Prompt create(Map model) { return new Prompt(render(model)); } + @Override + public Prompt create(Map model, ChatOptions modelOptions) { + return new Prompt(render(model), modelOptions); + } + public Set getInputVariables() { TokenStream tokens = this.st.impl.tokens; Set inputVariables = new HashSet<>(); diff --git a/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplateActions.java b/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplateActions.java index 7b54aa05e03..872d35f4c63 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplateActions.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/chat/prompt/PromptTemplateActions.java @@ -21,6 +21,10 @@ public interface PromptTemplateActions extends PromptTemplateStringActions { Prompt create(); + Prompt create(ChatOptions modelOptions); + Prompt create(Map model); + Prompt create(Map model, ChatOptions modelOptions); + }