Use record and use default record toString

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;
-    }
-
 }