From c30c775e6363de01704c3ad6f744e2ce572397a1 Mon Sep 17 00:00:00 2001 From: Tim Elfrink Date: Thu, 18 Sep 2025 10:32:18 +0200 Subject: [PATCH] Fix Nova Canvas model_id parameter handling for consistency Follow the same pattern as chat completions and embeddings by extracting model_id from optional_params into a variable, even though it's not used in image generation. This maintains code consistency across Bedrock services. --- .../bedrock/image/amazon_nova_canvas_transformation.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/litellm/llms/bedrock/image/amazon_nova_canvas_transformation.py b/litellm/llms/bedrock/image/amazon_nova_canvas_transformation.py index 39f97699d49..cd33e62af16 100644 --- a/litellm/llms/bedrock/image/amazon_nova_canvas_transformation.py +++ b/litellm/llms/bedrock/image/amazon_nova_canvas_transformation.py @@ -7,12 +7,12 @@ from litellm.types.llms.bedrock import ( AmazonNovaCanvasColorGuidedGenerationParams, AmazonNovaCanvasColorGuidedRequest, AmazonNovaCanvasImageGenerationConfig, + AmazonNovaCanvasInpaintingParams, + AmazonNovaCanvasInpaintingRequest, AmazonNovaCanvasRequestBase, AmazonNovaCanvasTextToImageParams, AmazonNovaCanvasTextToImageRequest, AmazonNovaCanvasTextToImageResponse, - AmazonNovaCanvasInpaintingParams, - AmazonNovaCanvasInpaintingRequest, ) from litellm.types.utils import ImageResponse @@ -68,8 +68,9 @@ class AmazonNovaCanvasConfig: task_type = optional_params.pop("taskType", "TEXT_IMAGE") image_generation_config = optional_params.pop("imageGenerationConfig", {}) - # Filter out model_id parameter to prevent "extraneous key" error from Bedrock API - optional_params.pop("model_id", None) + # Extract model_id parameter to prevent "extraneous key" error from Bedrock API + # Following the same pattern as chat completions and embeddings + unencoded_model_id = optional_params.pop("model_id", None) # noqa: F841 image_generation_config = {**image_generation_config, **optional_params} if task_type == "TEXT_IMAGE":