From a24b06b253ac8fabf21b6c1182c0bc3711a694bb Mon Sep 17 00:00:00 2001
From: Hiroyuki Otomo <otomo_hiroyuki@cyberagent.co.jp>
Date: Tue, 25 Feb 2025 12:58:20 +0900
Subject: [PATCH] feat: support Claude 3.7 Sonnet

---
 pr_agent/algo/__init__.py | 3 +++
 requirements.txt          | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index 37ca48ac..b56ff88f 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -43,6 +43,7 @@ MAX_TOKENS = {
     'vertex_ai/claude-3-opus@20240229': 100000,
     'vertex_ai/claude-3-5-sonnet@20240620': 100000,
     'vertex_ai/claude-3-5-sonnet-v2@20241022': 100000,
+    'vertex_ai/claude-3-7-sonnet@20250219': 100000,
     'vertex_ai/gemini-1.5-pro': 1048576,
     'vertex_ai/gemini-1.5-flash': 1048576,
     'vertex_ai/gemini-2.0-flash': 1048576,
@@ -58,6 +59,7 @@ MAX_TOKENS = {
     'anthropic/claude-3-opus-20240229': 100000,
     'anthropic/claude-3-5-sonnet-20240620': 100000,
     'anthropic/claude-3-5-sonnet-20241022': 100000,
+    'anthropic/claude-3-7-sonnet-20250219': 100000,
     'anthropic/claude-3-5-haiku-20241022': 100000,
     'bedrock/anthropic.claude-instant-v1': 100000,
     'bedrock/anthropic.claude-v2': 100000,
@@ -67,6 +69,7 @@ MAX_TOKENS = {
     'bedrock/anthropic.claude-3-5-haiku-20241022-v1:0': 100000,
     'bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0': 100000,
     'bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0': 100000,
+    'bedrock/anthropic.claude-3-7-sonnet-20250219-v1:0': 100000,
     "bedrock/us.anthropic.claude-3-5-sonnet-20241022-v2:0": 100000,
     'claude-3-5-sonnet': 100000,
     'groq/llama3-8b-8192': 8192,
diff --git a/requirements.txt b/requirements.txt
index 5fe1856f..8e542ab8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 aiohttp==3.9.5
-anthropic[vertex]==0.39.0
+anthropic[vertex]==0.47.1
 atlassian-python-api==3.41.4
 azure-devops==7.1.0b3
 azure-identity==1.15.0