awslabs.cdk-mcp-server 0.0.71717__tar.gz → 0.0.81004__tar.gz

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.cdk-mcp-server
-Version: 0.0.71717
+Version: 0.0.81004
 Summary: An AWS CDK MCP server that provides guidance on AWS Cloud Development Kit best practices, infrastructure as code patterns, and security compliance with CDK Nag. This server offers tools to validate infrastructure designs, explain CDK Nag rules, analyze suppressions, generate Bedrock Agent schemas, and discover Solutions Constructs patterns.
 Requires-Python: >=3.10
 Requires-Dist: aws-lambda-powertools>=2.30.0
@@ -41,10 +41,16 @@ MCP server for AWS Cloud Development Kit (CDK) best practices, infrastructure as
 
 ### Amazon Bedrock Agent Schema Generation
 
-- Generate OpenAPI schema for Bedrock Agent Action Groups
+- Use this tool when creating Bedrock Agents with Action Groups that use Lambda functions
 - Streamline the creation of Bedrock Agent schemas
 - Convert code files to compatible OpenAPI specifications
 
+#### Developer Notes
+
+- **Requirements**: Your Lambda function must use `BedrockAgentResolver` from AWS Lambda Powertools
+- **Lambda Dependencies**: If schema generation fails, a fallback script will be generated. If you see error messages about missing dependencies, install them and then run the script again.
+- **Integration**: Use the generated schema with `bedrock.ApiSchema.fromLocalAsset()` in your CDK code
+
 ## CDK Implementation Workflow
 
 This diagram provides a comprehensive view of the recommended CDK implementation workflow:
@@ -69,7 +75,8 @@ graph TD
     BA --> BS["GenerateBedrockAgentSchema"]
     BS -->|"Success"| JSON["openapi.json created"]
     BS -->|"Import Errors"| BSF["Tool generates<br/>generate_schema.py"]
-    BSF --> BSR["Run script manually:<br/>python generate_schema.py"]
+    BSF -->|"Missing dependencies?"| InstallDeps["Install dependencies"]
+    InstallDeps --> BSR["Run script manually:<br/>python generate_schema.py"]
     BSR --> JSON["openapi.json created"]
     
     %% Use schema in Agent CDK

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/README.md

@@ -30,10 +30,16 @@ MCP server for AWS Cloud Development Kit (CDK) best practices, infrastructure as
 
 ### Amazon Bedrock Agent Schema Generation
 
-- Generate OpenAPI schema for Bedrock Agent Action Groups
+- Use this tool when creating Bedrock Agents with Action Groups that use Lambda functions
 - Streamline the creation of Bedrock Agent schemas
 - Convert code files to compatible OpenAPI specifications
 
+#### Developer Notes
+
+- **Requirements**: Your Lambda function must use `BedrockAgentResolver` from AWS Lambda Powertools
+- **Lambda Dependencies**: If schema generation fails, a fallback script will be generated. If you see error messages about missing dependencies, install them and then run the script again.
+- **Integration**: Use the generated schema with `bedrock.ApiSchema.fromLocalAsset()` in your CDK code
+
 ## CDK Implementation Workflow
 
 This diagram provides a comprehensive view of the recommended CDK implementation workflow:
@@ -58,7 +64,8 @@ graph TD
     BA --> BS["GenerateBedrockAgentSchema"]
     BS -->|"Success"| JSON["openapi.json created"]
     BS -->|"Import Errors"| BSF["Tool generates<br/>generate_schema.py"]
-    BSF --> BSR["Run script manually:<br/>python generate_schema.py"]
+    BSF -->|"Missing dependencies?"| InstallDeps["Install dependencies"]
+    InstallDeps --> BSR["Run script manually:<br/>python generate_schema.py"]
     BSR --> JSON["openapi.json created"]
     
     %% Use schema in Agent CDK

awslabs_cdk_mcp_server-0.0.81004/awslabs/__init__.py

@@ -0,0 +1,10 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.

awslabs_cdk_mcp_server-0.0.81004/awslabs/cdk_mcp_server/__init__.py

@@ -0,0 +1,18 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+"""AWS CDK MCP server package."""
+
+# Import the minimal set of essential functions
+from awslabs.cdk_mcp_server.core.server import main, mcp
+
+__all__ = ['main', 'mcp']
+
+__version__ = '0.0.0'

