diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 41ee47b67..dbdf673b2 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -34,6 +34,7 @@ '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, @@ -44,12 +45,14 @@ '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, '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, diff --git a/requirements.txt b/requirements.txt index 05f13ad76..152e6a5b5 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 @@ -10,10 +10,10 @@ 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 +openai==1.52.1 pytest==7.4.0 PyGithub==1.59.* PyYAML==6.0.1