From 315a483aceb31729c416f158bb59dc60bcde6ae3 Mon Sep 17 00:00:00 2001 From: Cesar Garcia <128240629+Chesars@users.noreply.github.com> Date: Wed, 4 Mar 2026 22:32:30 -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> --- .../image_generation/transformation.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/litellm/llms/black_forest_labs/image_generation/transformation.py b/litellm/llms/black_forest_labs/image_generation/transformation.py index 76267491e3e..1ad866fac6d 100644 --- a/litellm/llms/black_forest_labs/image_generation/transformation.py +++ b/litellm/llms/black_forest_labs/image_generation/transformation.py @@ -57,11 +57,19 @@ class BlackForestLabsImageGenerationConfig(BaseImageGenerationConfig): Return list of OpenAI params supported by Black Forest Labs. Note: BFL uses different parameter names, these are mapped in map_openai_params. - """ return [ "n", # Number of images (BFL returns 1 per request, but ultra supports up to 4) "size", # Maps to width/height or aspect_ratio "quality", # Maps to raw mode for ultra + "seed", + "output_format", + "safety_tolerance", + "prompt_upsampling", + "raw", + "num_images", + "image_url", + "image_prompt_strength", + "aspect_ratio", ] def map_openai_params(