33 lines
956 B
Java
33 lines
956 B
Java
package com.pablotj.ia.chat.boot;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
public class PromptBuilder {
|
|
|
|
private final String systemPrompt;
|
|
private final List<String> turns = new ArrayList<>();
|
|
|
|
public PromptBuilder(String systemPrompt) {
|
|
this.systemPrompt = systemPrompt;
|
|
}
|
|
|
|
public void user(String message) {
|
|
turns.add("GPT4 Correct User: " + message + "<|end_of_turn|>");
|
|
}
|
|
|
|
public void assistant(String message) {
|
|
turns.add("GPT4 Correct Assistant: " + message + "<|end_of_turn|>");
|
|
}
|
|
|
|
public String build() {
|
|
StringBuilder sb = new StringBuilder();
|
|
sb.append(systemPrompt).append("<|end_of_turn|>\n");
|
|
for (String turn : turns) {
|
|
sb.append(turn).append("\n");
|
|
}
|
|
// Deja el último turno preparado para que el modelo continúe como "Assistant"
|
|
sb.append("GPT4 Correct Assistant: ");
|
|
return sb.toString();
|
|
}
|
|
} |