From 93abaae3cef30c0e8bed2afb75bf45dc7d23824e Mon Sep 17 00:00:00 2001
From: Ronald Mannak <ronaldmannak@me.com>
Date: Thu, 18 Jul 2024 21:07:47 -0700
Subject: [PATCH] Add support for GPT4o Mini

---
 Sources/CleverBird/chat/ChatModel.swift             | 5 +++++
 Sources/CleverBird/chat/ChatThread+tokenCount.swift | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/Sources/CleverBird/chat/ChatModel.swift b/Sources/CleverBird/chat/ChatModel.swift
index a531599..1a965d3 100644
--- 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
index e29a2e1..573ed85 100644
--- 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