Skip to content

Commit 284ed9e

Browse files
committed
Add new openai models
1 parent 741fe16 commit 284ed9e

1 file changed

Lines changed: 19 additions & 17 deletions

File tree

Tool/Sources/Preferences/Types/ChatGPTModel.swift

Lines changed: 19 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -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

2628
public 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

Comments
 (0)