From 495e2ccb7de0210902881af5a69af537b967203a Mon Sep 17 00:00:00 2001
From: dceoy <dnarsil+github@gmail.com>
Date: Sun, 28 Jul 2024 02:28:42 +0900
Subject: [PATCH] Add Llama 3.1 and Mixtral 8x7B for Groq

---
 pr_agent/algo/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index ca533001b..27b24c56e 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -47,5 +47,9 @@
     'bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0': 100000,
     'groq/llama3-8b-8192': 8192,
     'groq/llama3-70b-8192': 8192,
+    '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,
     'ollama/llama3': 4096,
 }