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

Merge pull request #1374 from KennyDizi/main

Browse Source 
Add Support for GPT-4o November 2024 Model and Update Configurations
This commit is contained in:
Tal
2024-12-02 08:23:26 +02:00
committed by GitHub
parent 1d6f87be3b 7d1b6c2f0a
commit b44030114e
3 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pr_agent/algo/__init__.py
View File

@ -19,6 +19,7 @@ MAX_TOKENS = {
'gpt-4o-mini': 128000, # 128K, but may be limited by config.max_model_tokens 'gpt-4o-mini': 128000, # 128K, but may be limited by config.max_model_tokens
'gpt-4o-mini-2024-07-18': 128000, # 128K, but may be limited by config.max_model_tokens 'gpt-4o-mini-2024-07-18': 128000, # 128K, but may be limited by config.max_model_tokens
'gpt-4o-2024-08-06': 128000, # 128K, but may be limited by config.max_model_tokens 'gpt-4o-2024-08-06': 128000, # 128K, but may be limited by config.max_model_tokens
'gpt-4o-2024-11-20': 128000, # 128K, but may be limited by config.max_model_tokens
'o1-mini': 128000, # 128K, but may be limited by config.max_model_tokens 'o1-mini': 128000, # 128K, but may be limited by config.max_model_tokens
'o1-mini-2024-09-12': 128000, # 128K, but may be limited by config.max_model_tokens 'o1-mini-2024-09-12': 128000, # 128K, but may be limited by config.max_model_tokens
'o1-preview': 128000, # 128K, but may be limited by config.max_model_tokens 'o1-preview': 128000, # 128K, but may be limited by config.max_model_tokens

4
pr_agent/settings/configuration.toml
View File

@ -1,8 +1,8 @@
[config] [config]
# models # models
model="gpt-4-turbo-2024-04-09" model="gpt-4-turbo-2024-04-09"
model_turbo="gpt-4o-2024-08-06" model_turbo="gpt-4o-2024-11-20"
fallback_models=["gpt-4o-2024-05-13"] fallback_models=["gpt-4o-2024-08-06"]
# CLI # CLI
git_provider="github" git_provider="github"
publish_output=true publish_output=true

2
requirements.txt
View File

@ -12,7 +12,7 @@ google-cloud-aiplatform==1.38.0
google-generativeai==0.8.3 google-generativeai==0.8.3
google-cloud-storage==2.10.0 google-cloud-storage==2.10.0
Jinja2==3.1.2 Jinja2==3.1.2
litellm==1.52.0 litellm==1.52.12
loguru==0.7.2 loguru==0.7.2
msrest==0.7.1 msrest==0.7.1
openai==1.54.1 openai==1.54.1