awslabs.nova-canvas-mcp-server 0.1.10652__py3-none-any.whl → 0.1.62303__py3-none-any.whl

awslabs/nova_canvas_mcp_server/__init__.py

@@ -1,3 +1,3 @@
 """awslabs.nova-canvas-mcp-server"""
 
-__version__ = "0.1.6"
+__version__ = '0.1.6'

awslabs/nova_canvas_mcp_server/consts.py

@@ -1,11 +1,11 @@
 # Constants
-NOVA_CANVAS_MODEL_ID = "amazon.nova-canvas-v1:0"
+NOVA_CANVAS_MODEL_ID = 'amazon.nova-canvas-v1:0'
 DEFAULT_WIDTH = 1024
 DEFAULT_HEIGHT = 1024
-DEFAULT_QUALITY = "standard"
+DEFAULT_QUALITY = 'standard'
 DEFAULT_CFG_SCALE = 6.5
 DEFAULT_NUMBER_OF_IMAGES = 1
-DEFAULT_OUTPUT_DIR = "output"  # Default directory inside workspace_dir
+DEFAULT_OUTPUT_DIR = 'output'  # Default directory inside workspace_dir
 
 
 # Nova Canvas Prompt Best Practices

awslabs/nova_canvas_mcp_server/models.py

@@ -15,8 +15,8 @@ class Quality(str, Enum):
         PREMIUM: Premium quality image generation with enhanced details.
     """
 
-    STANDARD = "standard"
-    PREMIUM = "premium"
+    STANDARD = 'standard'
+    PREMIUM = 'premium'
 
 
 class TaskType(str, Enum):
@@ -27,8 +27,8 @@ class TaskType(str, Enum):
         COLOR_GUIDED_GENERATION: Generate an image guided by both text and color palette.
     """
 
-    TEXT_IMAGE = "TEXT_IMAGE"
-    COLOR_GUIDED_GENERATION = "COLOR_GUIDED_GENERATION"
+    TEXT_IMAGE = 'TEXT_IMAGE'
+    COLOR_GUIDED_GENERATION = 'COLOR_GUIDED_GENERATION'
 
 
 class ImageGenerationConfig(BaseModel):
@@ -50,12 +50,10 @@ class ImageGenerationConfig(BaseModel):
     height: int = Field(default=1024, ge=320, le=4096)
     quality: Quality = Quality.STANDARD
     cfgScale: float = Field(default=6.5, ge=1.1, le=10.0)
-    seed: int = Field(
-        default_factory=lambda: random.randint(0, 858993459), ge=0, le=858993459
-    )
+    seed: int = Field(default_factory=lambda: random.randint(0, 858993459), ge=0, le=858993459)
     numberOfImages: int = Field(default=1, ge=1, le=5)
 
-    @field_validator("width", "height")
+    @field_validator('width', 'height')
     @classmethod
     def must_be_divisible_by_16(cls, v: int) -> int:
         """Validate that width and height are divisible by 16.
@@ -70,10 +68,10 @@ class ImageGenerationConfig(BaseModel):
             ValueError: If the value is not divisible by 16.
         """
         if v % 16 != 0:
-            raise ValueError("Value must be divisible by 16")
+            raise ValueError('Value must be divisible by 16')
         return v
 
-    @model_validator(mode="after")
+    @model_validator(mode='after')
     def validate_aspect_ratio_and_total_pixels(self):
         """Validate aspect ratio and total pixel count.
 
@@ -93,12 +91,12 @@ class ImageGenerationConfig(BaseModel):
         # Check aspect ratio between 1:4 and 4:1
         aspect_ratio = width / height
         if aspect_ratio < 0.25 or aspect_ratio > 4.0:
-            raise ValueError("Aspect ratio must be between 1:4 and 4:1")
+            raise ValueError('Aspect ratio must be between 1:4 and 4:1')
 
         # Check total pixel count
         total_pixels = width * height
         if total_pixels >= 4194304:
-            raise ValueError("Total pixel count must be less than 4,194,304")
+            raise ValueError('Total pixel count must be less than 4,194,304')
 
         return self
 
@@ -132,7 +130,7 @@ class ColorGuidedGenerationParams(BaseModel):
     text: str = Field(..., min_length=1, max_length=1024)
     negativeText: Optional[str] = Field(default=None, min_length=1, max_length=1024)
 
-    @field_validator("colors")
+    @field_validator('colors')
     @classmethod
     def validate_hex_colors(cls, v: List[str]) -> List[str]:
         """Validate that colors are in the correct hexadecimal format.
@@ -146,7 +144,7 @@ class ColorGuidedGenerationParams(BaseModel):
         Raises:
             ValueError: If any color is not a valid hexadecimal color in the format '#RRGGBB'.
         """
-        hex_pattern = re.compile(r"^#[0-9A-Fa-f]{6}$")
+        hex_pattern = re.compile(r'^#[0-9A-Fa-f]{6}$')
         for color in v:
             if not hex_pattern.match(color):
                 raise ValueError(
@@ -182,13 +180,13 @@ class TextImageRequest(BaseModel):
         """
         text_to_image_params = self.textToImageParams.model_dump()
         # Remove negativeText if it's None
