scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

style: apply ruff format to xai image and video files

Browse source
This commit is contained in:
hx 2026-09-16 10:00:57 +08:00
parent eb763350f5
commit d6bb450932
4 changed files with 9 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/llms/xai/image_edit/transformation.py
View file

@ -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")

13
litellm/llms/xai/image_generation/transformation.py
View file

@ -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"):

4
litellm/proxy/video_endpoints/endpoints.py
View file

@ -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

12
litellm/videos/main.py
View file

@ -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)