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

fix: logic

Browse source
This commit is contained in:
Abhinav Kumar 2025-07-26 11:40:40 +05:30
parent af2b66bb51
commit a8b8202567
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View file

@ -353,9 +353,10 @@ class LiteLLMAIHandler(BaseAiHandler):
kwargs = _process_litellm_extra_body(kwargs)
# Support for Bedrock custom inference profile via model_id
if get_settings().get("LITELLM.MODEL_ID", None) and 'bedrock/' in model:
kwargs["model_id"] = get_settings().litellm.model_id
get_logger().info(f"Using Bedrock custom inference profile: {get_settings().litellm.model_id}")
model_id = get_settings().get("litellm.model_id")
if model_id and 'bedrock/' in model:
kwargs["model_id"] = model_id
get_logger().info(f"Using Bedrock custom inference profile: {model_id}")
get_logger().debug("Prompts", artifact={"system": system, "user": user})