From 157251493acdc6d0a273700e2bf9a3a3a22432e6 Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 6 Nov 2024 07:52:58 +0700 Subject: [PATCH 1/6] Add support claude-3-5-haiku-20241022 model --- pr_agent/algo/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 83d001a15..f45cfd02a 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -31,6 +31,7 @@ 'vertex_ai/codechat-bison': 6144, 'vertex_ai/codechat-bison-32k': 32000, 'vertex_ai/claude-3-haiku@20240307': 100000, + 'vertex_ai/claude-3-5-haiku@20241022': 100000, 'vertex_ai/claude-3-sonnet@20240229': 100000, 'vertex_ai/claude-3-opus@20240229': 100000, 'vertex_ai/claude-3-5-sonnet@20240620': 100000, @@ -48,6 +49,7 @@ 'anthropic/claude-3-opus-20240229': 100000, 'anthropic/claude-3-5-sonnet-20240620': 100000, 'anthropic/claude-3-5-sonnet-20241022': 100000, + 'anthropic/claude-3-5-haiku-20241022': 100000, 'bedrock/anthropic.claude-instant-v1': 100000, 'bedrock/anthropic.claude-v2': 100000, 'bedrock/anthropic.claude-v2:1': 100000, From c394fc27672f6eb328a2d9e36dd2799807e5a9a0 Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 6 Nov 2024 07:55:19 +0700 Subject: [PATCH 2/6] Upgrade anthropic version to 0.39.0 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index ec88a4ca1..21d2d162f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ aiohttp==3.9.5 -anthropic[vertex]==0.37.1 +anthropic[vertex]==0.39.0 atlassian-python-api==3.41.4 azure-devops==7.1.0b3 azure-identity==1.15.0 From c5ed2f040a086c16bf34fc6997dbe24592819ccb Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 6 Nov 2024 07:56:30 +0700 Subject: [PATCH 3/6] Update litellm to v1..52.0 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 21d2d162f..b57fd4941 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,7 +12,7 @@ google-cloud-aiplatform==1.38.0 google-generativeai==0.8.3 google-cloud-storage==2.10.0 Jinja2==3.1.2 -litellm==1.50.2 +litellm==1.52.0 loguru==0.7.2 msrest==0.7.1 openai==1.52.1 From 0eacb3e35ee2169bf12d33c0d8d2485a75171ce3 Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 6 Nov 2024 08:07:19 +0700 Subject: [PATCH 4/6] Update openai to v1.54.1 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index b57fd4941..75a52b72b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,7 +15,7 @@ Jinja2==3.1.2 litellm==1.52.0 loguru==0.7.2 msrest==0.7.1 -openai==1.52.1 +openai==1.54.1 pytest==7.4.0 PyGithub==1.59.* PyYAML==6.0.1 From 04496f9b0eb2a246ef86ebcb43c228a2525c7524 Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 6 Nov 2024 08:07:44 +0700 Subject: [PATCH 5/6] Update tiktoken to v0.8.0 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 75a52b72b..afb6af682 100644 --- a/requirements.txt +++ b/requirements.txt @@ -22,7 +22,7 @@ PyYAML==6.0.1 python-gitlab==3.15.0 retry==0.9.2 starlette-context==0.3.6 -tiktoken==0.7.0 +tiktoken==0.8.0 ujson==5.8.0 uvicorn==0.22.0 tenacity==8.2.3 From af653a048f219fe916a16f260c22780e7db4abeb Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Thu, 7 Nov 2024 09:12:52 +0700 Subject: [PATCH 6/6] 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 f45cfd02a..2a9b1f34f 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -55,6 +55,7 @@ '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,