File tree Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Original file line number Diff line number Diff line change 1
1
public enum ChatModel : Codable {
2
2
case gpt35Turbo
3
+ case gpt4oMini
3
4
case gpt4
4
5
case gpt4Turbo
5
6
case gpt4o
@@ -12,6 +13,8 @@ public enum ChatModel: Codable {
12
13
switch modelString {
13
14
case _ where modelString. starts ( with: " gpt-3.5 " ) :
14
15
self = . gpt35Turbo
16
+ case _ where modelString. starts ( with: " gpt-4o-mini " ) :
17
+ self = . gpt4oMini
15
18
case _ where modelString. starts ( with: " gpt-4o " ) :
16
19
self = . gpt4o
17
20
case _ where modelString. starts ( with: " gpt-4-turbo " ) :
@@ -36,6 +39,8 @@ extension ChatModel: CustomStringConvertible {
36
39
switch self {
37
40
case . gpt35Turbo:
38
41
return " gpt-3.5-turbo "
42
+ case . gpt4oMini:
43
+ return " gpt-4o-mini "
39
44
case . gpt4o:
40
45
return " gpt-4o "
41
46
case . gpt4Turbo:
Original file line number Diff line number Diff line change @@ -19,7 +19,7 @@ extension ChatThread {
19
19
tokensPerMessage = 4
20
20
case . gpt4, . gpt4Turbo:
21
21
tokensPerMessage = 3
22
- case . gpt4o:
22
+ case . gpt4o, . gpt4oMini :
23
23
tokensPerMessage = 3
24
24
case . specific( _) :
25
25
tokensPerMessage = 3
You can’t perform that action at this time.
0 commit comments