-        if text_to_image_params.get("negativeText") is None:
-            text_to_image_params.pop("negativeText", None)
+        if text_to_image_params.get('negativeText') is None:
+            text_to_image_params.pop('negativeText', None)
 
         return {
-            "taskType": self.taskType,
-            "textToImageParams": text_to_image_params,
-            "imageGenerationConfig": self.imageGenerationConfig.model_dump()
+            'taskType': self.taskType,
+            'textToImageParams': text_to_image_params,
+            'imageGenerationConfig': self.imageGenerationConfig.model_dump()
             if self.imageGenerationConfig
             else None,
         }
@@ -206,9 +204,7 @@ class ColorGuidedRequest(BaseModel):
         imageGenerationConfig: Configuration for image generation.
     """
 
-    taskType: Literal[
-        TaskType.COLOR_GUIDED_GENERATION
-    ] = TaskType.COLOR_GUIDED_GENERATION
+    taskType: Literal[TaskType.COLOR_GUIDED_GENERATION] = TaskType.COLOR_GUIDED_GENERATION
     colorGuidedGenerationParams: ColorGuidedGenerationParams
     imageGenerationConfig: Optional[ImageGenerationConfig] = Field(
         default_factory=ImageGenerationConfig
@@ -223,13 +219,13 @@ class ColorGuidedRequest(BaseModel):
         """
         color_guided_params = self.colorGuidedGenerationParams.model_dump()
         # Remove negativeText if it's None
-        if color_guided_params.get("negativeText") is None:
-            color_guided_params.pop("negativeText", None)
+        if color_guided_params.get('negativeText') is None:
+            color_guided_params.pop('negativeText', None)
 
         return {
-            "taskType": self.taskType,
-            "colorGuidedGenerationParams": color_guided_params,
-            "imageGenerationConfig": self.imageGenerationConfig.model_dump()
+            'taskType': self.taskType,
+            'colorGuidedGenerationParams': color_guided_params,
+            'imageGenerationConfig': self.imageGenerationConfig.model_dump()
             if self.imageGenerationConfig
             else None,
         }

awslabs/nova_canvas_mcp_server/novacanvas.py

@@ -41,7 +41,7 @@ def save_generated_images(
     base64_images: List[str],
     filename: Optional[str] = None,
     number_of_images: int = DEFAULT_NUMBER_OF_IMAGES,
-    prefix: str = "nova_canvas",
+    prefix: str = 'nova_canvas',
     workspace_dir: Optional[str] = None,
 ) -> Dict[str, List]:
     """Save base64-encoded images to files.
@@ -56,7 +56,7 @@ def save_generated_images(
     Returns:
         Dictionary with lists of paths to the saved image files and PIL Image objects.
     """
-    logger.debug(f"Saving {len(base64_images)} images")
+    logger.debug(f'Saving {len(base64_images)} images')
     # Determine the output directory
     if workspace_dir:
         output_dir = os.path.join(workspace_dir, DEFAULT_OUTPUT_DIR)
