diff options
Diffstat (limited to 'vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java')
-rw-r--r-- | vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java b/vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java new file mode 100644 index 00000000000..e8392ca992e --- /dev/null +++ b/vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java @@ -0,0 +1,43 @@ +package ai.vespa.llm.completion; + +import com.yahoo.api.annotations.Beta; + +import java.util.Objects; + +/** + * A prompt which just consists of a string. + * + * @author bratseth + */ +@Beta +public class StringPrompt extends Prompt { + + private final String string; + + private StringPrompt(String string) { + this.string = Objects.requireNonNull(string); + } + + @Override + public String asString() { return string; } + + @Override + public StringPrompt append(String text) { + return StringPrompt.from(string + text); + } + + @Override + public StringPrompt append(Completion completion) { + return append(completion.text()); + } + + @Override + public String toString() { + return string; + } + + public static StringPrompt from(String string) { + return new StringPrompt(string); + } + +} |