ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-12 02:45:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
Show stats 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")