diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index cd535416..9e05ed9f 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -27,4 +27,5 @@ MAX_TOKENS = {
     'bedrock/anthropic.claude-v2': 100000,
     'bedrock/anthropic.claude-v2:1': 100000,
     'bedrock/anthropic.claude-3-sonnet-20240229-v1:0': 100000,
+    'bedrock/anthropic.claude-3-haiku-20240307-v1:0': 100000,
 }
diff --git a/requirements.txt b/requirements.txt
index 6646b2e5..97df2f93 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,7 +9,7 @@ GitPython==3.1.32
 google-cloud-aiplatform==1.35.0
 google-cloud-storage==2.10.0
 Jinja2==3.1.2
-litellm==1.29.7
+litellm==1.31.10
 loguru==0.7.2
 msrest==0.7.1
 openai==1.13.3