Skip to content

Commit ae8d75f

Browse files
committed
Add gemini models
1 parent 3eaca6b commit ae8d75f

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

Tool/Sources/Preferences/Types/GoogleGenerativeChatModel.swift

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
import Foundation
22

33
public enum GoogleGenerativeAIModel: String {
4+
case gemini15Pro = "gemini-1.5-pro"
5+
case gemini15Flash = "gemini-1.5-flash"
46
case geminiPro = "gemini-pro"
57
}
68

@@ -9,6 +11,10 @@ public extension GoogleGenerativeAIModel {
911
switch self {
1012
case .geminiPro:
1113
return 32768
14+
case .gemini15Flash:
15+
return 1_048_576
16+
case .gemini15Pro:
17+
return 2_097_152
1218
}
1319
}
1420
}

0 commit comments

Comments
 (0)