From d3a7041f0d5a76971f506abf9e7adb7366f3267e Mon Sep 17 00:00:00 2001
From: Kamakura <barnett.yuxiang@gmail.com>
Date: Tue, 4 Jun 2024 00:00:22 +0800
Subject: [PATCH] update alog/__init__.py

---
 pr_agent/algo/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index 43078f8d..163b45eb 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -36,5 +36,5 @@ MAX_TOKENS = {
     'bedrock/anthropic.claude-3-haiku-20240307-v1:0': 100000,
     'groq/llama3-8b-8192': 8192,
     'groq/llama3-70b-8192': 8192,
-    "ollama/llama3": 4096,
+    'llama/llama3': 4096,
 }