From c53c6aee7fddcc49e78ec9d1d87979f52fcf7320 Mon Sep 17 00:00:00 2001
From: Kamakura <barnett.yuxiang@gmail.com>
Date: Tue, 4 Jun 2024 15:09:30 +0800
Subject: [PATCH] fix wrong provider name

---
 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 163b45eb..ad89b052 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,
-    'llama/llama3': 4096,
+    'ollama/llama3': 4096,
 }