ai.vespa.llm.completion.StringPrompt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vespajlib Show documentation

Library for use in Java components of Vespa. Shared code which do not fit anywhere else.

There is a newer version: 8.441.21

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
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);
    }

}