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

feat(pr_agent): add support for gpt-4-turbo-preview model and update default settings

Browse Source
This commit is contained in:
Yuta Nishi
2024-03-26 14:47:05 +09:00
parent 9c284e64cf
commit d064a352ad
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pr_agent/algo/__init__.py
View File

@ -10,6 +10,7 @@ MAX_TOKENS = {
'gpt-4-32k': 32000, 'gpt-4-32k': 32000,
'gpt-4-1106-preview': 128000, # 128K, but may be limited by config.max_model_tokens 'gpt-4-1106-preview': 128000, # 128K, but may be limited by config.max_model_tokens
'gpt-4-0125-preview': 128000, # 128K, but may be limited by config.max_model_tokens 'gpt-4-0125-preview': 128000, # 128K, but may be limited by config.max_model_tokens
'gpt-4-turbo-preview': 128000, # 128K, but may be limited by config.max_model_tokens
'claude-instant-1': 100000, 'claude-instant-1': 100000,
'claude-2': 100000, 'claude-2': 100000,
'command-nightly': 4096, 'command-nightly': 4096,

4
pr_agent/settings/configuration.toml
View File

@ -1,6 +1,6 @@
[config] [config]
model="gpt-4" # "gpt-4-0125-preview" model="gpt-4" # "gpt-4-turbo-preview"
model_turbo="gpt-4-0125-preview" model_turbo="gpt-4-turbo-preview"
fallback_models=["gpt-3.5-turbo-16k"] fallback_models=["gpt-3.5-turbo-16k"]
git_provider="github" git_provider="github"
publish_output=true publish_output=true