From e126e98a7fd1ca21bf3c7581ecdd4977dccfc2e1 Mon Sep 17 00:00:00 2001
From: Sameer Kankute <sameer@berri.ai>
Date: Mon, 19 Jan 2026 19:10:43 +0530
Subject: [PATCH] Fix azure image mypy issues

---
 litellm/llms/azure_ai/image_edit/flux2_transformation.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/litellm/llms/azure_ai/image_edit/flux2_transformation.py b/litellm/llms/azure_ai/image_edit/flux2_transformation.py
index 87bae59ba0..77d46ff917 100644
--- a/litellm/llms/azure_ai/image_edit/flux2_transformation.py
+++ b/litellm/llms/azure_ai/image_edit/flux2_transformation.py
@@ -88,7 +88,7 @@ class AzureFoundryFlux2ImageEditConfig(OpenAIImageEditConfig):
         self,
         model: str,
         prompt: Optional[str],
-        image: FileTypes,
+        image: Optional[FileTypes],
         image_edit_optional_request_params: Dict,
         litellm_params: GenericLiteLLMParams,
         headers: dict,
@@ -102,6 +102,9 @@ class AzureFoundryFlux2ImageEditConfig(OpenAIImageEditConfig):
         if prompt is None:
             raise ValueError("FLUX 2 image edit requires a prompt.")
         
+        if image is None:
+            raise ValueError("FLUX 2 image edit requires an image.")
+        
         image_b64 = self._convert_image_to_base64(image)
 
         # Build request body with required params