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":