fix: remove deprecated image and change the default to the best and c…

…heapest one
kestra-io · Aug 9, 2024 · 15a0d65 · 15a0d65
1 parent 85cece7
commit 15a0d65
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/src/main/java/io/kestra/plugin/openai/ChatCompletion.java b/src/main/java/io/kestra/plugin/openai/ChatCompletion.java
@@ -46,7 +46,7 @@
                 "  - id: completion",
                 "    type: io.kestra.plugin.openai.ChatCompletion",
                 "    apiKey: \"yourOpenAIapiKey\"",
-                "    model: gpt-3.5-turbo-0613",
+                "    model: gpt-4o",
                 "    prompt: \"{{inputs.prompt}}\"",
                 "",
                 "  - id: response",
@@ -72,7 +72,7 @@
                 "  - id: prioritize_response",
                 "    type: io.kestra.plugin.openai.ChatCompletion",
                 "    apiKey: \"yourOpenAIapiKey\"",
-                "    model: gpt-4",
+                "    model: gpt-4o",
                 "    messages:",
                 "      - role: user",
                 "        content: \"{{ inputs.prompt }}\"",

diff --git a/src/test/java/io/kestra/plugin/openai/ChatCompletionTest.java b/src/test/java/io/kestra/plugin/openai/ChatCompletionTest.java
@@ -34,15 +34,15 @@ void runMessages() throws Exception {
 
         ChatCompletion task = ChatCompletion.builder()
             .apiKey(this.apiKey)
-            .model("gpt-3.5-turbo-0613")
+            .model("gpt-4o")
             .clientTimeout(30)
             .messages(messages)
             .build();
 
         ChatCompletion.Output runOutput = task.run(runContext);
 
         assertThat(runOutput.getChoices().get(0).getMessage().getContent(), containsString("Paris"));
-        assertThat(runOutput.getModel(), containsString("gpt-3.5-turbo"));
+        assertThat(runOutput.getModel(), containsString("gpt-4o"));
         assertThat(runOutput.getUsage().getPromptTokens(), is(14L));
     }
 
@@ -52,14 +52,14 @@ void runPrompt() throws Exception {
 
         ChatCompletion task = ChatCompletion.builder()
             .apiKey(this.apiKey)
-            .model("gpt-3.5-turbo")
+            .model("gpt-4o")
             .prompt("what is the capital of France?")
             .build();
 
         ChatCompletion.Output runOutput = task.run(runContext);
 
         assertThat(runOutput.getChoices().get(0).getMessage().getContent(), containsString("Paris"));
-        assertThat(runOutput.getModel(), containsString("gpt-3.5-turbo"));
+        assertThat(runOutput.getModel(), containsString("gpt-4o"));
         assertThat(runOutput.getUsage().getPromptTokens(), is(14L));
     }
 
@@ -74,15 +74,15 @@ void runMessagesWithPrompt() throws Exception {
 
         ChatCompletion task = ChatCompletion.builder()
             .apiKey(this.apiKey)
-            .model("gpt-3.5-turbo-0613")
+            .model("gpt-4o")
             .messages(messages)
             .prompt("and the capital of germany?")
             .build();
 
         ChatCompletion.Output runOutput = task.run(runContext);
 
         assertThat(runOutput.getChoices().get(0).getMessage().getContent(), containsString("Berlin"));
-        assertThat(runOutput.getModel(), containsString("gpt-3.5-turbo"));
+        assertThat(runOutput.getModel(), containsString("gpt-4o"));
         assertThat(runOutput.getUsage().getPromptTokens(), is(35L));
     }
 
@@ -120,7 +120,7 @@ void runFunction() throws Exception {
 
         ChatCompletion task = ChatCompletion.builder()
             .apiKey(this.apiKey)
-            .model("gpt-3.5-turbo")
+            .model("gpt-4o")
             .messages(messages)
             .functions(functions)
             .functionCall("auto")
@@ -131,7 +131,7 @@ void runFunction() throws Exception {
 
         assertThat(functionCall.getName(), containsString("test"));
         assertThat(functionCall.getArguments().get("location").toString(), containsString("Lyon"));
-        assertThat(runOutput.getModel(), containsString("gpt-3.5-turbo"));
+        assertThat(runOutput.getModel(), containsString("gpt-4o"));
     }
 
     @Test
@@ -174,7 +174,7 @@ void runFunctionWithEnumValues() throws Exception {
 
         ChatCompletion task = ChatCompletion.builder()
             .apiKey(this.apiKey)
-            .model("gpt-3.5-turbo")
+            .model("gpt-4o")
             .messages(messages)
             .functions(functions)
             .functionCall("record_customer_rating")
@@ -189,7 +189,7 @@ void runFunctionWithEnumValues() throws Exception {
             .get("food_eaten").toString()
             .toLowerCase(), containsString("steak"));
         assertThat(functionCall.getArguments().get("customer_name").toString(), containsString("John Smith"));
-        assertThat(runOutput.getModel(), containsString("gpt-3.5-turbo"));
+        assertThat(runOutput.getModel(), containsString("gpt-4o"));
     }
 
     private ChatMessage buildMessage(String role, String content) {