diff options
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/llm/Completion.java | 10 |
1 files changed, 1 insertions, 9 deletions
diff --git a/model-integration/src/main/java/ai/vespa/llm/Completion.java b/model-integration/src/main/java/ai/vespa/llm/Completion.java index 1b03ca04cb2..5f483a65186 100644 --- a/model-integration/src/main/java/ai/vespa/llm/Completion.java +++ b/model-integration/src/main/java/ai/vespa/llm/Completion.java @@ -11,7 +11,7 @@ import java.util.Objects; * @author bratseth */ @Beta -public class Completion { +public record Completion(String text, FinishReason finishReason) { public enum FinishReason { @@ -23,9 +23,6 @@ public class Completion { } - private final String text; - private final FinishReason finishReason; - public Completion(String text, FinishReason finishReason) { this.text = Objects.requireNonNull(text); this.finishReason = Objects.requireNonNull(finishReason); @@ -41,9 +38,4 @@ public class Completion { return new Completion(text, FinishReason.stop); } - @Override - public String toString() { - return text; - } - } |