diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 2a9b1f34f..10b50184d 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -19,6 +19,7 @@ '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-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-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 diff --git a/pr_agent/settings/configuration.toml b/pr_agent/settings/configuration.toml index 4d31f9b20..264439b8a 100644 --- a/pr_agent/settings/configuration.toml +++ b/pr_agent/settings/configuration.toml @@ -1,8 +1,8 @@ [config] # models model="gpt-4-turbo-2024-04-09" -model_turbo="gpt-4o-2024-08-06" -fallback_models=["gpt-4o-2024-05-13"] +model_turbo="gpt-4o-2024-11-20" +fallback_models=["gpt-4o-2024-08-06"] # CLI git_provider="github" publish_output=true diff --git a/requirements.txt b/requirements.txt index afb6af682..9b6bc5af9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,7 +12,7 @@ google-cloud-aiplatform==1.38.0 google-generativeai==0.8.3 google-cloud-storage==2.10.0 Jinja2==3.1.2 -litellm==1.52.0 +litellm==1.52.12 loguru==0.7.2 msrest==0.7.1 openai==1.54.1