@@ -8,79 +8,70 @@ public enum ChatGPTModel: String, CaseIterable {
88 case gpt4 = " gpt-4 "
99 case gpt432k = " gpt-4-32k "
1010 case gpt4Turbo = " gpt-4-turbo "
11- case gpt40314 = " gpt-4-0314 "
12- case gpt40613 = " gpt-4-0613 "
13- case gpt41106Preview = " gpt-4-1106-preview "
1411 case gpt4VisionPreview = " gpt-4-vision-preview "
15- case gpt4TurboPreview = " gpt-4-turbo-preview "
16- case gpt4Turbo20240409 = " gpt-4-turbo-2024-04-09 "
17- case gpt35Turbo1106 = " gpt-3.5-turbo-1106 "
18- case gpt35Turbo0125 = " gpt-3.5-turbo-0125 "
1912 case gpt432k0314 = " gpt-4-32k-0314 "
2013 case gpt432k0613 = " gpt-4-32k-0613 "
2114 case gpt40125 = " gpt-4-0125-preview "
15+ case gpt4_1 = " gpt-4.1 "
16+ case gpt4_1Mini = " gpt-4.1-mini "
17+ case gpt4_1Nano = " gpt-4.1-nano "
2218 case o1 = " o1 "
2319 case o1Preview = " o1-preview "
24- case o1Preview20240912 = " o1-preview-2024-09-12 "
25- case o1Mini = " o1-mini "
26- case o1Mini20240912 = " o1-mini-2024-09-12 "
20+ case o1Pro = " o1-pro "
2721 case o3Mini = " o3-mini "
22+ case o3 = " o3 "
23+ case o4Mini = " o4-mini "
2824}
2925
3026public extension ChatGPTModel {
3127 var maxToken : Int {
3228 switch self {
3329 case . gpt4:
3430 return 8192
35- case . gpt40314:
36- return 8192
3731 case . gpt432k:
3832 return 32768
3933 case . gpt432k0314:
4034 return 32768
4135 case . gpt35Turbo:
4236 return 16385
43- case . gpt35Turbo1106:
44- return 16385
45- case . gpt35Turbo0125:
46- return 16385
4737 case . gpt35Turbo16k:
4838 return 16385
49- case . gpt40613:
50- return 8192
5139 case . gpt432k0613:
5240 return 32768
53- case . gpt41106Preview:
54- return 128_000
5541 case . gpt4VisionPreview:
5642 return 128_000
57- case . gpt4TurboPreview:
58- return 128_000
5943 case . gpt40125:
6044 return 128_000
6145 case . gpt4Turbo:
6246 return 128_000
63- case . gpt4Turbo20240409:
64- return 128_000
6547 case . gpt4o:
6648 return 128_000
6749 case . gpt4oMini:
6850 return 128_000
69- case . o1Preview, . o1Preview20240912:
70- return 128_000
71- case . o1Mini, . o1Mini20240912:
51+ case . o1Preview:
7252 return 128_000
7353 case . o1:
7454 return 200_000
7555 case . o3Mini:
7656 return 200_000
57+ case . gpt4_1:
58+ return 1_047_576
59+ case . gpt4_1Mini:
60+ return 1_047_576
61+ case . gpt4_1Nano:
62+ return 1_047_576
63+ case . o1Pro:
64+ return 200_000
65+ case . o3:
66+ return 200_000
67+ case . o4Mini:
68+ return 200_000
7769 }
7870 }
7971
8072 var supportsImages : Bool {
8173 switch self {
82- case . gpt4VisionPreview, . gpt4Turbo, . gpt4Turbo20240409, . gpt4o, . gpt4oMini, . o1Preview,
83- . o1Preview20240912, . o1Mini, . o1Mini20240912, . o1, . o3Mini:
74+ case . gpt4VisionPreview, . gpt4Turbo, . gpt4o, . gpt4oMini, . o1Preview, . o1, . o3Mini:
8475 return true
8576 default :
8677 return false
@@ -89,7 +80,7 @@ public extension ChatGPTModel {
8980
9081 var supportsTemperature : Bool {
9182 switch self {
92- case . o1Preview, . o1Preview20240912 , . o1Mini , . o1Mini20240912 , . o1, . o3Mini:
83+ case . o1Preview, . o1, . o3Mini:
9384 return false
9485 default :
9586 return true
@@ -98,7 +89,7 @@ public extension ChatGPTModel {
9889
9990 var supportsSystemPrompt : Bool {
10091 switch self {
101- case . o1Preview, . o1Preview20240912 , . o1Mini , . o1Mini20240912 , . o1, . o3Mini:
92+ case . o1Preview, . o1, . o3Mini:
10293 return false
10394 default :
10495 return true
0 commit comments