From af653a048f219fe916a16f260c22780e7db4abeb Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Thu, 7 Nov 2024 09:12:52 +0700 Subject: [PATCH] Add support model bedrock/anthropic.claude-3-5-haiku-20241022-v1:0 --- pr_agent/algo/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index f45cfd02..2a9b1f34 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -55,6 +55,7 @@ MAX_TOKENS = { 'bedrock/anthropic.claude-v2:1': 100000, 'bedrock/anthropic.claude-3-sonnet-20240229-v1:0': 100000, '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, 'claude-3-5-sonnet': 100000,