diff --git a/litellm/llms/xai/image_edit/transformation.py b/litellm/llms/xai/image_edit/transformation.py
index c82c810442e..15762d76391 100644
--- a/litellm/llms/xai/image_edit/transformation.py
+++ b/litellm/llms/xai/image_edit/transformation.py
@@ -153,7 +153,7 @@ class XAIImageEditConfig(BaseImageEditConfig):
             raise ValueError("xAI image edit requires at least one reference image.")
 
         n: Final = image_edit_optional_request_params.get("n")
-        request: Final[dict[str, Any]] = {
+        request: Final[dict[str, object]] = {
             "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)}),
diff --git a/litellm/llms/xai/image_generation/transformation.py b/litellm/llms/xai/image_generation/transformation.py
index 62dd7a7f707..327f7891ae2 100644
--- a/litellm/llms/xai/image_generation/transformation.py
+++ b/litellm/llms/xai/image_generation/transformation.py
@@ -17,6 +17,7 @@ from litellm.types.utils import ImageObject, ImageResponse
 
 if TYPE_CHECKING:
     import tiktoken
+
     from litellm.litellm_core_utils.litellm_logging import Logging as LiteLLMLoggingObj
 
 _SIZE_TO_ASPECT_RATIO: Final = {