xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-01 19:30:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1302 from KennyDizi/main

Browse Source 
Add support for new Claude models and update dependencies
This commit is contained in:
Tal
2024-10-24 08:40:58 +03:00
committed by GitHub
parent 7b5a3d45bd b4700bd7c0
commit c077c71fdb
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pr_agent/algo/__init__.py
View File

@ -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,
@ -44,12 +45,14 @@ 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,
'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,

6
requirements.txt
View File

@ -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