From e7de81c29319d605703a874c3f1fa56f61122e8b Mon Sep 17 00:00:00 2001
From: Theo Kanning <TheoKanning@users.noreply.github.com>
Date: Wed, 13 Dec 2023 16:37:13 -0600
Subject: [PATCH] Remove references to deprecated models (#438)

---
 README.md                                                   | 2 +-
 example/src/main/java/example/OpenAiApiExample.java         | 2 +-
 .../java/com/theokanning/openai/service/CompletionTest.java | 4 ++--
 .../java/com/theokanning/openai/service/EmbeddingTest.java  | 2 +-
 .../test/java/com/theokanning/openai/service/ModelTest.java | 6 +++---
 5 files changed, 8 insertions(+), 8 deletions(-)

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