Add new GPT4o model (#43)

SwiftBeta · May 13, 2024 · 841299d · 841299d
1 parent a72d111
commit 841299d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/Demo/Demo/ChatCompletions/ChatCompletionsViewModel.swift b/Demo/Demo/ChatCompletions/ChatCompletionsViewModel.swift
@@ -23,7 +23,7 @@ final class ChatCompletionsViewModel {
                                                                    maxTokens: 2000)
         if isStream {
             do {
-                for try await newMessage in try await openAI.createChatCompletionsStream(model: .gpt4(.base),
+                for try await newMessage in try await openAI.createChatCompletionsStream(model: .gpt4o(.base),
                                                                                          messages: messages,
                                                                                          optionalParameters: optionalParameters) {
                     onReceiveStream(newMessage: newMessage)

diff --git a/README.md b/README.md
@@ -233,7 +233,7 @@ let optionalParameters = ChatCompletionsOptionalParameters(
 do {
     // Create a chat completion stream using the OpenAI API.
     let stream = try await openAI.createChatCompletionsStream(
-        model: .gpt4(.base), // Specify the model, here GPT-4 base model.
+        model: .gpt4o(.base), // Specify the model, here GPT-4 base model.
         messages: messages, // Provide the conversation messages.
         optionalParameters: optionalParameters // Include the optional parameters.
     )

diff --git a/Sources/SwiftOpenAI/OpenAI/DataModels/OpenAIModelType.swift b/Sources/SwiftOpenAI/OpenAI/DataModels/OpenAIModelType.swift
@@ -1,12 +1,15 @@
 import Foundation
 
 public enum OpenAIModelType {
+    case gpt4o(GPT4o)
     case gpt4(GPT4)
     case gpt3_5(GPT3_5)
     case embedding(EmbeddingModel)
 
     var name: String {
         switch self {
+        case .gpt4o(let gpt4oModel):
+            return gpt4oModel.rawValue
         case .gpt4(let gpt4Model):
             return gpt4Model.rawValue
         case .gpt3_5(let gpt3_5Model):
@@ -28,6 +31,11 @@ public enum OpenAIImageModelType {
     }
 }
 
+public enum GPT4o: String {
+    case base = "gpt-4o"
+    case gpt_4o_2024_05_13 = "gpt-4o-2024-05-13"
+}
+
 public enum GPT4: String {
     case base = "gpt-4-turbo"
     case gpt_4_turbo_2024_04_09 = "gpt-4-turbo-2024-04-09"