diff --git a/README.md b/README.md index b3626255..0ca55a82 100644 --- a/README.md +++ b/README.md @@ -61,7 +61,7 @@ If you're looking for the fastest solution, import the `service` module and use OpenAiService service = new OpenAiService("your_token"); CompletionRequest completionRequest = CompletionRequest.builder() .prompt("Somebody once told me the world is gonna roll me") - .model("ada") + .model("babbage-002"") .echo(true) .build(); service.createCompletion(completionRequest).getChoices().forEach(System.out::println); diff --git a/example/src/main/java/example/OpenAiApiExample.java b/example/src/main/java/example/OpenAiApiExample.java index 7a245ef9..52ae1ccf 100644 --- a/example/src/main/java/example/OpenAiApiExample.java +++ b/example/src/main/java/example/OpenAiApiExample.java @@ -19,7 +19,7 @@ public static void main(String... args) { System.out.println("\nCreating completion..."); CompletionRequest completionRequest = CompletionRequest.builder() - .model("ada") + .model("babbage-002") .prompt("Somebody once told me the world is gonna roll me") .echo(true) .user("testing") diff --git a/service/src/test/java/com/theokanning/openai/service/CompletionTest.java b/service/src/test/java/com/theokanning/openai/service/CompletionTest.java index c2ad90a1..69284c3b 100644 --- a/service/src/test/java/com/theokanning/openai/service/CompletionTest.java +++ b/service/src/test/java/com/theokanning/openai/service/CompletionTest.java @@ -21,7 +21,7 @@ public class CompletionTest { @Test void createCompletion() { CompletionRequest completionRequest = CompletionRequest.builder() - .model("ada") + .model("babbage-002") .prompt("Somebody once told me the world is gonna roll me") .echo(true) .n(5) @@ -39,7 +39,7 @@ void createCompletion() { @Test void streamCompletion() { CompletionRequest completionRequest = CompletionRequest.builder() - .model("ada") + .model("babbage-002") .prompt("Somebody once told me the world is gonna roll me") .echo(true) .n(1) diff --git a/service/src/test/java/com/theokanning/openai/service/EmbeddingTest.java b/service/src/test/java/com/theokanning/openai/service/EmbeddingTest.java index 89bb01af..3320dcea 100644 --- a/service/src/test/java/com/theokanning/openai/service/EmbeddingTest.java +++ b/service/src/test/java/com/theokanning/openai/service/EmbeddingTest.java @@ -18,7 +18,7 @@ public class EmbeddingTest { @Test void createEmbeddings() { EmbeddingRequest embeddingRequest = EmbeddingRequest.builder() - .model("text-similarity-babbage-001") + .model("text-embedding-ada-002") .input(Collections.singletonList("The food was delicious and the waiter...")) .build(); diff --git a/service/src/test/java/com/theokanning/openai/service/ModelTest.java b/service/src/test/java/com/theokanning/openai/service/ModelTest.java index 31d23da9..637ed5e6 100644 --- a/service/src/test/java/com/theokanning/openai/service/ModelTest.java +++ b/service/src/test/java/com/theokanning/openai/service/ModelTest.java @@ -23,9 +23,9 @@ void listModels() { @Test void getModel() { - Model ada = service.getModel("ada"); + Model model = service.getModel("babbage-002"); - assertEquals("ada", ada.id); - assertEquals("openai", ada.ownedBy); + assertEquals("babbage-002", model.id); + assertEquals("system", model.ownedBy); } }