diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 0ba7aa84..ca533001 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -17,6 +17,7 @@ MAX_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,