diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index f8dcd7fc..dd6febfd 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -87,7 +87,8 @@ MAX_TOKENS = { USER_MESSAGE_ONLY_MODELS = [ "deepseek/deepseek-reasoner", "o1-mini", - "o1-mini-2024-09-12" + "o1-mini-2024-09-12", + "o1-preview" ] NO_SUPPORT_TEMPERATURE_MODELS = [ @@ -98,4 +99,5 @@ NO_SUPPORT_TEMPERATURE_MODELS = [ "o1-2024-12-17", "o3-mini", "o3-mini-2025-01-31", + "o1-preview" ]