remove unused fields in Message classes

spring-projects · Aug 10, 2023 · 0f26819 · 0f26819
1 parent 8189459
commit 0f26819
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 31 deletions.
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
@@ -22,25 +22,18 @@
 
 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()));
 	}
 
 	@Override
 	public Prompt create(Map<String, Object> model) {
-		return new Prompt(new AssistantMessage(render(model), this.example));
+		return new Prompt(new AssistantMessage(render(model)));
 	}
 
 }
diff --git a/...g-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/AssistantMessage.java b/...g-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<String, Object> properties) {
+	public AssistantMessage(String content, Map<String, Object> properties) {
 		super(MessageType.ASSISTANT, content, properties);
-		this.example = example;
-	}
-
-	public boolean isExample() {
-		return example;
 	}
 
 }
diff --git a/...ng-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/FunctionMessage.java b/...ng-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<String, Object> properties) {
+	public FunctionMessage(String content, Map<String, Object> properties) {
 		super(MessageType.SYSTEM, content, properties);
-		this.functionName = functionName;
-	}
-
-	public String getFunctionName() {
-		return functionName;
 	}
 
 }