From d9f37011a2243a66f04762b96ff90a1493023001 Mon Sep 17 00:00:00 2001
From: Cesar Garcia <128240629+Chesars@users.noreply.github.com>
Date: Wed, 4 Mar 2026 19:20:03 -0300
Subject: [PATCH] Update
 litellm/llms/black_forest_labs/image_generation/transformation.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
---
 .../llms/black_forest_labs/image_generation/transformation.py    | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/llms/black_forest_labs/image_generation/transformation.py b/litellm/llms/black_forest_labs/image_generation/transformation.py
index 7a97adafe80..88ad9a57c18 100644
--- a/litellm/llms/black_forest_labs/image_generation/transformation.py
+++ b/litellm/llms/black_forest_labs/image_generation/transformation.py
@@ -250,6 +250,7 @@ class BlackForestLabsImageGenerationConfig(BaseImageGenerationConfig):
         raw_response: httpx.Response,
         model_response: ImageResponse,
         logging_obj: LiteLLMLoggingObj,
+        **kwargs,
     ) -> ImageResponse:
         """
         Transform Black Forest Labs response to OpenAI-compatible ImageResponse.