diff --git a/Sources/SwiftOpenAI/OpenAI/DataModels/OpenAIModelType.swift b/Sources/SwiftOpenAI/OpenAI/DataModels/OpenAIModelType.swift index 201c97e..5cb5bbf 100644 --- a/Sources/SwiftOpenAI/OpenAI/DataModels/OpenAIModelType.swift +++ b/Sources/SwiftOpenAI/OpenAI/DataModels/OpenAIModelType.swift @@ -2,6 +2,7 @@ import Foundation public enum OpenAIModelType { case gpt4o(GPT4o) + case gpt4o_mini(GPT4o_mini) case gpt4(GPT4) case gpt3_5(GPT3_5) case embedding(EmbeddingModel) @@ -10,6 +11,8 @@ public enum OpenAIModelType { switch self { case .gpt4o(let gpt4oModel): return gpt4oModel.rawValue + case .gpt4o_mini(let gpt4oMiniModel): + return gpt4oMiniModel.rawValue case .gpt4(let gpt4Model): return gpt4Model.rawValue case .gpt3_5(let gpt3_5Model): @@ -36,6 +39,10 @@ public enum GPT4o: String { case gpt_4o_2024_05_13 = "gpt-4o-2024-05-13" } +public enum GPT4o_mini: String { + case base = "gpt-4o-mini" +} + public enum GPT4: String { case base = "gpt-4-turbo" case gpt_4_turbo_2024_04_09 = "gpt-4-turbo-2024-04-09"