awslabs_cdk_mcp_server-0.0.81004/awslabs/cdk_mcp_server/core/__init__.py

@@ -0,0 +1,11 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+"""Core modules for the AWS CDK MCP server."""

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/core/resources.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """AWS CDK MCP resource handlers."""
 
 import logging

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/core/search_utils.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """Common search utilities for AWS CDK MCP Server."""
 
 import re

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/core/server.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """AWS CDK MCP server implementation."""
 
 import argparse

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/core/tools.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """AWS CDK MCP tool handlers."""
 
 import logging
@@ -230,6 +241,38 @@ async def bedrock_schema_generator_from_file(
         output_path=output_path,
     )
 
+    # Add comprehensive next steps for successful schema generation
+    if result.get('status') == 'success':
+        output_filename = os.path.basename(output_path)
+        output_dir = os.path.dirname(output_path)
+        lambda_dir = os.path.dirname(os.path.abspath(lambda_code_path))
+        lambda_name = os.path.basename(os.path.dirname(lambda_code_path))
+
+        # Create a more comprehensive integration example
+        result['next_steps'] = {
+            'success_message': f'Schema successfully generated and saved to {output_path}',
+            'integration_steps': [
+                '1. Ensure your Lambda function has the right permissions:',
+                '   - Add bedrock.amazonaws.com as a principal in permissions',
+                '   - Include Lambda Powertools and Pydantic as layers',
+                '2. Add the ActionGroup to your Bedrock Agent:',
+                '   - Create an action group with your Lambda as the executor',
+                '   - Use the generated schema with ApiSchema.fromLocalAsset()',
+                '3. Deploy your CDK stack',
+            ],
+            'cdk_example': [
+                '// Add the Action Group to your agent',
+                'agent.addActionGroup(new bedrock.AgentActionGroup({',
+                f"  name: '{lambda_name}-action-group',",
+                f"  description: 'Action group for {lambda_name}',",
+                '  executor: bedrock.ActionGroupExecutor.fromlambdaFunction(yourLambdaFunction),',
+                '  apiSchema: bedrock.ApiSchema.fromLocalAsset(',
+                f"    path.join(__dirname, '{os.path.relpath(output_dir, lambda_dir)}', '{output_filename}')",
+                '  )',
+                '}));',
+            ],
+        }
+
     # If fallback script was generated, save it to a file instead of returning it in the response
     if result.get('status') == 'error' and result.get('fallback_script'):
         # Save the script to a file

awslabs_cdk_mcp_server-0.0.81004/awslabs/cdk_mcp_server/data/__init__.py

@@ -0,0 +1,11 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+"""Data modules for the AWS CDK MCP server."""

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/data/cdk_nag_parser.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """CDK Nag rules parsing utilities."""
 
 import httpx

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/data/construct_descriptions.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """GenAI CDK construct descriptions."""
 
 from typing import Dict

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/data/genai_cdk_loader.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """GenAI CDK constructs static content loader."""
 
 import logging

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/data/lambda_powertools_loader.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """Lambda Powertools guidance loader module."""
 
 import os

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/data/schema_generator.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """Schema generator for Bedrock Agent Action Groups."""
 
 import importlib.util
@@ -121,12 +132,13 @@ def main():
 
             # Generate the OpenAPI schema
             print("Generating OpenAPI schema...")
+            # Note: This might show a UserWarning about Pydantic v2 and OpenAPI versions
             openapi_schema = json.loads(app.get_openapi_json_schema(openapi_version="3.0.0"))
 
             # Fix Pydantic v2 issue (forcing OpenAPI 3.0.0)
             if openapi_schema.get("openapi") != "3.0.0":
                 openapi_schema["openapi"] = "3.0.0"
-                print("Fixed OpenAPI version to 3.0.0 (Pydantic v2 issue)")
+                print("Note: Adjusted OpenAPI version for compatibility with Bedrock Agents")
 
             # Fix operationIds
             for path in openapi_schema['paths']:
@@ -158,12 +170,19 @@ def main():
                 json.dump(openapi_schema, f, indent=2)
 
             print(f"Schema successfully generated and saved to {{OUTPUT_PATH}}")
+            print("Next steps: Use this schema in your CDK code with bedrock.ApiSchema.fromLocalAsset()")
             return True
 
         except Exception as simplified_error:
             print(f"Error with simplified version: {{str(simplified_error)}}")
