exo/lm-lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Combined unnecessary dual prompt template parameters instruction and text to user

Browse Source
This commit is contained in:
extio-de
2025-06-27 17:17:27 +02:00
parent 4e52c1c4ce
commit 5b5f56fa15
12 changed files with 34 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/main/java/de/extio/lmlib/prompt/VicunaPromptStrategy.java
View File

@@ -6,16 +6,15 @@ import org.springframework.stereotype.Component;
public class VicunaPromptStrategy implements PromptStrategy {
@Override
public StringBuilder start(final String system, final String question, final String text) {
public StringBuilder start(final String system, final String user) {
final StringBuilder prompt = new StringBuilder();
prompt.append(system);
prompt.append(" USER: ");
prompt.append(question);
if (!question.isEmpty() && !text.isEmpty()) {
if (system.isEmpty()) {
prompt.append(system);
prompt.append("\n");
}
prompt.append(text);
prompt.append(" ASSISTANT:");
prompt.append("USER: ");
prompt.append(user);
prompt.append("\nASSISTANT:");
return prompt;
}
@@ -30,7 +29,7 @@ public class VicunaPromptStrategy implements PromptStrategy {
prompt.append(assistant);
prompt.append("</s>USER: ");
prompt.append(user);
prompt.append(" ASSISTANT:");
prompt.append("\nASSISTANT:");
}
@Override