aboutsummaryrefslogtreecommitdiffstats
path: root/vespajlib/src/main/java/ai/vespa/llm/completion
diff options
context:
space:
mode:
Diffstat (limited to 'vespajlib/src/main/java/ai/vespa/llm/completion')
-rw-r--r--vespajlib/src/main/java/ai/vespa/llm/completion/Completion.java6
-rw-r--r--vespajlib/src/main/java/ai/vespa/llm/completion/Prompt.java2
-rw-r--r--vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java1
-rw-r--r--vespajlib/src/main/java/ai/vespa/llm/completion/package-info.java4
4 files changed, 8 insertions, 5 deletions
diff --git a/vespajlib/src/main/java/ai/vespa/llm/completion/Completion.java b/vespajlib/src/main/java/ai/vespa/llm/completion/Completion.java
index 30645b5151f..ea784013812 100644
--- a/vespajlib/src/main/java/ai/vespa/llm/completion/Completion.java
+++ b/vespajlib/src/main/java/ai/vespa/llm/completion/Completion.java
@@ -1,4 +1,4 @@
-// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// 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;
@@ -19,8 +19,10 @@ public record Completion(String text, FinishReason finishReason) {
length,
/** The completion is the predicted ending of the prompt. */
- stop
+ stop,
+ /** The completion is not finished yet, more tokens are incoming. */
+ none
}
public Completion(String text, FinishReason finishReason) {
diff --git a/vespajlib/src/main/java/ai/vespa/llm/completion/Prompt.java b/vespajlib/src/main/java/ai/vespa/llm/completion/Prompt.java
index d5d0247d6b0..44dfb8499a8 100644
--- a/vespajlib/src/main/java/ai/vespa/llm/completion/Prompt.java
+++ b/vespajlib/src/main/java/ai/vespa/llm/completion/Prompt.java
@@ -1,4 +1,4 @@
-// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// 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;
diff --git a/vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java b/vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java
index e8392ca992e..9e702c79a7a 100644
--- a/vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java
+++ b/vespajlib/src/main/java/ai/vespa/llm/completion/StringPrompt.java
@@ -1,3 +1,4 @@
+// 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;
diff --git a/vespajlib/src/main/java/ai/vespa/llm/completion/package-info.java b/vespajlib/src/main/java/ai/vespa/llm/completion/package-info.java
index 79898c694ca..57c2b3f3364 100644
--- a/vespajlib/src/main/java/ai/vespa/llm/completion/package-info.java
+++ b/vespajlib/src/main/java/ai/vespa/llm/completion/package-info.java
@@ -1,4 +1,4 @@
-// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
@ExportPackage
@PublicApi
package ai.vespa.llm.completion;
@@ -8,4 +8,4 @@ import com.yahoo.osgi.annotation.ExportPackage;
/**
* Classes for generating text completions with language models.
- */ \ No newline at end of file
+ */