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

remove unused fields in Message classes

Browse Source
This commit is contained in:
Mark Pollack
2023-08-10 15:58:11 -04:00
parent 8189459e5b
commit 0f268194cc
3 changed files with 4 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
spring-ai-core/src/main/java/org/springframework/ai/core/prompt/AssistantPromptTemplate.java
View File

@@ -22,17 +22,10 @@ import java.util.Map;
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 class AssistantPromptTemplate extends PromptTemplate {
@Override
public Prompt create(Map<String, Object> model) {
return new Prompt(new AssistantMessage(render(model), this.example));
return new Prompt(new AssistantMessage(render(model)));
}
}

14
spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/AssistantMessage.java
View File

@@ -26,24 +26,12 @@ import java.util.Map;
*/
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;
}
}

12
spring-ai-core/src/main/java/org/springframework/ai/core/prompt/messages/FunctionMessage.java
View File

@@ -20,20 +20,12 @@ import java.util.Map;
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;
}
}