From e6c56c7355acd5ad64fc97d80ecd2d9d8f4f6f3d Mon Sep 17 00:00:00 2001 From: Tal Date: Wed, 9 Oct 2024 08:56:31 +0300 Subject: [PATCH] Update pr_agent/algo/ai_handlers/litellm_ai_handler.py Co-authored-by: codiumai-pr-agent-pro[bot] <151058649+codiumai-pr-agent-pro[bot]@users.noreply.github.com> --- pr_agent/algo/ai_handlers/litellm_ai_handler.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pr_agent/algo/ai_handlers/litellm_ai_handler.py b/pr_agent/algo/ai_handlers/litellm_ai_handler.py index e1587f1d..12bdb4f7 100644 --- a/pr_agent/algo/ai_handlers/litellm_ai_handler.py +++ b/pr_agent/algo/ai_handlers/litellm_ai_handler.py @@ -187,7 +187,8 @@ class LiteLLMAIHandler(BaseAiHandler): {"type": "image_url", "image_url": {"url": img_path}}] # Currently O1 does not support separate system and user prompts - if model.startswith('o1-'): + O1_MODEL_PREFIX = 'o1-' + if model.startswith(O1_MODEL_PREFIX): user = f"{system}\n\n\n{user}" system = "" get_logger().info(f"Using O1 model, combining system and user prompts")