File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11import Foundation
22
33public 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
1717public 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}
You can’t perform that action at this time.
0 commit comments