scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix(bfl): close docstring in get_supported_openai_params, prevent quality/n param leak for non-ultra models

Browse source
This commit is contained in:
Chesars 2026-03-04 22:36:13 -03:00
parent 315a483ace
commit f3cb45765b
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/black_forest_labs/image_generation/transformation.py
View file

@ -57,6 +57,7 @@ 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
@ -95,13 +96,13 @@ class BlackForestLabsImageGenerationConfig(BaseImageGenerationConfig):
if k == "size" and v:
self._map_size_param(v, optional_params)
elif k == "n":
# BFL uses num_images for ultra model
if "ultra" in model.lower():
optional_params["num_images"] = v
elif k == "quality" and v == "hd":
# Map 'hd' quality to raw mode for more natural look
if "ultra" in model.lower():
# non-ultra: silently skip (n=1 is BFL default)
elif k == "quality":
if v == "hd" and "ultra" in model.lower():
optional_params["raw"] = True
# other quality values have no BFL mapping
else:
optional_params[k] = v
elif not drop_params: