diff --git a/litellm/llms/xai/image_edit/transformation.py b/litellm/llms/xai/image_edit/transformation.py index 4ba55f6c339..9ef0b51dca7 100644 --- a/litellm/llms/xai/image_edit/transformation.py +++ b/litellm/llms/xai/image_edit/transformation.py @@ -86,12 +86,7 @@ class XAIImageEditConfig(BaseImageEditConfig): resolved_base: Final = ( XAIOAuthAuthenticator().get_api_base() if should_use_xai_oauth(litellm_params) and not XAIModelInfo.get_api_key(api_key) - else ( - api_base - or get_secret_str("XAI_API_BASE") - or get_secret_str("XAI_OAUTH_API_BASE") - or XAI_API_BASE - ) + else (api_base or get_secret_str("XAI_API_BASE") or get_secret_str("XAI_OAUTH_API_BASE") or XAI_API_BASE) ) base: Final = (resolved_base or XAI_API_BASE).rstrip("/") if base.endswith("/v1"): @@ -129,8 +124,7 @@ class XAIImageEditConfig(BaseImageEditConfig): model=model, llm_provider="xai", message=( - "Missing xAI credentials for image edit. " - "Pass api_key / XAI_API_KEY, or set use_xai_oauth=True." + "Missing xAI credentials for image edit. Pass api_key / XAI_API_KEY, or set use_xai_oauth=True." ), ) headers["Authorization"] = f"Bearer {dynamic_api_key}" @@ -159,11 +153,7 @@ class XAIImageEditConfig(BaseImageEditConfig): request: Final[dict[str, Any]] = { "model": XAIModelInfo.get_base_model(model) or model, **({"prompt": prompt} if prompt is not None else {}), - **( - {"image": image_payloads[0]} - if len(image_payloads) == 1 - else {"images": list(image_payloads)} - ), + **({"image": image_payloads[0]} if len(image_payloads) == 1 else {"images": list(image_payloads)}), **{ key: image_edit_optional_request_params[key] for key in ("aspect_ratio", "resolution") diff --git a/litellm/llms/xai/image_generation/transformation.py b/litellm/llms/xai/image_generation/transformation.py index 6dd8318a139..4e220e82386 100644 --- a/litellm/llms/xai/image_generation/transformation.py +++ b/litellm/llms/xai/image_generation/transformation.py @@ -49,11 +49,7 @@ class XAIImageGenerationConfig(BaseImageGenerationConfig): ) -> dict: supported_params: Final = frozenset(self.get_supported_openai_params(model)) allowed: Final = supported_params | _XAI_NATIVE_PARAMS - unknown: Final = tuple( - key - for key in non_default_params - if key not in optional_params and key not in allowed - ) + unknown: Final = tuple(key for key in non_default_params if key not in optional_params and key not in allowed) if unknown and not drop_params: raise ValueError( f"Parameter {unknown[0]} is not supported for model {model}. " @@ -86,12 +82,7 @@ class XAIImageGenerationConfig(BaseImageGenerationConfig): resolved_base: Final = ( XAIOAuthAuthenticator().get_api_base() if should_use_xai_oauth(litellm_params) and not XAIModelInfo.get_api_key(api_key) - else ( - api_base - or get_secret_str("XAI_API_BASE") - or get_secret_str("XAI_OAUTH_API_BASE") - or XAI_API_BASE - ) + else (api_base or get_secret_str("XAI_API_BASE") or get_secret_str("XAI_OAUTH_API_BASE") or XAI_API_BASE) ) base: Final = (resolved_base or XAI_API_BASE).rstrip("/") if base.endswith("/v1"): diff --git a/litellm/proxy/video_endpoints/endpoints.py b/litellm/proxy/video_endpoints/endpoints.py index 6f13d88341f..6da6301aac5 100644 --- a/litellm/proxy/video_endpoints/endpoints.py +++ b/litellm/proxy/video_endpoints/endpoints.py @@ -269,9 +269,7 @@ async def video_status( if resolved_model: data["model"] = resolved_model - custom_llm_provider: Final = ( - explicit_provider or infer_video_provider_from_model(resolved_model) or "openai" - ) + custom_llm_provider: Final = explicit_provider or infer_video_provider_from_model(resolved_model) or "openai" if custom_llm_provider: data["custom_llm_provider"] = custom_llm_provider diff --git a/litellm/videos/main.py b/litellm/videos/main.py index b95225d04c5..c01d1343c91 100644 --- a/litellm/videos/main.py +++ b/litellm/videos/main.py @@ -362,9 +362,7 @@ def video_content( litellm_call_id: Final[str | None] = kwargs.get("litellm_call_id", None) _is_async: Final = kwargs.pop("async_call", False) is True - custom_llm_provider = _provider_for_video_id( - video_id, custom_llm_provider, kwargs.get("model") - ) + custom_llm_provider = _provider_for_video_id(video_id, custom_llm_provider, kwargs.get("model")) # get llm provider logic litellm_params: Final = GenericLiteLLMParams(**kwargs) @@ -456,9 +454,7 @@ async def avideo_content( loop: Final = asyncio.get_event_loop() kwargs["async_call"] = True - custom_llm_provider = _provider_for_video_id( - video_id, custom_llm_provider, kwargs.get("model") - ) + custom_llm_provider = _provider_for_video_id(video_id, custom_llm_provider, kwargs.get("model")) func: Final = partial( video_content, @@ -1062,9 +1058,7 @@ def video_status( response: Final = VideoObject(**mock_response) return response - custom_llm_provider = _provider_for_video_id( - video_id, custom_llm_provider, kwargs.get("model") - ) + custom_llm_provider = _provider_for_video_id(video_id, custom_llm_provider, kwargs.get("model")) # get llm provider logic litellm_params: Final = GenericLiteLLMParams(**kwargs)