awslabs.nova-canvas-mcp-server 0.1.10652__py3-none-any.whl → 0.1.31004__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.10652.dist-info → awslabs_nova_canvas_mcp_server-0.1.31004.dist-info}/METADATA

@@ -1,15 +1,15 @@
 Metadata-Version: 2.4
 Name: awslabs.nova-canvas-mcp-server
-Version: 0.1.10652
+Version: 0.1.31004
 Summary: An AWS Labs Model Context Protocol (MCP) server for Amazon Nova Canvas
-Requires-Python: >=3.13
+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
 
-# Nova Canvas MCP Server
+# Amazon Nova Canvas MCP Server
 
 MCP server for generating images using Amazon Nova Canvas
 
@@ -31,7 +31,7 @@ MCP server for generating images using Amazon Nova Canvas
 ## 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`
+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
@@ -39,9 +39,7 @@ MCP server for generating images using Amazon 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`):
+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
 {
@@ -50,8 +48,9 @@ Add the server to your MCP client config (e.g. for Amazon Q CLI MCP, `~/.aws/ama
       "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
+        "AWS_PROFILE": "your-aws-profile",
+        "AWS_REGION": "us-east-1",
+        "FASTMCP_LOG_LEVEL": "ERROR"
       },
       "disabled": false,
       "autoApprove": []
@@ -66,8 +65,8 @@ The MCP server uses the AWS profile specified in the `AWS_PROFILE` environment v
 
 ```json
 "env": {
-  "AWS_PROFILE": "your-aws-profile",  // Specify which AWS profile to use
-  "AWS_REGION": "us-east-1"           // Region where Bedrock is available
+  "AWS_PROFILE": "your-aws-profile",
+  "AWS_REGION": "us-east-1"
 }
 ```
 

awslabs_nova_canvas_mcp_server-0.1.31004.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.31004.dist-info/METADATA,sha256=ffG9LiNQ-eUWnr0fBCoUHNV8Dtkim_gWvM368BPuGv4,2932
+awslabs_nova_canvas_mcp_server-0.1.31004.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+awslabs_nova_canvas_mcp_server-0.1.31004.dist-info/entry_points.txt,sha256=v8V4vn8YuugOSL7w_sUxz-M0EDZNZU2_ydJZDd31pGI,94
+awslabs_nova_canvas_mcp_server-0.1.31004.dist-info/RECORD,,

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,,