From 4047e71268c040b1d4456deccfaf5f433121dff1 Mon Sep 17 00:00:00 2001 From: Peter Dave Hello Date: Sat, 24 May 2025 23:52:18 +0800 Subject: [PATCH] Complete Gemini 2.5 Flash model info --- pr_agent/algo/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 23c9795b..a2b35e71 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -64,10 +64,13 @@ MAX_TOKENS = { 'vertex_ai/gemini-1.5-flash': 1048576, 'vertex_ai/gemini-2.0-flash': 1048576, 'vertex_ai/gemini-2.5-flash-preview-04-17': 1048576, + 'vertex_ai/gemini-2.5-flash-preview-05-20': 1048576, 'vertex_ai/gemma2': 8200, 'gemini/gemini-1.5-pro': 1048576, 'gemini/gemini-1.5-flash': 1048576, 'gemini/gemini-2.0-flash': 1048576, + 'gemini/gemini-2.5-flash-preview-04-17': 1048576, + 'gemini/gemini-2.5-flash-preview-05-20': 1048576, 'gemini/gemini-2.5-pro-preview-03-25': 1048576, 'gemini/gemini-2.5-pro-preview-05-06': 1048576, 'codechat-bison': 6144,