xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-03 04:10:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add a null check for user_message_only_models before using it

Browse Source
This commit is contained in:
Trung Dinh
2025-01-31 11:53:05 +07:00
parent 7eb26b3220
commit 48377e3c81
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View File

@ -198,8 +198,8 @@ class LiteLLMAIHandler(BaseAiHandler):
{"type": "image_url", "image_url": {"url": img_path}}]
# Currently, some models do not support a separate system and user prompts
user_message_only_models = get_settings().config.user_message_only_models
if user_message_only_models and any(entry in model for entry in user_message_only_models):
user_message_only_models = get_settings().config.user_message_only_models or []
if user_message_only_models and any(entry.lower() in model.lower() for entry in user_message_only_models):
user = f"{system}\n\n\n{user}"
system = ""
get_logger().info(f"Using model {model}, combining system and user prompts")