Updates max tokens

Ironclad · Dec 14, 2023 · 0c0138a · 0c0138a
1 parent 66d808f
commit 0c0138a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/core/src/plugins/google/google.ts b/packages/core/src/plugins/google/google.ts
@@ -9,15 +9,15 @@ export type GoogleModel = {
 
 export const googleModels = {
   'gemini-pro': {
-    maxTokens: 32000,
+    maxTokens: 32760,
     cost: {
       prompt: NaN,
       completion: NaN,
     },
     displayName: 'Gemini Pro',
   },
   'gemini-pro-vision': {
-    maxTokens: 16000,
+    maxTokens: 16384,
     cost: {
       prompt: NaN,
       completion: NaN,