diff --git a/pr_agent/algo/ai_handlers/litellm_ai_handler.py b/pr_agent/algo/ai_handlers/litellm_ai_handler.py index 453973f6..50d14b21 100644 --- a/pr_agent/algo/ai_handlers/litellm_ai_handler.py +++ b/pr_agent/algo/ai_handlers/litellm_ai_handler.py @@ -200,10 +200,10 @@ class LiteLLMAIHandler(BaseAiHandler): # Currently, model OpenAI o1 series does not support a separate system and user prompts O1_MODEL_PREFIX = 'o1' model_type = model.split('/')[-1] if '/' in model else model - if model_type.startswith(O1_MODEL_PREFIX): + if (model_type.startswith(O1_MODEL_PREFIX)) or ("deepseek-reasoner" in model): user = f"{system}\n\n\n{user}" system = "" - get_logger().info(f"Using O1 model, combining system and user prompts") + get_logger().info(f"Using model {model}, combining system and user prompts") messages = [{"role": "user", "content": user}] kwargs = { "model": model,