From 1f836e405d4a7a1313a94d7bf3cb7b1aeb48be5a Mon Sep 17 00:00:00 2001
From: Hiroyuki Otomo <otomo_hiroyuki@cyberagent.co.jp>
Date: Sat, 24 May 2025 09:45:27 +0900
Subject: [PATCH] fix: reflect comments

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

diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index f9a055f1..dbda2f43 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -87,15 +87,17 @@ 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-opus-4-20250514-v1:0': 100000,
+    'bedrock/anthropic.claude-opus-4-20250514-v1:0': 200000,
     'bedrock/anthropic.claude-3-haiku-20240307-v1:0': 100000,
     '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': 200000,
     'bedrock/anthropic.claude-sonnet-4-20250514-v1:0': 200000,
+    "bedrock/us.anthropic.claude-opus-4-20250514-v1:0": 200000,
     "bedrock/us.anthropic.claude-3-5-sonnet-20241022-v2:0": 100000,
     "bedrock/us.anthropic.claude-3-7-sonnet-20250219-v1:0": 200000,
+    "bedrock/us.anthropic.claude-sonnet-4-20250514-v1:0": 200000,
     'claude-3-5-sonnet': 100000,
     'groq/meta-llama/llama-4-scout-17b-16e-instruct': 131072,
     'groq/meta-llama/llama-4-maverick-17b-128e-instruct': 131072,