ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2026-01-03 21:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1051 from KennyDizi/main

Browse source 
Add support for gpt-4o-mini and gpt-4o-mini-2024-07-18 models
This commit is contained in:
Tal 2024-07-23 09:23:56 +03:00 committed by GitHub
commit 32604d8103
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pr_agent/algo/__init__.py
View file

@ -16,6 +16,8 @@ MAX_TOKENS = {
'gpt-4-turbo-preview': 128000, # 128K, but may be limited by config.max_model_tokens
'gpt-4-turbo-2024-04-09': 128000, # 128K, but may be limited by config.max_model_tokens
'gpt-4-turbo': 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
'claude-instant-1': 100000,
'claude-2': 100000,
'command-nightly': 4096,