From c5bd09e2c96f18dafeac4294eeeadc368105b458 Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 23 Oct 2024 07:10:22 +0700 Subject: [PATCH 1/6] Add model anthropic/claude-3-5-sonnet-20241022 --- 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 41ee47b6..d43e673c 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -44,6 +44,7 @@ MAX_TOKENS = { 'anthropic.claude-v2': 100000, 'anthropic/claude-3-opus-20240229': 100000, 'anthropic/claude-3-5-sonnet-20240620': 100000, + 'anthropic/claude-3-5-sonnet-20241022': 100000, 'bedrock/anthropic.claude-instant-v1': 100000, 'bedrock/anthropic.claude-v2': 100000, 'bedrock/anthropic.claude-v2:1': 100000, From f1d0401f824820a49141917dbdddb3b1218cc908 Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 23 Oct 2024 07:11:33 +0700 Subject: [PATCH 2/6] Add model vertex_ai/claude-3-5-sonnet-v2@20241022 --- 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 d43e673c..b435da4f 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -34,6 +34,7 @@ MAX_TOKENS = { 'vertex_ai/claude-3-sonnet@20240229': 100000, '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/gemini-1.5-pro': 1048576, 'vertex_ai/gemini-1.5-flash': 1048576, 'vertex_ai/gemma2': 8200, From 6c55a2720ab93b821e1f1f08169326428fcf293a Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 23 Oct 2024 07:12:39 +0700 Subject: [PATCH 3/6] Update litellm to v1.50.2 to support all new added models --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 05f13ad7..5870cd2e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,7 +10,7 @@ GitPython==3.1.41 google-cloud-aiplatform==1.38.0 google-cloud-storage==2.10.0 Jinja2==3.1.2 -litellm==1.43.13 +litellm==1.50.2 loguru==0.7.2 msrest==0.7.1 openai==1.46.0 From d491a942ccd77949fec45a691937a1d1991a6c55 Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 23 Oct 2024 07:20:35 +0700 Subject: [PATCH 4/6] Update openai to v1.52.1 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 5870cd2e..2686da3b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,7 +13,7 @@ Jinja2==3.1.2 litellm==1.50.2 loguru==0.7.2 msrest==0.7.1 -openai==1.46.0 +openai==1.52.1 pytest==7.4.0 PyGithub==1.59.* PyYAML==6.0.1 From a95755426254389a3fca6d2e35e9223a7e6fc884 Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 23 Oct 2024 07:44:58 +0700 Subject: [PATCH 5/6] Update anthropic to v0.37.1 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 2686da3b..152e6a5b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ aiohttp==3.9.5 -anthropic[vertex]==0.21.3 +anthropic[vertex]==0.37.1 atlassian-python-api==3.41.4 azure-devops==7.1.0b3 azure-identity==1.15.0 From b4700bd7c0198c12ed7f7fc817691c75e91c809e Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Wed, 23 Oct 2024 08:08:04 +0700 Subject: [PATCH 6/6] Add support bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0 model --- 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 b435da4f..dbdf673b 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -52,6 +52,7 @@ MAX_TOKENS = { 'bedrock/anthropic.claude-3-sonnet-20240229-v1:0': 100000, 'bedrock/anthropic.claude-3-haiku-20240307-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, 'groq/llama3-8b-8192': 8192, 'groq/llama3-70b-8192': 8192,