From d62796ac681684fae8407222a6f5b26a5c5990ef Mon Sep 17 00:00:00 2001
From: koid <yukinori.koide@gunosy.com>
Date: Wed, 13 Mar 2024 11:14:04 +0900
Subject: [PATCH] update max_tokens

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

diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index 7c537e3a..cd535416 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -23,4 +23,8 @@ MAX_TOKENS = {
     'anthropic.claude-v1': 100000,
     'anthropic.claude-v2': 100000,
     'anthropic/claude-3-opus-20240229': 100000,
+    'bedrock/anthropic.claude-instant-v1': 100000,
+    'bedrock/anthropic.claude-v2': 100000,
+    'bedrock/anthropic.claude-v2:1': 100000,
+    'bedrock/anthropic.claude-3-sonnet-20240229-v1:0': 100000,
 }