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): add BFL-specific params to image edit get_supported_openai_params for consistency

Browse source
This commit is contained in:
Chesars 2026-03-04 23:04:48 -03:00
parent f3cb45765b
commit fa165a68d9
2 changed files with 20 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
litellm/llms/black_forest_labs/image_edit/transformation.py
View file

@ -54,7 +54,20 @@ class BlackForestLabsImageEditConfig(BaseImageEditConfig):
Note: BFL uses different parameter names, these are mapped in map_openai_params.
"""
return []
return [
"seed",
"output_format",
"safety_tolerance",
"prompt_upsampling",
"aspect_ratio",
"steps",
"guidance",
"grow_mask",
"top",
"bottom",
"left",
"right",
]
def map_openai_params(
self,

9
tests/test_litellm/llms/black_forest_labs/image_edit/test_bfl_image_edit_transformation.py
View file

@ -46,9 +46,12 @@ class TestBlackForestLabsImageEditTransformation:
"""Test that supported OpenAI params are returned correctly."""
params = self.config.get_supported_openai_params(self.model)
# BFL image edit currently returns an empty list since it uses
# different parameter names mapped in map_openai_params
assert params == []
# BFL image edit supports BFL-specific params passed through directly
assert isinstance(params, list)
assert len(params) > 0
assert "seed" in params
assert "output_format" in params
assert "safety_tolerance" in params
def test_map_openai_params_basic(self):
"""Test mapping of OpenAI params to BFL params."""