Add support for GPT4o Mini (#23)

btfranklin · Jul 19, 2024 · d100fb1 · d100fb1
1 parent b30d43f
commit d100fb1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/Sources/CleverBird/chat/ChatModel.swift b/Sources/CleverBird/chat/ChatModel.swift
@@ -1,5 +1,6 @@
 public enum ChatModel: Codable {
     case gpt35Turbo
+    case gpt4oMini
     case gpt4
     case gpt4Turbo
     case gpt4o
@@ -12,6 +13,8 @@ public enum ChatModel: Codable {
         switch modelString {
         case _ where modelString.starts(with: "gpt-3.5"):
             self = .gpt35Turbo
+        case _ where modelString.starts(with: "gpt-4o-mini"):
+            self = .gpt4oMini
         case _ where modelString.starts(with: "gpt-4o"):
             self = .gpt4o
         case _ where modelString.starts(with: "gpt-4-turbo"):
@@ -36,6 +39,8 @@ extension ChatModel: CustomStringConvertible {
         switch self {
         case .gpt35Turbo:
             return "gpt-3.5-turbo"
+        case .gpt4oMini:
+            return "gpt-4o-mini"
         case .gpt4o:
             return "gpt-4o"
         case .gpt4Turbo:

diff --git a/Sources/CleverBird/chat/ChatThread+tokenCount.swift b/Sources/CleverBird/chat/ChatThread+tokenCount.swift
@@ -19,7 +19,7 @@ extension ChatThread {
             tokensPerMessage = 4
         case .gpt4, .gpt4Turbo:
             tokensPerMessage = 3
-        case .gpt4o:
+        case .gpt4o, .gpt4oMini:
             tokensPerMessage = 3
         case .specific(_):
             tokensPerMessage = 3