diff options
Diffstat (limited to 'model-integration/src/main/java/ai/vespa/llm/StringPrompt.java')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/llm/StringPrompt.java | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/model-integration/src/main/java/ai/vespa/llm/StringPrompt.java b/model-integration/src/main/java/ai/vespa/llm/StringPrompt.java new file mode 100644 index 00000000000..0af8388dfb1 --- /dev/null +++ b/model-integration/src/main/java/ai/vespa/llm/StringPrompt.java @@ -0,0 +1,43 @@ +package ai.vespa.llm; + +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); + } + +} |