From ce105abdfbf1fc69feea4cf0d4ae8c2ed2472f9f Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Fri, 16 Jan 2026 17:41:42 +0530 Subject: [PATCH] Fix all mypy issues --- litellm/llms/azure_ai/image_edit/flux2_transformation.py | 5 ++++- litellm/llms/base_llm/image_edit/transformation.py | 2 +- .../llms/bedrock/image_edit/stability_transformation.py | 9 ++++++--- litellm/llms/gemini/image_edit/transformation.py | 5 ++++- litellm/llms/openai/image_edit/dalle2_transformation.py | 7 +++++-- litellm/llms/openai/image_edit/transformation.py | 5 ++++- litellm/llms/recraft/image_edit/transformation.py | 5 ++++- litellm/llms/stability/image_edit/transformations.py | 9 ++++++--- .../vertex_ai/image_edit/vertex_gemini_transformation.py | 5 ++++- .../vertex_ai/image_edit/vertex_imagen_transformation.py | 5 ++++- 10 files changed, 42 insertions(+), 15 deletions(-) diff --git a/litellm/llms/azure_ai/image_edit/flux2_transformation.py b/litellm/llms/azure_ai/image_edit/flux2_transformation.py index caa3905667..87bae59ba0 100644 --- a/litellm/llms/azure_ai/image_edit/flux2_transformation.py +++ b/litellm/llms/azure_ai/image_edit/flux2_transformation.py @@ -87,7 +87,7 @@ class AzureFoundryFlux2ImageEditConfig(OpenAIImageEditConfig): def transform_image_edit_request( self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict, litellm_params: GenericLiteLLMParams, @@ -99,6 +99,9 @@ class AzureFoundryFlux2ImageEditConfig(OpenAIImageEditConfig): FLUX 2 uses the same endpoint for generation and editing, with the image passed as base64 in the JSON body. """ + if prompt is None: + raise ValueError("FLUX 2 image edit requires a prompt.") + image_b64 = self._convert_image_to_base64(image) # Build request body with required params diff --git a/litellm/llms/base_llm/image_edit/transformation.py b/litellm/llms/base_llm/image_edit/transformation.py index d522675296..cc72348037 100644 --- a/litellm/llms/base_llm/image_edit/transformation.py +++ b/litellm/llms/base_llm/image_edit/transformation.py @@ -92,7 +92,7 @@ class BaseImageEditConfig(ABC): def transform_image_edit_request( self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict, litellm_params: GenericLiteLLMParams, diff --git a/litellm/llms/bedrock/image_edit/stability_transformation.py b/litellm/llms/bedrock/image_edit/stability_transformation.py index bcaf0923f6..e8b7781298 100644 --- a/litellm/llms/bedrock/image_edit/stability_transformation.py +++ b/litellm/llms/bedrock/image_edit/stability_transformation.py @@ -21,18 +21,18 @@ Supported models: API Reference: https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters.html """ -import json import base64 +import json from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple import httpx from litellm.llms.base_llm.image_edit.transformation import BaseImageEditConfig from litellm.types.images.main import ImageEditOptionalRequestParams -from litellm.types.router import GenericLiteLLMParams from litellm.types.llms.stability import ( OPENAI_SIZE_TO_STABILITY_ASPECT_RATIO, ) +from litellm.types.router import GenericLiteLLMParams from litellm.types.utils import FileTypes, ImageObject, ImageResponse from litellm.utils import get_model_info @@ -153,7 +153,7 @@ class BedrockStabilityImageEditConfig(BaseImageEditConfig): def transform_image_edit_request( self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict, litellm_params: GenericLiteLLMParams, @@ -164,6 +164,9 @@ class BedrockStabilityImageEditConfig(BaseImageEditConfig): Returns the request body dict that will be JSON-encoded by the handler. """ + if prompt is None: + raise ValueError("Bedrock Stability image edit requires a prompt.") + # Build Bedrock Stability request data: Dict[str, Any] = { "prompt": prompt, diff --git a/litellm/llms/gemini/image_edit/transformation.py b/litellm/llms/gemini/image_edit/transformation.py index 78a7ff9546..0015155b47 100644 --- a/litellm/llms/gemini/image_edit/transformation.py +++ b/litellm/llms/gemini/image_edit/transformation.py @@ -80,7 +80,7 @@ class GeminiImageEditConfig(BaseImageEditConfig): def transform_image_edit_request( # type: ignore[override] self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict[str, Any], litellm_params: GenericLiteLLMParams, @@ -90,6 +90,9 @@ class GeminiImageEditConfig(BaseImageEditConfig): if not inline_parts: raise ValueError("Gemini image edit requires at least one image.") + if prompt is None: + raise ValueError("Gemini image edit requires a prompt.") + contents = [ { "parts": inline_parts + [{"text": prompt}], diff --git a/litellm/llms/openai/image_edit/dalle2_transformation.py b/litellm/llms/openai/image_edit/dalle2_transformation.py index 37e92be17a..13531546d2 100644 --- a/litellm/llms/openai/image_edit/dalle2_transformation.py +++ b/litellm/llms/openai/image_edit/dalle2_transformation.py @@ -1,5 +1,5 @@ from io import BufferedReader -from typing import TYPE_CHECKING, Any, Dict, List, Tuple, cast +from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, cast from httpx._types import RequestFiles @@ -30,7 +30,7 @@ class DallE2ImageEditConfig(OpenAIImageEditConfig): def transform_image_edit_request( self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict, litellm_params: GenericLiteLLMParams, @@ -41,6 +41,9 @@ class DallE2ImageEditConfig(OpenAIImageEditConfig): DALL-E-2 only accepts a single image with field name "image" (not "image[]"). """ + if prompt is None: + raise ValueError("DALL-E-2 image edit requires a prompt.") + request = ImageEditRequestParams( model=model, image=image, diff --git a/litellm/llms/openai/image_edit/transformation.py b/litellm/llms/openai/image_edit/transformation.py index 1b90d96fa9..9edad9ee2c 100644 --- a/litellm/llms/openai/image_edit/transformation.py +++ b/litellm/llms/openai/image_edit/transformation.py @@ -79,7 +79,7 @@ class OpenAIImageEditConfig(BaseImageEditConfig): def transform_image_edit_request( self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict, litellm_params: GenericLiteLLMParams, @@ -91,6 +91,9 @@ class OpenAIImageEditConfig(BaseImageEditConfig): Handles multipart/form-data for images. Uses "image[]" field name to support multiple images (e.g., for gpt-image-1). """ + if prompt is None: + raise ValueError("OpenAI image edit requires a prompt.") + request = ImageEditRequestParams( model=model, image=image, diff --git a/litellm/llms/recraft/image_edit/transformation.py b/litellm/llms/recraft/image_edit/transformation.py index 533a510860..9bf46704ed 100644 --- a/litellm/llms/recraft/image_edit/transformation.py +++ b/litellm/llms/recraft/image_edit/transformation.py @@ -101,7 +101,7 @@ class RecraftImageEditConfig(BaseImageEditConfig): def transform_image_edit_request( self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict, litellm_params: GenericLiteLLMParams, @@ -114,6 +114,9 @@ class RecraftImageEditConfig(BaseImageEditConfig): https://www.recraft.ai/docs#image-to-image """ + if prompt is None: + raise ValueError("Recraft image edit requires a prompt.") + request_body: RecraftImageEditRequestParams = RecraftImageEditRequestParams( model=model, prompt=prompt, diff --git a/litellm/llms/stability/image_edit/transformations.py b/litellm/llms/stability/image_edit/transformations.py index 173fae2d6f..013e3f27a0 100644 --- a/litellm/llms/stability/image_edit/transformations.py +++ b/litellm/llms/stability/image_edit/transformations.py @@ -14,11 +14,11 @@ from httpx._types import RequestFiles from litellm.llms.base_llm.image_edit.transformation import BaseImageEditConfig from litellm.secret_managers.main import get_secret_str from litellm.types.images.main import ImageEditOptionalRequestParams -from litellm.types.router import GenericLiteLLMParams from litellm.types.llms.stability import ( OPENAI_SIZE_TO_STABILITY_ASPECT_RATIO, STABILITY_EDIT_ENDPOINTS, ) +from litellm.types.router import GenericLiteLLMParams from litellm.types.utils import FileTypes, ImageObject, ImageResponse from litellm.utils import get_model_info @@ -170,7 +170,7 @@ class StabilityImageEditConfig(BaseImageEditConfig): def transform_image_edit_request( self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict, litellm_params: GenericLiteLLMParams, @@ -186,9 +186,12 @@ class StabilityImageEditConfig(BaseImageEditConfig): # Populate multipart form-data as separate text fields (data) and files. # Stability expects prompt/output_format/etc. as normal form fields, not file parts. data: Dict[str, Any] = { - "prompt": prompt, "output_format": "png", # Default to PNG } + + # Add prompt only if provided (some Stability endpoints don't require it) + if prompt is not None: + data["prompt"] = prompt # Handle image parameter - could be a single file or list image_file = image[0] if isinstance(image, list) else image # type: ignore files: Dict[str, Any] = {"image": image_file} diff --git a/litellm/llms/vertex_ai/image_edit/vertex_gemini_transformation.py b/litellm/llms/vertex_ai/image_edit/vertex_gemini_transformation.py index 174d05cf7c..154d5669eb 100644 --- a/litellm/llms/vertex_ai/image_edit/vertex_gemini_transformation.py +++ b/litellm/llms/vertex_ai/image_edit/vertex_gemini_transformation.py @@ -151,7 +151,7 @@ class VertexAIGeminiImageEditConfig(BaseImageEditConfig, VertexLLM): def transform_image_edit_request( # type: ignore[override] self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict[str, Any], litellm_params: GenericLiteLLMParams, @@ -161,6 +161,9 @@ class VertexAIGeminiImageEditConfig(BaseImageEditConfig, VertexLLM): if not inline_parts: raise ValueError("Vertex AI Gemini image edit requires at least one image.") + if prompt is None: + raise ValueError("Vertex AI Gemini image edit requires a prompt.") + # Correct format for Vertex AI Gemini image editing contents = { "role": "USER", diff --git a/litellm/llms/vertex_ai/image_edit/vertex_imagen_transformation.py b/litellm/llms/vertex_ai/image_edit/vertex_imagen_transformation.py index 1515e6cbe9..337a4bd4dd 100644 --- a/litellm/llms/vertex_ai/image_edit/vertex_imagen_transformation.py +++ b/litellm/llms/vertex_ai/image_edit/vertex_imagen_transformation.py @@ -143,7 +143,7 @@ class VertexAIImagenImageEditConfig(BaseImageEditConfig, VertexLLM): def transform_image_edit_request( # type: ignore[override] self, model: str, - prompt: str, + prompt: Optional[str], image: FileTypes, image_edit_optional_request_params: Dict[str, Any], litellm_params: GenericLiteLLMParams, @@ -156,6 +156,9 @@ class VertexAIImagenImageEditConfig(BaseImageEditConfig, VertexLLM): if not reference_images: raise ValueError("Vertex AI Imagen image edit requires at least one reference image.") + if prompt is None: + raise ValueError("Vertex AI Imagen image edit requires a prompt.") + # Correct Imagen instances format instances = [ {