Skip to content

Commit 79a6913

Browse files
committed
Merge branch 'feature/add-new-model' into develop
2 parents ea8cfe8 + 2c9f217 commit 79a6913

1 file changed

Lines changed: 18 additions & 12 deletions

File tree

Tool/Sources/Preferences/ChatGPTModel.swift

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,20 @@
11
import Foundation
22

33
public enum ChatGPTModel: String {
4+
case gpt35Turbo = "gpt-3.5-turbo"
5+
case gpt35Turbo16k = "gpt-3.5-turbo-16k"
46
case gpt4 = "gpt-4"
5-
6-
case gpt40314 = "gpt-4-0314"
7-
87
case gpt432k = "gpt-4-32k"
9-
10-
case gpt432k0314 = "gpt-4-32k-0314"
11-
12-
case gpt35Turbo = "gpt-3.5-turbo"
13-
8+
case gpt40314 = "gpt-4-0314"
9+
case gpt40613 = "gpt-4-0613"
1410
case gpt35Turbo0301 = "gpt-3.5-turbo-0301"
11+
case gpt35Turbo0613 = "gpt-3.5-turbo-0613"
12+
case gpt35Turbo16k0613 = "gpt-3.5-turbo-16k-0613"
13+
case gpt432k0314 = "gpt-4-32k-0314"
14+
case gpt432k0613 = "gpt-4-32k-0613"
1515
}
1616

1717
public extension ChatGPTModel {
18-
var endpoint: String {
19-
"https://api.openai.com/v1/chat/completions"
20-
}
21-
2218
var maxToken: Int {
2319
switch self {
2420
case .gpt4:
@@ -33,6 +29,16 @@ public extension ChatGPTModel {
3329
return 4096
3430
case .gpt35Turbo0301:
3531
return 4096
32+
case .gpt35Turbo0613:
33+
return 4096
34+
case .gpt35Turbo16k:
35+
return 16384
36+
case .gpt35Turbo16k0613:
37+
return 16384
38+
case .gpt40613:
39+
return 8192
40+
case .gpt432k0613:
41+
return 32768
3642
}
3743
}
3844
}

0 commit comments

Comments
 (0)