Fix all mypy issues

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

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,

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,

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}],

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,

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,

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,

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}

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",

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 = [
             {