diff --git a/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/AssistantPromptTemplate.java b/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/AssistantPromptTemplate.java index d30e3302d8..78927fc322 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/AssistantPromptTemplate.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/AssistantPromptTemplate.java @@ -22,17 +22,10 @@ public class AssistantPromptTemplate extends PromptTemplate { - private boolean example = false; - public AssistantPromptTemplate(String template) { super(template); } - public AssistantPromptTemplate(String template, boolean example) { - super(template); - this.example = example; - } - @Override public Prompt create() { return new Prompt(new AssistantMessage(render())); @@ -40,7 +33,7 @@ public Prompt create() { @Override public Prompt create(Map model) { - return new Prompt(new AssistantMessage(render(model), this.example)); + return new Prompt(new AssistantMessage(render(model))); } } diff --git a/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/AssistantMessage.java b/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/AssistantMessage.java index f1ced57174..537967ea6e 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/AssistantMessage.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/AssistantMessage.java @@ -26,24 +26,12 @@ */ public class AssistantMessage extends AbstractMessage { - private boolean example = false; - public AssistantMessage(String content) { super(MessageType.ASSISTANT, content); } - public AssistantMessage(String content, boolean example) { - super(MessageType.ASSISTANT, content); - this.example = example; - } - - public AssistantMessage(String content, boolean example, Map properties) { + public AssistantMessage(String content, Map properties) { super(MessageType.ASSISTANT, content, properties); - this.example = example; - } - - public boolean isExample() { - return example; } } diff --git a/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/FunctionMessage.java b/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/FunctionMessage.java index 2c82813e8d..6b427c2ddd 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/FunctionMessage.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/FunctionMessage.java @@ -20,20 +20,12 @@ public class FunctionMessage extends AbstractMessage { - private String functionName; - - public FunctionMessage(String content, String functionName) { + public FunctionMessage(String content) { super(MessageType.SYSTEM, content); - this.functionName = functionName; } - public FunctionMessage(String content, String functionName, Map properties) { + public FunctionMessage(String content, Map properties) { super(MessageType.SYSTEM, content, properties); - this.functionName = functionName; - } - - public String getFunctionName() { - return functionName; } }