diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 83d001a1..2a9b1f34 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -31,6 +31,7 @@ MAX_TOKENS = { '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,11 +49,13 @@ MAX_TOKENS = { '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, '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, diff --git a/requirements.txt b/requirements.txt index ec88a4ca..afb6af68 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 @@ -12,17 +12,17 @@ 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 +openai==1.54.1 pytest==7.4.0 PyGithub==1.59.* 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