@@ -73,27 +73,25 @@ def save_generated_images(
         # Generate filename if not provided
         if filename:
             image_filename = (
-                f"{filename}_{i + 1}.png" if number_of_images > 1 else f"{filename}.png"
+                f'{filename}_{i + 1}.png' if number_of_images > 1 else f'{filename}.png'
             )
         else:
             # Generate a random filename
-            random_id = "".join(
-                random.choices("abcdefghijklmnopqrstuvwxyz0123456789", k=8)
-            )
-            image_filename = f"{prefix}_{random_id}_{i + 1}.png"
+            random_id = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789', k=8))
+            image_filename = f'{prefix}_{random_id}_{i + 1}.png'
 
         # Decode the base64 image data
         image_data = base64.b64decode(base64_image_data)
 
         # Save the image
         image_path = os.path.join(output_dir, image_filename)
-        with open(image_path, "wb") as file:
+        with open(image_path, 'wb') as file:
             file.write(image_data)
         # Convert to absolute path
         abs_image_path = os.path.abspath(image_path)
         saved_paths.append(abs_image_path)
 
-    return {"paths": saved_paths}
+    return {'paths': saved_paths}
 
 
 async def invoke_nova_canvas(
@@ -112,24 +110,22 @@ async def invoke_nova_canvas(
     Raises:
         Exception: If the API call fails.
     """
-    logger.debug("Invoking Nova Canvas API")
+    logger.debug('Invoking Nova Canvas API')
 
     # Convert the request payload to JSON
     request = json.dumps(request_model_dict)
 
     try:
         # Invoke the model
-        logger.info(f"Sending request to Nova Canvas model: {NOVA_CANVAS_MODEL_ID}")
-        response = bedrock_runtime_client.invoke_model(
-            modelId=NOVA_CANVAS_MODEL_ID, body=request
-        )
+        logger.info(f'Sending request to Nova Canvas model: {NOVA_CANVAS_MODEL_ID}')
+        response = bedrock_runtime_client.invoke_model(modelId=NOVA_CANVAS_MODEL_ID, body=request)
 
         # Decode the response body
-        result = json.loads(response["body"].read())
-        logger.info("Nova Canvas API call successful")
+        result = json.loads(response['body'].read())
+        logger.info('Nova Canvas API call successful')
         return result
     except Exception as e:
-        logger.error(f"Nova Canvas API call failed: {str(e)}")
+        logger.error(f'Nova Canvas API call failed: {str(e)}')
         raise
 
 
@@ -169,22 +165,18 @@ async def generate_image_with_text(
         ImageGenerationResponse: An object containing the paths to the generated images,
         PIL Image objects, and status information.
     """
-    logger.debug(
-        f"Generating text-to-image with prompt: '{prompt[:30]}...' ({width}x{height})"
-    )
+    logger.debug(f"Generating text-to-image with prompt: '{prompt[:30]}...' ({width}x{height})")
 
     try:
         # Validate input parameters using Pydantic
         try:
-            logger.debug("Validating parameters and creating request model")
+            logger.debug('Validating parameters and creating request model')
 
             # Create image generation config
             config = ImageGenerationConfig(
                 width=width,
                 height=height,
-                quality=Quality.STANDARD
-                if quality == DEFAULT_QUALITY
-                else Quality.PREMIUM,
+                quality=Quality.STANDARD if quality == DEFAULT_QUALITY else Quality.PREMIUM,
                 cfgScale=cfg_scale,
                 seed=seed if seed is not None else random.randint(0, 858993459),
                 numberOfImages=number_of_images,
@@ -193,9 +185,7 @@ async def generate_image_with_text(
             # Create text-to-image params
             # The Nova Canvas API doesn't accept null for negativeText
             if negative_prompt is not None:
-                text_params = TextToImageParams(
-                    text=prompt, negativeText=negative_prompt
-                )
+                text_params = TextToImageParams(text=prompt, negativeText=negative_prompt)
             else:
                 text_params = TextToImageParams(text=prompt)
 
@@ -206,13 +196,13 @@ async def generate_image_with_text(
 
             # Convert model to dictionary
             request_model_dict = request_model.to_api_dict()
-            logger.info("Request validation successful")
+            logger.info('Request validation successful')
 
         except Exception as e:
-            logger.error(f"Parameter validation failed: {str(e)}")
+            logger.error(f'Parameter validation failed: {str(e)}')
             return ImageGenerationResponse(
-                status="error",
-                message=f"Validation error: {str(e)}",
+                status='error',
+                message=f'Validation error: {str(e)}',
                 paths=[],
                 prompt=prompt,
                 negative_prompt=negative_prompt,
@@ -220,36 +210,34 @@ async def generate_image_with_text(
 
         try:
             # Invoke the Nova Canvas API
-            logger.debug("Sending request to Nova Canvas API")
-            model_response = await invoke_nova_canvas(
-                request_model_dict, bedrock_runtime_client
-            )
+            logger.debug('Sending request to Nova Canvas API')
+            model_response = await invoke_nova_canvas(request_model_dict, bedrock_runtime_client)
 
             # Extract the image data
-            base64_images = model_response["images"]
-            logger.info(f"Received {len(base64_images)} images from Nova Canvas API")
+            base64_images = model_response['images']
+            logger.info(f'Received {len(base64_images)} images from Nova Canvas API')
 
             # Save the generated images
             result = save_generated_images(
                 base64_images,
                 filename,
                 number_of_images,
-                prefix="nova_canvas",
+                prefix='nova_canvas',
                 workspace_dir=workspace_dir,
             )
 
             logger.info(f'Successfully generated {len(result["paths"])} image(s)')
             return ImageGenerationResponse(
-                status="success",
+                status='success',
                 message=f'Generated {len(result["paths"])} image(s)',
-                paths=result["paths"],
+                paths=result['paths'],
                 prompt=prompt,
                 negative_prompt=negative_prompt,
             )
         except Exception as e:
-            logger.error(f"Image generation failed: {str(e)}")
+            logger.error(f'Image generation failed: {str(e)}')
             return ImageGenerationResponse(
-                status="error",
+                status='error',
                 message=str(e),
                 paths=[],
                 prompt=prompt,
@@ -257,9 +245,9 @@ async def generate_image_with_text(
             )
 
     except Exception as e:
-        logger.error(f"Unexpected error in generate_image_with_text: {str(e)}")
+        logger.error(f'Unexpected error in generate_image_with_text: {str(e)}')
         return ImageGenerationResponse(
-            status="error",
+            status='error',
             message=str(e),
             paths=[],
             prompt=prompt,
@@ -312,17 +300,13 @@ async def generate_image_with_colors(
     try:
         # Validate input parameters using Pydantic
         try:
-            logger.debug(
-                "Validating parameters and creating color-guided request model"
-            )
+            logger.debug('Validating parameters and creating color-guided request model')
 
             # Create image generation config
             config = ImageGenerationConfig(
                 width=width,
                 height=height,
-                quality=Quality.STANDARD
-                if quality == DEFAULT_QUALITY
-                else Quality.PREMIUM,
+                quality=Quality.STANDARD if quality == DEFAULT_QUALITY else Quality.PREMIUM,
                 cfgScale=cfg_scale,
                 seed=seed if seed is not None else random.randint(0, 858993459),
                 numberOfImages=number_of_images,
@@ -349,13 +333,13 @@ async def generate_image_with_colors(
 
             # Convert model to dictionary
             request_model_dict = request_model.to_api_dict()
-            logger.info("Color-guided request validation successful")
+            logger.info('Color-guided request validation successful')
 
         except Exception as e:
-            logger.error(f"Color-guided parameter validation failed: {str(e)}")
+            logger.error(f'Color-guided parameter validation failed: {str(e)}')
             return ImageGenerationResponse(
-                status="error",
-                message=f"Validation error: {str(e)}",
+                status='error',
+                message=f'Validation error: {str(e)}',
                 paths=[],
                 prompt=prompt,
                 negative_prompt=negative_prompt,
@@ -364,39 +348,35 @@ async def generate_image_with_colors(
 
         try:
             # Invoke the Nova Canvas API
-            logger.debug("Sending color-guided request to Nova Canvas API")
-            model_response = await invoke_nova_canvas(
-                request_model_dict, bedrock_runtime_client
-            )
+            logger.debug('Sending color-guided request to Nova Canvas API')
+            model_response = await invoke_nova_canvas(request_model_dict, bedrock_runtime_client)
 
             # Extract the image data
-            base64_images = model_response["images"]
-            logger.info(f"Received {len(base64_images)} images from Nova Canvas API")
+            base64_images = model_response['images']
+            logger.info(f'Received {len(base64_images)} images from Nova Canvas API')
 
             # Save the generated images
             result = save_generated_images(
                 base64_images,
                 filename,
                 number_of_images,
-                prefix="nova_canvas_color",
+                prefix='nova_canvas_color',
                 workspace_dir=workspace_dir,
             )
 
-            logger.info(
-                f'Successfully generated {len(result["paths"])} color-guided image(s)'
-            )
+            logger.info(f'Successfully generated {len(result["paths"])} color-guided image(s)')
             return ImageGenerationResponse(
-                status="success",
+                status='success',
                 message=f'Generated {len(result["paths"])} image(s)',
-                paths=result["paths"],
+                paths=result['paths'],
                 prompt=prompt,
                 negative_prompt=negative_prompt,
                 colors=colors,
             )
         except Exception as e:
-            logger.error(f"Color-guided image generation failed: {str(e)}")
+            logger.error(f'Color-guided image generation failed: {str(e)}')
             return ImageGenerationResponse(
-                status="error",
+                status='error',
                 message=str(e),
                 paths=[],
                 prompt=prompt,
@@ -405,9 +385,9 @@ async def generate_image_with_colors(
             )
 
     except Exception as e:
-        logger.error(f"Unexpected error in generate_image_with_colors: {str(e)}")
+        logger.error(f'Unexpected error in generate_image_with_colors: {str(e)}')
         return ImageGenerationResponse(
-            status="error",
+            status='error',
             message=str(e),
             paths=[],
             prompt=prompt,

awslabs/nova_canvas_mcp_server/server.py

@@ -25,7 +25,7 @@ from typing import TYPE_CHECKING, List, Optional
 
 # Logging
 logger.remove()
-logger.add(sys.stderr, level=os.getenv("FASTMCP_LOG_LEVEL", "WARNING"))
+logger.add(sys.stderr, level=os.getenv('FASTMCP_LOG_LEVEL', 'WARNING'))
 
 # Bedrock Runtime Client typing
 if TYPE_CHECKING:
@@ -36,25 +36,23 @@ else:
 
 # Bedrock Runtime Client
 bedrock_runtime_client: BedrockRuntimeClient
-aws_region: str = os.environ.get("AWS_REGION", "us-east-1")
+aws_region: str = os.environ.get('AWS_REGION', 'us-east-1')
 
 try:
-    if aws_profile := os.environ.get("AWS_PROFILE"):
+    if aws_profile := os.environ.get('AWS_PROFILE'):
         bedrock_runtime_client = boto3.Session(
             profile_name=aws_profile, region_name=aws_region
-        ).client("bedrock-runtime")
+        ).client('bedrock-runtime')
     else:
-        bedrock_runtime_client = boto3.Session(region_name=aws_region).client(
-            "bedrock-runtime"
-        )
+        bedrock_runtime_client = boto3.Session(region_name=aws_region).client('bedrock-runtime')
 except Exception as e:
-    logger.error(f"Error creating bedrock runtime client: {str(e)}")
+    logger.error(f'Error creating bedrock runtime client: {str(e)}')
     raise
 
 
 # Create the MCP server Pwith detailed instructions
 mcp = FastMCP(
-    "awslabs-nova-canvas-mcp-server",
+    'awslabs-nova-canvas-mcp-server',
     instructions=f"""
 # Amazon Nova Canvas Image Generation
 
@@ -73,33 +71,33 @@ Generate an image from a text prompt and color palette using Amazon Nova Canvas.
 {PROMPT_INSTRUCTIONS}
 """,
     dependencies=[
-        "pydantic",
-        "boto3",
+        'pydantic',
+        'boto3',
     ],
 )
 
 
-@mcp.tool(name="generate_image")
+@mcp.tool(name='generate_image')
 async def mcp_generate_image(
     ctx: Context,
     prompt: str = Field(
-        description="The text description of the image to generate (1-1024 characters)"
+        description='The text description of the image to generate (1-1024 characters)'
     ),
     negative_prompt: Optional[str] = Field(
         default=None,
-        description="Text to define what not to include in the image (1-1024 characters)",
+        description='Text to define what not to include in the image (1-1024 characters)',
     ),
     filename: Optional[str] = Field(
         default=None,
-        description="The name of the file to save the image to (without extension)",
+        description='The name of the file to save the image to (without extension)',
     ),
     width: int = Field(
         default=DEFAULT_WIDTH,
-        description="The width of the generated image (320-4096, divisible by 16)",
+        description='The width of the generated image (320-4096, divisible by 16)',
     ),
     height: int = Field(
         default=DEFAULT_HEIGHT,
-        description="The height of the generated image (320-4096, divisible by 16)",
+        description='The height of the generated image (320-4096, divisible by 16)',
     ),
     quality: str = Field(
         default=DEFAULT_QUALITY,
@@ -107,14 +105,12 @@ async def mcp_generate_image(
     ),
     cfg_scale: float = Field(
         default=DEFAULT_CFG_SCALE,
-        description="How strongly the image adheres to the prompt (1.1-10.0)",
-    ),
-    seed: Optional[int] = Field(
-        default=None, description="Seed for generation (0-858,993,459)"
+        description='How strongly the image adheres to the prompt (1.1-10.0)',
     ),
+    seed: Optional[int] = Field(default=None, description='Seed for generation (0-858,993,459)'),
     number_of_images: int = Field(
         default=DEFAULT_NUMBER_OF_IMAGES,
-        description="The number of images to generate (1-5)",
+        description='The number of images to generate (1-5)',
     ),
     workspace_dir: Optional[str] = Field(
         default=None,
@@ -161,7 +157,7 @@ async def mcp_generate_image(
 
     try:
         logger.info(
-            f"Generating image with text prompt, quality: {quality}, cfg_scale: {cfg_scale}"
+            f'Generating image with text prompt, quality: {quality}, cfg_scale: {cfg_scale}'
         )
         response = await generate_image_with_text(
             prompt=prompt,
@@ -177,62 +173,58 @@ async def mcp_generate_image(
             workspace_dir=workspace_dir,
         )
 
-        if response.status == "success":
+        if response.status == 'success':
             # return response.paths
             return McpImageGenerationResponse(
-                status="success",
-                paths=[f"file://{path}" for path in response.paths],
+                status='success',
+                paths=[f'file://{path}' for path in response.paths],
             )
         else:
-            logger.error(f"Image generation returned error status: {response.message}")
-            await ctx.error(f"Failed to generate image: {response.message}")  # type: ignore
+            logger.error(f'Image generation returned error status: {response.message}')
+            await ctx.error(f'Failed to generate image: {response.message}')  # type: ignore
             # Return empty image or raise exception based on requirements
-            raise Exception(f"Failed to generate image: {response.message}")
+            raise Exception(f'Failed to generate image: {response.message}')
     except Exception as e:
-        logger.error(f"Error in mcp_generate_image: {str(e)}")
-        await ctx.error(f"Error generating image: {str(e)}")  # type: ignore
+        logger.error(f'Error in mcp_generate_image: {str(e)}')
+        await ctx.error(f'Error generating image: {str(e)}')  # type: ignore
         raise
 
 
-@mcp.tool(name="generate_image_with_colors")
+@mcp.tool(name='generate_image_with_colors')
 async def mcp_generate_image_with_colors(
     ctx: Context,
     prompt: str = Field(
-        description="The text description of the image to generate (1-1024 characters)"
+        description='The text description of the image to generate (1-1024 characters)'
     ),
     colors: List[str] = Field(
         description='List of up to 10 hexadecimal color values (e.g., "#FF9800")'
     ),
     negative_prompt: Optional[str] = Field(
         default=None,
-        description="Text to define what not to include in the image (1-1024 characters)",
+        description='Text to define what not to include in the image (1-1024 characters)',
     ),
     filename: Optional[str] = Field(
         default=None,
-        description="The name of the file to save the image to (without extension)",
+        description='The name of the file to save the image to (without extension)',
     ),
     width: int = Field(
         default=1024,
-        description="The width of the generated image (320-4096, divisible by 16)",
+        description='The width of the generated image (320-4096, divisible by 16)',
     ),
     height: int = Field(
         default=1024,
-        description="The height of the generated image (320-4096, divisible by 16)",
+        description='The height of the generated image (320-4096, divisible by 16)',
     ),
     quality: str = Field(
-        default="standard",
+        default='standard',
         description='The quality of the generated image ("standard" or "premium")',
     ),
     cfg_scale: float = Field(
         default=6.5,
-        description="How strongly the image adheres to the prompt (1.1-10.0)",
-    ),
-    seed: Optional[int] = Field(
-        default=None, description="Seed for generation (0-858,993,459)"
-    ),
-    number_of_images: int = Field(
-        default=1, description="The number of images to generate (1-5)"
+        description='How strongly the image adheres to the prompt (1.1-10.0)',
     ),
+    seed: Optional[int] = Field(default=None, description='Seed for generation (0-858,993,459)'),
+    number_of_images: int = Field(default=1, description='The number of images to generate (1-5)'),
     workspace_dir: Optional[str] = Field(
         default=None,
         description="The current workspace directory where the image should be saved. CRITICAL: Assistant must always provide this parameter to save images to the user's current project.",
@@ -274,9 +266,9 @@ async def mcp_generate_image_with_colors(
     )
 
     try:
-        color_hex_list = ", ".join(colors[:3]) + (", ..." if len(colors) > 3 else "")
+        color_hex_list = ', '.join(colors[:3]) + (', ...' if len(colors) > 3 else '')
         logger.info(
-            f"Generating color-guided image with colors: [{color_hex_list}], quality: {quality}"
+            f'Generating color-guided image with colors: [{color_hex_list}], quality: {quality}'
         )
 
         response = await generate_image_with_colors(
@@ -294,51 +286,45 @@ async def mcp_generate_image_with_colors(
             workspace_dir=workspace_dir,
         )
 
-        if response.status == "success":
+        if response.status == 'success':
             return McpImageGenerationResponse(
-                status="success",
-                paths=[f"file://{path}" for path in response.paths],
+                status='success',
+                paths=[f'file://{path}' for path in response.paths],
             )
         else:
             logger.error(
-                f"Color-guided image generation returned error status: {response.message}"
-            )
-            await ctx.error(
-                f"Failed to generate color-guided image: {response.message}"
-            )
-            raise Exception(
-                f"Failed to generate color-guided image: {response.message}"
+                f'Color-guided image generation returned error status: {response.message}'
             )
+            await ctx.error(f'Failed to generate color-guided image: {response.message}')
+            raise Exception(f'Failed to generate color-guided image: {response.message}')
     except Exception as e:
-        logger.error(f"Error in mcp_generate_image_with_colors: {str(e)}")
-        await ctx.error(f"Error generating color-guided image: {str(e)}")
+        logger.error(f'Error in mcp_generate_image_with_colors: {str(e)}')
+        await ctx.error(f'Error generating color-guided image: {str(e)}')
         raise
 
 
 def main():
     """Run the MCP server with CLI argument support."""
-    logger.info("Starting nova-canvas-mcp-server MCP server")
+    logger.info('Starting nova-canvas-mcp-server MCP server')
 
     parser = argparse.ArgumentParser(
-        description="MCP server for generating images using Amazon Nova Canvas"
-    )
-    parser.add_argument("--sse", action="store_true", help="Use SSE transport")
-    parser.add_argument(
-        "--port", type=int, default=8888, help="Port to run the server on"
+        description='MCP server for generating images using Amazon Nova Canvas'
     )
+    parser.add_argument('--sse', action='store_true', help='Use SSE transport')
+    parser.add_argument('--port', type=int, default=8888, help='Port to run the server on')
 
     args = parser.parse_args()
-    logger.debug(f"Parsed arguments: sse={args.sse}, port={args.port}")
+    logger.debug(f'Parsed arguments: sse={args.sse}, port={args.port}')
 
     # Run server with appropriate transport
     if args.sse:
-        logger.info(f"Using SSE transport on port {args.port}")
+        logger.info(f'Using SSE transport on port {args.port}')
         mcp.settings.port = args.port
-        mcp.run(transport="sse")
+        mcp.run(transport='sse')
     else:
-        logger.info("Using standard stdio transport")
+        logger.info('Using standard stdio transport')
         mcp.run()
 
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     main()

awslabs_nova_canvas_mcp_server-0.1.62303.dist-info/METADATA

@@ -0,0 +1,91 @@
+Metadata-Version: 2.4
+Name: awslabs.nova-canvas-mcp-server
+Version: 0.1.62303
+Summary: An AWS Labs Model Context Protocol (MCP) server for Amazon Nova Canvas
+Requires-Python: >=3.10
+Requires-Dist: boto3>=1.37.24
+Requires-Dist: loguru>=0.7.3
+Requires-Dist: mcp[cli]>=1.6.0
+Requires-Dist: pydantic>=2.11.1
+Description-Content-Type: text/markdown
+
+# Amazon Nova Canvas MCP Server
+
+[![smithery badge](https://smithery.ai/badge/@awslabs/nova-canvas-mcp-server)](https://smithery.ai/server/@awslabs/nova-canvas-mcp-server)
+
+MCP server for generating images using Amazon Nova Canvas
+
+## Features
+
+### Text-based image generation
+
+- Create images from text prompts with `generate_image`
+- Customizable dimensions (320-4096px), quality options, and negative prompting
+- Supports multiple image generation (1-5) in single request
+- Adjustable parameters like cfg_scale (1.1-10.0) and seeded generation
+
+### Color-guided image generation
+
+- Generate images with specific color palettes using `generate_image_with_colors`
+- Define up to 10 hex color values to influence the image style and mood
+- Same customization options as text-based generation
+
+### Workspace integration
+
+- Images saved to user-specified workspace directories with automatic folder creation
+
+### AWS authentication
+
+- Uses AWS profiles for secure access to Amazon Nova Canvas services
+
+## Prerequisites
+
+1. Install `uv` from [Astral](https://docs.astral.sh/uv/getting-started/installation/) or the [GitHub README](https://github.com/astral-sh/uv#installation)
+2. Install Python using `uv python install 3.10`
+3. Set up AWS credentials with access to Amazon Bedrock and Nova Canvas
+   - You need an AWS account with Amazon Bedrock and Amazon Nova Canvas enabled
+   - Configure AWS credentials with `aws configure` or environment variables
+   - Ensure your IAM role/user has permissions to use Amazon Bedrock and Nova Canvas
+
+## Installation
+
+Here are some ways you can work with MCP across AWS, and we'll be adding support to more products including Amazon Q Developer CLI soon: (e.g. for Amazon Q Developer CLI MCP, `~/.aws/amazonq/mcp.json`):
+
+```json
+{
+  "mcpServers": {
+    "awslabs.nova-canvas-mcp-server": {
+      "command": "uvx",
+      "args": ["awslabs.nova-canvas-mcp-server@latest"],
+      "env": {
+        "AWS_PROFILE": "your-aws-profile",
+        "AWS_REGION": "us-east-1",
+        "FASTMCP_LOG_LEVEL": "ERROR"
+      },
+      "disabled": false,
+      "autoApprove": []
+    }
+  }
+}
+```
+
+### Installing via Smithery
+
+To install Amazon Nova Canvas MCP Server for Claude Desktop automatically via [Smithery](https://smithery.ai/server/@awslabs/nova-canvas-mcp-server):
+
+```bash
+npx -y @smithery/cli install @awslabs/nova-canvas-mcp-server --client claude
+```
+
+### AWS Authentication
+
+The MCP server uses the AWS profile specified in the `AWS_PROFILE` environment variable. If not provided, it defaults to the "default" profile in your AWS configuration file.
+
+```json
+"env": {
+  "AWS_PROFILE": "your-aws-profile",
+  "AWS_REGION": "us-east-1"
+}
+```
+
+Make sure the AWS profile has permissions to access Amazon Bedrock and Amazon Nova Canvas. The MCP server creates a boto3 session using the specified profile to authenticate with AWS services. Your AWS IAM credentials remain on your local machine and are strictly used for using the Amazon Bedrock model APIs.

awslabs_nova_canvas_mcp_server-0.1.62303.dist-info/RECORD

@@ -0,0 +1,10 @@
+awslabs/__init__.py,sha256=4zfFn3N0BkvQmMTAIvV_QAbKp6GWzrwaUN17YeRoChM,115
+awslabs/nova_canvas_mcp_server/__init__.py,sha256=D1JLDjoqRWgJm28RaKjBtIzAsuc31Ilg8r5-hv8I0ZU,60
+awslabs/nova_canvas_mcp_server/consts.py,sha256=1qnIsWXKsg7R8JpWalgns0vPmBAHu6f9oI8hylhBuuo,2590
+awslabs/nova_canvas_mcp_server/models.py,sha256=tYJeeTKhU_6OJxKJBvKyAzgC46-Dexx_o5up539jqi4,9935
+awslabs/nova_canvas_mcp_server/novacanvas.py,sha256=ltHqnH5slEfq52jF69xE59pam1pEXmIRaGYnWFtfC04,15017
+awslabs/nova_canvas_mcp_server/server.py,sha256=Tdar_AviuKRoBTlWvSEdIMdlz2xuKTxtQMH7y8_OxKA,12326
+awslabs_nova_canvas_mcp_server-0.1.62303.dist-info/METADATA,sha256=k8E6a0HgSzk-HgSNIr2yW_M5FDIQAI1jAQsaNv8pq2c,3328
+awslabs_nova_canvas_mcp_server-0.1.62303.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+awslabs_nova_canvas_mcp_server-0.1.62303.dist-info/entry_points.txt,sha256=v8V4vn8YuugOSL7w_sUxz-M0EDZNZU2_ydJZDd31pGI,94
+awslabs_nova_canvas_mcp_server-0.1.62303.dist-info/RECORD,,

awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/METADATA

@@ -1,74 +0,0 @@
-Metadata-Version: 2.4
-Name: awslabs.nova-canvas-mcp-server
-Version: 0.1.10652
-Summary: An AWS Labs Model Context Protocol (MCP) server for Amazon Nova Canvas
-Requires-Python: >=3.13
-Requires-Dist: boto3>=1.37.24
-Requires-Dist: loguru>=0.7.3
-Requires-Dist: mcp[cli]>=1.6.0
-Requires-Dist: pydantic>=2.11.1
-Description-Content-Type: text/markdown
-
-# Nova Canvas MCP Server
-
-MCP server for generating images using Amazon Nova Canvas
-
-## Features
-
-- **Text-based image generation** - Create images from text prompts with `generate_image`
-  - Customizable dimensions (320-4096px), quality options, and negative prompting
-  - Supports multiple image generation (1-5) in single request
-  - Adjustable parameters like cfg_scale (1.1-10.0) and seeded generation
-
-- **Color-guided image generation** - Generate images with specific color palettes using `generate_image_with_colors`
-  - Define up to 10 hex color values to influence the image style and mood
-  - Same customization options as text-based generation
-
-- **Workspace integration** - Images saved to user-specified workspace directories with automatic folder creation
-
-- **AWS authentication** - Uses AWS profiles for secure access to Amazon Nova Canvas services
-
-## Prerequisites
-
-1. Install `uv` from [Astral](https://docs.astral.sh/uv/getting-started/installation/) or the [GitHub README](https://github.com/astral-sh/uv#installation)
-2. Install Python using `uv python install 3.13`
-3. Set up AWS credentials with access to Amazon Bedrock and Nova Canvas
-   - You need an AWS account with Amazon Bedrock and Amazon Nova Canvas enabled
-   - Configure AWS credentials with `aws configure` or environment variables
-   - Ensure your IAM role/user has permissions to use Amazon Bedrock and Nova Canvas
-
-## Installation
-
-Install the MCP server:
-
-Add the server to your MCP client config (e.g. for Amazon Q CLI MCP, `~/.aws/amazonq/mcp.json`):
-
-```json
-{
-  "mcpServers": {
-    "awslabs.nova-canvas-mcp-server": {
-      "command": "uvx",
-      "args": ["awslabs.nova-canvas-mcp-server@latest"],
-      "env": {
-        "AWS_PROFILE": "your-aws-profile",  // Optional: specify AWS profile
-        "AWS_REGION": "us-east-1"           // Required: region where Bedrock is available
-      },
-      "disabled": false,
-      "autoApprove": []
-    }
-  }
-}
-```
-
-### AWS Authentication
-
-The MCP server uses the AWS profile specified in the `AWS_PROFILE` environment variable. If not provided, it defaults to the "default" profile in your AWS configuration file.
-
-```json
-"env": {
-  "AWS_PROFILE": "your-aws-profile",  // Specify which AWS profile to use
-  "AWS_REGION": "us-east-1"           // Region where Bedrock is available
-}
-```
-
-Make sure the AWS profile has permissions to access Amazon Bedrock and Amazon Nova Canvas. The MCP server creates a boto3 session using the specified profile to authenticate with AWS services. Your AWS IAM credentials remain on your local machine and are strictly used for using the Amazon Bedrock model APIs.

awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/RECORD

@@ -1,10 +0,0 @@
-awslabs/__init__.py,sha256=4zfFn3N0BkvQmMTAIvV_QAbKp6GWzrwaUN17YeRoChM,115
-awslabs/nova_canvas_mcp_server/__init__.py,sha256=W7bzpkX1o3FURvzhQujITA3csYdIPA41UrCBnl99m8A,60
-awslabs/nova_canvas_mcp_server/consts.py,sha256=jWKxZvmHi_5l2IULP5mHJLI3FLTx_4Y_KmC390ySWvA,2590
-awslabs/nova_canvas_mcp_server/models.py,sha256=VxqfLSAfe3MiNWDP61tKu3UUOzgwqE2ko0U5n4Fxz-E,9963
-awslabs/nova_canvas_mcp_server/novacanvas.py,sha256=pLnEB4qmQ3_IO2VgatmTi164g1NEhS11UkTL0uVjiEk,15305
-awslabs/nova_canvas_mcp_server/server.py,sha256=mNN8MHLnpz9ZrGYZFTismDM0pRWsoQ9lknTmaPapgN0,12464
-awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/METADATA,sha256=EtSFRGpmUQrtJgGMmTNdyNRNrVIUbh7VTHPXyytglbI,2981
-awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/entry_points.txt,sha256=v8V4vn8YuugOSL7w_sUxz-M0EDZNZU2_ydJZDd31pGI,94
-awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/RECORD,,