xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-14 17:50:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: prevent LITELLM.EXTRA_BODY from overriding existing parameters in LiteLLMAIHandler

Browse Source
This commit is contained in:
Abhinav Kumar
2025-07-07 21:20:25 +05:30
parent 12af211c13
commit 6a55bbcd23
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -365,11 +365,14 @@ class LiteLLMAIHandler(BaseAiHandler):
kwargs["extra_headers"] = litellm_extra_headers kwargs["extra_headers"] = litellm_extra_headers
# Support for custom OpenAI body fields (e.g., Flex Processing) # Support for custom OpenAI body fields (e.g., Flex Processing)
if get_settings().get("LITELLM.EXTRA_BODY", None): if get_settings().litellm.extra_body:
try: try:
litellm_extra_body = json.loads(get_settings().litellm.extra_body) litellm_extra_body = json.loads(get_settings().litellm.extra_body)
if not isinstance(litellm_extra_body, dict): if not isinstance(litellm_extra_body, dict):
raise ValueError("LITELLM.EXTRA_BODY must be a JSON object") raise ValueError("LITELLM.EXTRA_BODY must be a JSON object")
colliding_keys = kwargs.keys() & litellm_extra_body.keys()
if colliding_keys:
raise ValueError(f"LITELLM.EXTRA_BODY cannot override existing parameters: {', '.join(colliding_keys)}")
kwargs.update(litellm_extra_body) kwargs.update(litellm_extra_body)
except json.JSONDecodeError as e: except json.JSONDecodeError as e:
raise ValueError(f"LITELLM.EXTRA_BODY contains invalid JSON: {str(e)}") raise ValueError(f"LITELLM.EXTRA_BODY contains invalid JSON: {str(e)}")