-            print("You may need to manually modify the script to handle this error.")
-            print("Focus on preserving the BedrockAgentResolver app definition and routes.")
+            if "No module named" in str(simplified_error):
+                missing_dep = str(simplified_error).split("'")[-2] if "'" in str(simplified_error) else str(simplified_error).split("No module named ")[-1].strip()
+                print("To resolve this error, install the missing dependency:")
+                print("    pip install " + missing_dep.replace('_', '-'))
+                print("Then run this script again.")
+            else:
+                print("You may need to manually modify the script to handle this error.")
+                print("Focus on preserving the BedrockAgentResolver app definition and routes.")
             return False
 
     except Exception as e:

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/awslabs/cdk_mcp_server/data/solutions_constructs_parser.py

@@ -1,3 +1,14 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
 """AWS Solutions Constructs patterns parser module."""
 
 import httpx

awslabs_cdk_mcp_server-0.0.81004/awslabs/cdk_mcp_server/server.py

@@ -0,0 +1,18 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+
+"""AWS CDK MCP server implementation."""
+
+from awslabs.cdk_mcp_server.core.server import main
+
+
+if __name__ == '__main__':
+    main()

awslabs_cdk_mcp_server-0.0.81004/awslabs/cdk_mcp_server/static/__init__.py

@@ -0,0 +1,19 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+# with the License. A copy of the License is located at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions
+# and limitations under the License.
+from importlib import resources
+
+
+with (
+    resources.files('awslabs.cdk_mcp_server.static')
+    .joinpath('CDK_GENERAL_GUIDANCE.md')
+    .open('r') as f
+):
+    CDK_GENERAL_GUIDANCE = f.read()

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "awslabs.cdk-mcp-server"
-version = "0.0.071717"
+version = "0.0.081004"
 description = "An AWS CDK MCP server that provides guidance on AWS Cloud Development Kit best practices, infrastructure as code patterns, and security compliance with CDK Nag. This server offers tools to validate infrastructure designs, explain CDK Nag rules, analyze suppressions, generate Bedrock Agent schemas, and discover Solutions Constructs patterns."
 readme = "README.md"
 requires-python = ">=3.10"

{awslabs_cdk_mcp_server-0.0.71717 → awslabs_cdk_mcp_server-0.0.81004}/uv.lock

@@ -50,7 +50,7 @@ wheels = [
 
 [[package]]
 name = "awslabs-cdk-mcp-server"
-version = "0.0.31004"
+version = "0.0.62303"
 source = { editable = "." }
 dependencies = [
     { name = "aws-lambda-powertools" },

awslabs_cdk_mcp_server-0.0.71717/awslabs/__init__.py

@@ -1,2 +0,0 @@
-# This file is part of the awslabs namespace.
-# It is intentionally minimal to support PEP 420 namespace packages.

awslabs_cdk_mcp_server-0.0.71717/awslabs/cdk_mcp_server/__init__.py

@@ -1,8 +0,0 @@
-"""AWS CDK MCP server package."""
-
-# Import the minimal set of essential functions
-from awslabs.cdk_mcp_server.core.server import main, mcp
-
-__all__ = ['main', 'mcp']
-
-__version__ = '0.0.0'

awslabs_cdk_mcp_server-0.0.71717/awslabs/cdk_mcp_server/core/__init__.py

@@ -1 +0,0 @@
-"""Core modules for the AWS CDK MCP server."""

awslabs_cdk_mcp_server-0.0.71717/awslabs/cdk_mcp_server/data/__init__.py

@@ -1 +0,0 @@
-"""Data modules for the AWS CDK MCP server."""

awslabs_cdk_mcp_server-0.0.71717/awslabs/cdk_mcp_server/server.py

@@ -1,7 +0,0 @@
-"""AWS CDK MCP server implementation."""
-
-from awslabs.cdk_mcp_server.core.server import main
-
-
-if __name__ == '__main__':
-    main()

awslabs_cdk_mcp_server-0.0.71717/awslabs/cdk_mcp_server/static/__init__.py

@@ -1,9 +0,0 @@
-from importlib import resources
-
-
-with (
-    resources.files('awslabs.cdk_mcp_server.static')
-    .joinpath('CDK_GENERAL_GUIDANCE.md')
-    .open('r') as f
-):
-    CDK_GENERAL_GUIDANCE = f.read()