diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index cfa1bbe30..72eb79120 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -66,10 +66,10 @@
     'claude-3-5-sonnet': 100000,
     'groq/llama3-8b-8192': 8192,
     'groq/llama3-70b-8192': 8192,
+    'groq/llama-3.1-8b-instant': 8192,
+    'groq/llama-3.3-70b-versatile': 128000,
     'groq/mixtral-8x7b-32768': 32768,
-    'groq/llama-3.1-8b-instant': 131072,
-    'groq/llama-3.1-70b-versatile': 131072,
-    'groq/llama-3.1-405b-reasoning': 131072,
+    'groq/gemma2-9b-it': 8192,
     'ollama/llama3': 4096,
     'watsonx/meta-llama/llama-3-8b-instruct': 4096,
     "watsonx/meta-llama/llama-3-70b-instruct": 4096,