diff options
author | Lester Solbakken <lesters@oath.com> | 2023-10-16 16:56:40 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2023-10-16 16:56:40 +0200 |
commit | 0bab8eefe3443cc6f7befc13607b3c23602998b4 (patch) | |
tree | 89898e0d5b62930c2ed2b85c5c131df045faefe8 /vespajlib/src/main/java/ai/vespa/llm/test/MockLanguageModel.java | |
parent | 0ccfe8aab8c12ecd518f882a048f8a13fb2084f1 (diff) |
Add OpenAI async client
Diffstat (limited to 'vespajlib/src/main/java/ai/vespa/llm/test/MockLanguageModel.java')
-rw-r--r-- | vespajlib/src/main/java/ai/vespa/llm/test/MockLanguageModel.java | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/ai/vespa/llm/test/MockLanguageModel.java b/vespajlib/src/main/java/ai/vespa/llm/test/MockLanguageModel.java index 16e9c4e1848..db1b42fbbac 100644 --- a/vespajlib/src/main/java/ai/vespa/llm/test/MockLanguageModel.java +++ b/vespajlib/src/main/java/ai/vespa/llm/test/MockLanguageModel.java @@ -7,6 +7,8 @@ import ai.vespa.llm.completion.Prompt; import com.yahoo.api.annotations.Beta; import java.util.List; +import java.util.concurrent.CompletableFuture; +import java.util.function.Consumer; import java.util.function.Function; /** @@ -26,6 +28,11 @@ public class MockLanguageModel implements LanguageModel { return completer.apply(prompt); } + @Override + public CompletableFuture<Completion.FinishReason> completeAsync(Prompt prompt, Consumer<Completion> action) { + throw new RuntimeException("Not implemented"); + } + public static class Builder { private Function<Prompt, List<Completion>> completer = prompt -> List.of(Completion.from("")); |