Merge pull request #1397 from KennyDizi/main

Add support for `gemini/gemini-2.0-flash-exp` model
Codium-ai · Dec 16, 2024 · a768969 · a768969
2 parents 9ef9198 + 03d2bea
commit a768969
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
@@ -42,6 +42,7 @@
     'vertex_ai/gemma2': 8200,
     'gemini/gemini-1.5-pro': 1048576,
     'gemini/gemini-1.5-flash': 1048576,
+    'gemini/gemini-2.0-flash-exp': 1048576,
     'codechat-bison': 6144,
     'codechat-bison-32k': 32000,
     'anthropic.claude-instant-v1': 100000,