From ea64cd5d151eef8fc103cc711c867236b238a0d6 Mon Sep 17 00:00:00 2001 From: mateo-berri <277851410+mateo-berri@users.noreply.github.com> Date: Fri, 17 Jul 2026 14:23:41 -0400 Subject: [PATCH] fix(together_ai): pass legacy standard image params through verbatim --- .../together_ai/image_generation/transformation.py | 2 +- ...st_together_ai_image_generation_transformation.py | 12 +++++++++++- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/litellm/llms/together_ai/image_generation/transformation.py b/litellm/llms/together_ai/image_generation/transformation.py index bf2d3bd5c31..5465fec7dbb 100644 --- a/litellm/llms/together_ai/image_generation/transformation.py +++ b/litellm/llms/together_ai/image_generation/transformation.py @@ -29,7 +29,7 @@ else: class TogetherAIImageGenerationConfig(BaseImageGenerationConfig): def get_supported_openai_params(self, model: str) -> list[OpenAIImageGenerationOptionalParams]: - return ["n", "size", "response_format"] + return ["n", "size", "response_format", "quality", "style", "user"] def map_openai_params( self, diff --git a/tests/test_litellm/llms/together_ai/image_generation/test_together_ai_image_generation_transformation.py b/tests/test_litellm/llms/together_ai/image_generation/test_together_ai_image_generation_transformation.py index 66326a65ea0..ce54af45fdc 100644 --- a/tests/test_litellm/llms/together_ai/image_generation/test_together_ai_image_generation_transformation.py +++ b/tests/test_litellm/llms/together_ai/image_generation/test_together_ai_image_generation_transformation.py @@ -1,4 +1,3 @@ -import json from typing import Optional import httpx @@ -68,6 +67,17 @@ def test_map_openai_params_maps_size_to_width_height(): assert mapped == {"width": 1792, "height": 1024, "n": 2} +def test_map_openai_params_passes_unmapped_supported_params_verbatim(): + config = TogetherAIImageGenerationConfig() + mapped = config.map_openai_params( + non_default_params={"quality": "hd", "style": "vivid", "user": "user-123"}, + optional_params={}, + model="black-forest-labs/FLUX.1-kontext-pro", + drop_params=False, + ) + assert mapped == {"quality": "hd", "style": "vivid", "user": "user-123"} + + def test_map_openai_params_maps_b64_json_response_format_to_base64(): config = TogetherAIImageGenerationConfig() mapped = config.map_openai_params(