@@ -14,13 +14,15 @@ public enum ChatGPTModel: String {
1414 case gpt4VisionPreview = " gpt-4-vision-preview "
1515 case gpt4TurboPreview = " gpt-4-turbo-preview "
1616 case gpt4Turbo20240409 = " gpt-4-turbo-2024-04-09 "
17- case gpt35Turbo0613 = " gpt-3.5-turbo-0613 "
1817 case gpt35Turbo1106 = " gpt-3.5-turbo-1106 "
1918 case gpt35Turbo0125 = " gpt-3.5-turbo-0125 "
20- case gpt35Turbo16k0613 = " gpt-3.5-turbo-16k-0613 "
2119 case gpt432k0314 = " gpt-4-32k-0314 "
2220 case gpt432k0613 = " gpt-4-32k-0613 "
2321 case gpt40125 = " gpt-4-0125-preview "
22+ case o1Preview = " o1-preview "
23+ case o1Preview20240912 = " o1-preview-2024-09-12 "
24+ case o1Mini = " o1-mini "
25+ case o1Mini20240912 = " o1-mini-2024-09-12 "
2426}
2527
2628public extension ChatGPTModel {
@@ -36,47 +38,47 @@ public extension ChatGPTModel {
3638 return 32768
3739 case . gpt35Turbo:
3840 return 16385
39- case . gpt35Turbo0613:
40- return 4096
4141 case . gpt35Turbo1106:
4242 return 16385
4343 case . gpt35Turbo0125:
4444 return 16385
4545 case . gpt35Turbo16k:
4646 return 16385
47- case . gpt35Turbo16k0613:
48- return 16385
4947 case . gpt40613:
5048 return 8192
5149 case . gpt432k0613:
5250 return 32768
5351 case . gpt41106Preview:
54- return 128000
52+ return 128_000
5553 case . gpt4VisionPreview:
56- return 128000
54+ return 128_000
5755 case . gpt4TurboPreview:
58- return 128000
56+ return 128_000
5957 case . gpt40125:
60- return 128000
58+ return 128_000
6159 case . gpt4Turbo:
62- return 128000
60+ return 128_000
6361 case . gpt4Turbo20240409:
64- return 128000
62+ return 128_000
6563 case . gpt4o:
66- return 128000
64+ return 128_000
6765 case . gpt4oMini:
68- return 128000
66+ return 128_000
67+ case . o1Preview, . o1Preview20240912:
68+ return 128_000
69+ case . o1Mini, . o1Mini20240912:
70+ return 128_000
6971 }
7072 }
71-
73+
7274 var supportsImages : Bool {
7375 switch self {
74- case . gpt4VisionPreview, . gpt4Turbo, . gpt4Turbo20240409, . gpt4o:
76+ case . gpt4VisionPreview, . gpt4Turbo, . gpt4Turbo20240409, . gpt4o, . gpt4oMini, . o1Preview,
77+ . o1Preview20240912, . o1Mini, . o1Mini20240912:
7578 return true
7679 default :
7780 return false
7881 }
7982 }
8083}
8184
82- extension ChatGPTModel : CaseIterable { }
0 commit comments