replace deprecated openAI model in test

Signed-off-by: Hailong Cui <[email protected]>
Hailong-am · Jan 9, 2024 · 0a59fd1 · 0a59fd1
1 parent 19cec10
commit 0a59fd1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java
@@ -39,7 +39,7 @@ public class RestMLRemoteInferenceIT extends MLCommonsRestTestCase {
         + "    \"content_type\": \"application/json\",\n"
         + "    \"max_tokens\": 7,\n"
         + "    \"temperature\": 0,\n"
-        + "    \"model\": \"text-davinci-003\"\n"
+        + "    \"model\": \"gpt-3.5-turbo-instruct\"\n"
         + "  },\n"
         + "  \"credential\": {\n"
         + "    \"openAI_key\": \""
@@ -265,7 +265,7 @@ public void testOpenAIEditsModel() throws IOException, InterruptedException {
             + "      \"endpoint\": \"api.openai.com\",\n"
             + "      \"auth\": \"API_Key\",\n"
             + "      \"content_type\": \"application/json\",\n"
-            + "      \"model\": \"text-davinci-edit-001\"\n"
+            + "      \"model\": \"gpt-3.5-turbo\"\n"
             + "  },\n"
             + "  \"credential\": {\n"
             + "      \"openAI_key\": \""
@@ -276,7 +276,7 @@ public void testOpenAIEditsModel() throws IOException, InterruptedException {
             + "      {\n"
             + "      \"action_type\": \"predict\",\n"
             + "          \"method\": \"POST\",\n"
-            + "          \"url\": \"https://api.openai.com/v1/edits\",\n"
+            + "          \"url\": \"https://api.openai.com/v1/chat/completions\",\n"
             + "          \"headers\": { \n"
             + "          \"Authorization\": \"Bearer ${credential.openAI_key}\"\n"
             + "          },\n"