Get rid of third party openai client

2 files changed, 29 insertions, 3 deletions

diff --git a/vespajlib/src/test/java/ai/vespa/llm/client/openai/OpenAiClientCompletionTest.java b/vespajlib/src/test/java/ai/vespa/llm/client/openai/OpenAiClientCompletionTest.java
new file mode 100644
index 00000000000..961a02afea3
--- /dev/null
+++ b/vespajlib/src/test/java/ai/vespa/llm/client/openai/OpenAiClientCompletionTest.java
@@ -0,0 +1,29 @@
+package ai.vespa.llm.client.openai;
+
+import ai.vespa.llm.completion.Completion;
+import ai.vespa.llm.completion.StringPrompt;
+import org.junit.jupiter.api.Disabled;
+import org.junit.jupiter.api.Test;
+
+/**
+ * @author bratseth
+ */
+public class OpenAiClientCompletionTest {
+
+    @Test
+    @Disabled
+    public void testClient() {
+        var client = new OpenAiClient.Builder("your token here").build();
+        String input = "You are an unhelpful assistant who never answers questions straightforwardly. " +
+                       "Be as long-winded as possible. Are humans smarter than cats?";
+        StringPrompt prompt = StringPrompt.from(input);
+        System.out.print(prompt);
+        for (int i = 0; i < 10; i++) {
+            var completion = client.complete(prompt).get(0);
+            System.out.print(completion.text());
+            if (completion.finishReason() == Completion.FinishReason.stop) break;
+            prompt = prompt.append(completion.text());
+        }
+    }
+
+}
diff --git a/vespajlib/src/test/java/ai/vespa/llm/completion/CompletionTest.java b/vespajlib/src/test/java/ai/vespa/llm/completion/CompletionTest.java
index 1c794c64d1a..26508228ab6 100644
--- a/vespajlib/src/test/java/ai/vespa/llm/completion/CompletionTest.java
+++ b/vespajlib/src/test/java/ai/vespa/llm/completion/CompletionTest.java
@@ -1,8 +1,5 @@
 package ai.vespa.llm.completion;
 
-import ai.vespa.llm.completion.Completion;
-import ai.vespa.llm.completion.Prompt;
-import ai.vespa.llm.completion.StringPrompt;
 import ai.vespa.llm.test.MockLanguageModel;
 import org.junit.jupiter.api.Test;