Update litellm/llms/openrouter/image_edit/transformation.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Cesar Garcia 2026-03-03 15:16:12 -03:00 committed by GitHub
parent f6264a9c0f
commit fecb301684
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -133,12 +133,11 @@ class OpenRouterImageEditConfig(BaseImageEditConfig):
api_base: Optional[str],
litellm_params: dict,
) -> str:
if api_base:
if not api_base.endswith("/chat/completions"):
api_base = api_base.rstrip("/")
return f"{api_base}/chat/completions"
return api_base
return "https://openrouter.ai/api/v1/chat/completions"
base_url = api_base or get_secret_str("OPENROUTER_API_BASE") or "https://openrouter.ai/api/v1"
base_url = base_url.rstrip("/")
if not base_url.endswith("/chat/completions"):
return f"{base_url}/chat/completions"
return base_url
def transform_image_edit_request(
self,