awslabs.terraform-mcp-server 0.0.2__py3-none-any.whl → 0.0.7__py3-none-any.whl

awslabs/terraform_mcp_server/impl/tools/__init__.py

@@ -1,5 +1,6 @@
 """Tool implementations for the terraform MCP server."""
 
+from .search_user_provided_module import search_user_provided_module_impl
 from .execute_terraform_command import execute_terraform_command_impl
 from .search_aws_provider_docs import search_aws_provider_docs_impl
 from .search_awscc_provider_docs import search_awscc_provider_docs_impl
@@ -7,6 +8,7 @@ from .search_specific_aws_ia_modules import search_specific_aws_ia_modules_impl
 from .run_checkov_scan import run_checkov_scan_impl
 
 __all__ = [
+    'search_user_provided_module_impl',
     'execute_terraform_command_impl',
     'search_aws_provider_docs_impl',
     'search_awscc_provider_docs_impl',

awslabs/terraform_mcp_server/impl/tools/search_user_provided_module.py

@@ -0,0 +1,346 @@
+# 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.
+"""Implementation of user provided module from the Terraform registry search tool."""
+
+import re
+import requests
+import traceback
+from awslabs.terraform_mcp_server.impl.tools.utils import (
+    clean_description,
+    extract_outputs_from_readme,
+    get_github_release_details,
+    get_variables_tf,
+)
+from awslabs.terraform_mcp_server.models import (
+    SearchUserProvidedModuleRequest,
+    SearchUserProvidedModuleResult,
+    TerraformOutput,
+    TerraformVariable,
+)
+from loguru import logger
+from typing import Any, Dict, Optional, Tuple
+from urllib.parse import urlparse
+
+
+async def search_user_provided_module_impl(
+    request: SearchUserProvidedModuleRequest,
+) -> SearchUserProvidedModuleResult:
+    """Analyze a Terraform module from the registry.
+
+    This tool takes a Terraform registry module URL and analyzes its input variables,
+    output variables, README, and other details to provide comprehensive information
+    about the module.
+
+    Parameters:
+        request: Details about the Terraform module to analyze
+
+    Returns:
+        A SearchUserProvidedModuleResult object containing module information
+    """
+    logger.info(f'Analyzing Terraform module: {request.module_url}')
+
+    try:
+        # Parse the module URL to extract namespace, name, and provider
+        module_parts = parse_module_url(request.module_url)
+        if not module_parts:
+            return SearchUserProvidedModuleResult(
+                status='error',
+                module_name='unknown',
+                module_url=request.module_url,
+                module_version='unknown',
+                module_description='',
+                variables=[],
+                outputs=[],
+                readme_content=None,
+                error_message=f'Invalid module URL format: {request.module_url}. Expected format: [namespace]/[name]/[provider] or registry.terraform.io/[namespace]/[name]/[provider]',
+            )
+
+        namespace, name, provider = module_parts
+
+        # Fetch module details from Terraform Registry
+        module_details = await get_module_details(namespace, name, provider, request.version)
+        if not module_details:
+            return SearchUserProvidedModuleResult(
+                status='error',
+                module_name=name,
+                module_url=request.module_url,
+                module_version=request.version or 'latest',
+                module_description='',
+                variables=[],
+                outputs=[],
+                readme_content=None,
+                error_message=f'Failed to fetch module details from Terraform Registry: {request.module_url}',
+            )
+
+        # Extract module information
+        module_version = module_details.get('version', request.version or 'latest')
+        module_description = clean_description(module_details.get('description', ''))
+        readme_content = module_details.get('readme_content', '')
+
+        # Get variables and outputs
+        variables = []
+        outputs = []
+
+        # Extract variables from module details
+        if 'variables' in module_details and module_details['variables']:
+            variables = [TerraformVariable(**var_data) for var_data in module_details['variables']]
+        elif 'root' in module_details and 'inputs' in module_details['root']:
+            # Extract from registry API format
+            for var_name, var_data in module_details['root']['inputs'].items():
+                variables.append(
+                    TerraformVariable(
+                        name=var_name,
+                        type=var_data.get('type', ''),
+                        description=var_data.get('description', ''),
+                        default=var_data.get('default'),
+                        required=var_data.get('required', True),
+                    )
+                )
+
+        # Extract outputs from module details
+        if 'outputs' in module_details and module_details['outputs']:
+            outputs = [
+                TerraformOutput(name=output['name'], description=output.get('description', ''))
+                for output in module_details['outputs']
+            ]
+        elif 'root' in module_details and 'outputs' in module_details['root']:
+            # Extract from registry API format
+            for output_name, output_data in module_details['root']['outputs'].items():
+                outputs.append(
+                    TerraformOutput(
+                        name=output_name,
+                        description=output_data.get('description', ''),
+                    )
+                )
+        elif readme_content:
+            # Try to extract outputs from README
+            extracted_outputs = extract_outputs_from_readme(readme_content)
+            if extracted_outputs:
+                outputs = [
+                    TerraformOutput(name=output['name'], description=output.get('description', ''))
+                    for output in extracted_outputs
+                ]
+
+        # Create the result
+        result = SearchUserProvidedModuleResult(
+            status='success',
+            module_name=name,
+            module_url=request.module_url,
+            module_version=module_version,
+            module_description=module_description,
+            variables=variables,
+            outputs=outputs,
+            readme_content=readme_content,
+            error_message=None,
+        )
+
+        return result
+
+    except Exception as e:
+        logger.error(f'Error analyzing Terraform module: {e}')
+        logger.debug(f'Stack trace: {traceback.format_exc()}')
+        return SearchUserProvidedModuleResult(
+            status='error',
+            module_name=request.module_url.split('/')[-2]
+            if '/' in request.module_url
+            else 'unknown',
+            module_url=request.module_url,
+            module_version=request.version or 'latest',
+            module_description='',
+            variables=[],
+            outputs=[],
+            readme_content=None,
+            error_message=f'Error analyzing Terraform module: {str(e)}',
+        )
+
+
+def parse_module_url(module_url: str) -> Optional[Tuple[str, str, str]]:
+    """Parse a Terraform module URL to extract namespace, name, and provider.
+
+    Args:
+        module_url: The module URL or identifier (e.g., "hashicorp/consul/aws" or "registry.terraform.io/hashicorp/consul/aws")
+
+    Returns:
+        Tuple containing (namespace, name, provider) or None if invalid format
+    """
+    # First, handle registry.terraform.io URLs (with or without scheme)
+    parsed_url = None
+
+    # If URL has a scheme (http://, https://)
+    if '://' in module_url:
+        parsed_url = urlparse(module_url)
+    # For URLs without scheme, add a dummy scheme to enable proper URL parsing
+    else:
+        parsed_url = urlparse(f'https://{module_url}')
+
+    # Check if this is a registry.terraform.io URL
+    if parsed_url.netloc == 'registry.terraform.io':
+        # Extract path and remove leading slash
+        path = parsed_url.path.lstrip('/')
+        parts = path.split('/')
+    else:
+        # Simple module path format (namespace/name/provider)
+        parts = module_url.split('/')
+
+    # Ensure we have at least namespace/name/provider
+    if len(parts) < 3:
+        return None
+
+    namespace = parts[0]
+    name = parts[1]
+    provider = parts[2]
+
+    return namespace, name, provider
+
+
+async def get_module_details(
+    namespace: str, name: str, provider: str, version: Optional[str] = None
+) -> Dict[str, Any]:
+    """Fetch detailed information about a Terraform module from the registry.
+
+    Args:
+        namespace: The module namespace (e.g., hashicorp)
+        name: The module name (e.g., consul)
+        provider: The provider (e.g., aws)
+        version: Optional specific version to fetch
+
+    Returns:
+        Dictionary containing module details
+    """
+    logger.info(f'Fetching details for module {namespace}/{name}/{provider}')
+
+    try:
+        # Get basic module info via API
+        details_url = f'https://registry.terraform.io/v1/modules/{namespace}/{name}/{provider}'
+        if version:
+            details_url += f'/{version}'
+
+        logger.debug(f'Making API request to: {details_url}')
+
+        response = requests.get(details_url)
+        response.raise_for_status()
+
+        details = response.json()
+        logger.debug(
+            f'Received module details. Status code: {response.status_code}, Content size: {len(response.text)} bytes'
+        )
+
+        # Get the version
+        module_version = version or details.get('version', '')
+        if not module_version and 'latest' in details and 'version' in details['latest']:
+            module_version = details['latest']['version']
+
+        # Try to get README content and version details
+        readme_content = None
+        version_details = None
+
+        # APPROACH 1: Try to see if the registry API provides README content directly
+        logger.debug('Checking for README content in API response')
+        if 'readme' in details and details['readme']:
+            readme_content = details['readme']
+            logger.info(
+                f'Found README content directly in API response: {len(readme_content)} chars'
+            )
+
+        # APPROACH 2: Try using the GitHub repo URL for README content and version details
+        if 'source' in details:
+            source_url = details.get('source')
+            # Validate GitHub URL using regex
+            if isinstance(source_url, str) and re.match(r'https://github.com/', source_url):
+                logger.info(f'Found GitHub source URL: {source_url}')
+
+                # Extract GitHub owner and repo
+                github_parts = re.match(r'https://github.com/([^/]+)/([^/]+)', source_url)
+                if github_parts:
+                    owner, repo = github_parts.groups()
+                    logger.info(f'Extracted GitHub repo: {owner}/{repo}')
+
+                    # Get version details from GitHub
+                    github_version_info = await get_github_release_details(owner, repo)
+                    version_details = github_version_info['details']
+                    version_from_github = github_version_info['version']
+
+                    if version_from_github:
+                        logger.info(f'Found version from GitHub: {version_from_github}')
+                        if not module_version:
+                            module_version = version_from_github
+
+                    # Get variables.tf content and parsed variables
+                    variables_content, variables = await get_variables_tf(owner, repo, 'main')
+                    if variables_content and variables:
+                        logger.info(f'Found variables.tf with {len(variables)} variables')
+                        details['variables_content'] = variables_content
+                        details['variables'] = [var.dict() for var in variables]
+                    else:
+                        # Try master branch as fallback
+                        variables_content, variables = await get_variables_tf(
+                            owner, repo, 'master'
+                        )
+                        if variables_content and variables:
+                            logger.info(
+                                f'Found variables.tf in master branch with {len(variables)} variables'
+                            )
+                            details['variables_content'] = variables_content
+                            details['variables'] = [var.dict() for var in variables]
+
+                    # If README content not already found, try fetching it from GitHub
+                    if not readme_content:
+                        logger.debug(f'Fetching README from GitHub source: {source_url}')
+
+                        # Try main branch first, then fall back to master if needed
+                        for branch in ['main', 'master']:
+                            raw_readme_url = f'https://raw.githubusercontent.com/{owner}/{repo}/{branch}/README.md'
+                            logger.debug(f'Trying to fetch README from: {raw_readme_url}')
+
+                            readme_response = requests.get(raw_readme_url)
+                            if readme_response.status_code == 200:
+                                readme_content = readme_response.text
+                                logger.info(
+                                    f'Successfully fetched README from GitHub ({branch}): {len(readme_content)} chars'
+                                )
+                                break
+
+        # Add readme_content to details if available
+        if readme_content:
+            logger.info(f'Successfully extracted README content ({len(readme_content)} chars)')
+
+            # Extract outputs from README content
+            outputs = extract_outputs_from_readme(readme_content)
+            if outputs:
+                logger.info(f'Extracted {len(outputs)} outputs from README')
+                details['outputs'] = outputs
+
+            # Trim if too large
+            if len(readme_content) > 8000:
+                logger.debug(
+                    f'README content exceeds 8000 characters ({len(readme_content)}), truncating...'
+                )
+                readme_content = readme_content[:8000] + '...\n[README truncated due to length]'
+                logger.debug('README content truncated')
+
+            details['readme_content'] = readme_content
+        else:
+            logger.warning('No README content found through any method')
+
+        # Add version details if available
+        if version_details:
+            logger.info('Adding version details to response')
+            details['version_details'] = version_details
+
+        # Add version to details
+        details['version'] = module_version
+
+        return details
+
+    except Exception as e:
+        logger.error(f'Error fetching module details: {e}')
+        logger.debug(f'Stack trace: {traceback.format_exc()}')
+        return {}

awslabs/terraform_mcp_server/models/__init__.py

@@ -10,6 +10,8 @@ from .models import (
     CheckovScanResult,
     TerraformVariable,
     TerraformOutput,
+    SearchUserProvidedModuleRequest,
+    SearchUserProvidedModuleResult,
 )
 
 __all__ = [
@@ -24,4 +26,6 @@ __all__ = [
     'CheckovScanResult',
     'TerraformVariable',
     'TerraformOutput',
+    'SearchUserProvidedModuleRequest',
+    'SearchUserProvidedModuleResult',
 ]

awslabs/terraform_mcp_server/models/models.py

@@ -258,3 +258,47 @@ class CheckovScanResult(BaseModel):
     )
     summary: Dict[str, Any] = Field({}, description='Summary of scan results')
     raw_output: Optional[str] = Field(None, description='Raw output from Checkov')
+
+
+class SearchUserProvidedModuleRequest(BaseModel):
+    """Request model for searching user-provided Terraform modules.
+
+    Attributes:
+        module_url: URL of the Terraform module in the registry (e.g., 'hashicorp/consul/aws').
+        version: Optional specific version of the module to analyze.
+        variables: Optional dictionary of variables to use when analyzing the module.
+    """
+
+    module_url: str = Field(
+        ..., description='URL or identifier of the Terraform module (e.g., "hashicorp/consul/aws")'
+    )
+    version: Optional[str] = Field(None, description='Specific version of the module to analyze')
+    variables: Optional[Dict[str, Any]] = Field(
+        None, description='Variables to use when analyzing the module'
+    )
+
+
+class SearchUserProvidedModuleResult(BaseModel):
+    """Result model for searching user-provided Terraform modules.
+
+    Attributes:
+        status: Execution status (success/error).
+        module_name: Name of the analyzed module.
+        module_url: URL of the module in the registry.
+        module_version: Version of the module that was analyzed.
+        module_description: Description of the module.
+        variables: List of variables defined by the module.
+        outputs: List of outputs provided by the module.
+        readme_content: The README content of the module.
+        error_message: Optional error message if execution failed.
+    """
+
+    status: Literal['success', 'error']
+    module_name: str
+    module_url: str
+    module_version: str
+    module_description: str
+    variables: List[TerraformVariable] = Field([], description='Variables defined by the module')
+    outputs: List[TerraformOutput] = Field([], description='Outputs provided by the module')
+    readme_content: Optional[str] = Field(None, description='README content of the module')
+    error_message: Optional[str] = Field(None, description='Error message if execution failed')

awslabs/terraform_mcp_server/server.py

@@ -12,11 +12,14 @@ from awslabs.terraform_mcp_server.impl.tools import (
     search_aws_provider_docs_impl,
     search_awscc_provider_docs_impl,
     search_specific_aws_ia_modules_impl,
+    search_user_provided_module_impl,
 )
 from awslabs.terraform_mcp_server.models import (
     CheckovScanRequest,
     CheckovScanResult,
     ModuleSearchResult,
+    SearchUserProvidedModuleRequest,
+    SearchUserProvidedModuleResult,
     TerraformAWSCCProviderDocsResult,
     TerraformAWSProviderDocsResult,
     TerraformExecutionRequest,
@@ -29,7 +32,7 @@ from awslabs.terraform_mcp_server.static import (
 )
 from mcp.server.fastmcp import FastMCP
 from pydantic import Field
-from typing import Dict, List, Literal, Optional
+from typing import Any, Dict, List, Literal, Optional
 
 
 mcp = FastMCP(
@@ -264,6 +267,54 @@ async def run_checkov_scan(
     return await run_checkov_scan_impl(request)
 
 
+@mcp.tool(name='SearchUserProvidedModule')
+async def search_user_provided_module(
+    module_url: str = Field(
+        ..., description='URL or identifier of the Terraform module (e.g., "hashicorp/consul/aws")'
+    ),
+    version: Optional[str] = Field(None, description='Specific version of the module to analyze'),
+    variables: Optional[Dict[str, Any]] = Field(
+        None, description='Variables to use when analyzing the module'
+    ),
+) -> SearchUserProvidedModuleResult:
+    """Search for a user-provided Terraform registry module and understand its inputs, outputs, and usage.
+
+    This tool takes a Terraform registry module URL and analyzes its input variables,
+    output variables, README, and other details to provide comprehensive information
+    about the module.
+
+    The module URL should be in the format "namespace/name/provider" (e.g., "hashicorp/consul/aws")
+    or "registry.terraform.io/namespace/name/provider".
+
+    Examples:
+        - To search for the HashiCorp Consul module:
+          search_user_provided_module(module_url='hashicorp/consul/aws')
+
+        - To search for a specific version of a module:
+          search_user_provided_module(module_url='terraform-aws-modules/vpc/aws', version='3.14.0')
+
+        - To search for a module with specific variables:
+          search_user_provided_module(
+              module_url='terraform-aws-modules/eks/aws',
+              variables={'cluster_name': 'my-cluster', 'vpc_id': 'vpc-12345'}
+          )
+
+    Parameters:
+        module_url: URL or identifier of the Terraform module (e.g., "hashicorp/consul/aws")
+        version: Optional specific version of the module to analyze
+        variables: Optional dictionary of variables to use when analyzing the module
+
+    Returns:
+        A SearchUserProvidedModuleResult object containing module information
+    """
+    request = SearchUserProvidedModuleRequest(
+        module_url=module_url,
+        version=version,
+        variables=variables,
+    )
+    return await search_user_provided_module_impl(request)
+
+
 # * Resources
 @mcp.resource(
     name='terraform_development_workflow',

awslabs/terraform_mcp_server/static/MCP_INSTRUCTIONS.md

@@ -70,6 +70,10 @@ When implementing specific AWS resources (only after confirming no suitable AWS-
 3. `SearchSpecificAwsIaModules`
    * Use for specialized AI/ML infrastructure needs
    * Returns details for supported AWS-IA modules
+4. `SearchUserProvidedModule`
+   * Analyze any Terraform Registry module by URL or identifier
+   * Extract input variables, output variables, and README content
+   * Understand module usage and configuration options
 
 ### Command Execution Tools
 
@@ -108,6 +112,9 @@ The AWSCC provider (Cloud Control API-based) offers:
 - "Is this VPC configuration secure? Let's scan it with Checkov."
 - "Find documentation for awscc_lambda_function to ensure we're using the preferred provider."
 - "We need a Bedrock implementation for RAG. Let's search for AWS-IA modules that can help."
+- "Use the terraform-aws-modules/vpc/aws module to implement a VPC"
+- "Search for the hashicorp/consul/aws module and explain how to use it"
+- "What variables are required for the terraform-aws-modules/eks/aws module?"
 
 ## Best Practices
 

awslabs/terraform_mcp_server/static/TERRAFORM_WORKFLOW_GUIDE.md

@@ -83,6 +83,11 @@ flowchart TD
         * FIRST check for specialized AWS-IA modules (`SearchSpecificAwsIaModules` tool)
         * If no suitable module exists, THEN use AWSCC provider resources (`SearchAwsccProviderDocs` tool)
         * ONLY fall back to traditional AWS provider (`SearchAwsProviderDocs` tool) when the above options don't meet requirements
+    - When a user provides a specific Terraform Registry module to use:
+        * Use the `SearchUserProvidedModule` tool to analyze the module
+        * Extract input variables, output variables, and README content
+        * Understand module usage and configuration options
+        * Provide guidance on how to use the module correctly
     - MCP Resources and tools to consult:
         - Resources
             - *terraform_development_workflow* to consult this guide and to use it to ensure you're following the development workflow correctly
@@ -91,6 +96,7 @@ flowchart TD
             - *terraform_aws_provider_resources_listing* for available AWS resources
         - Tools
             - *SearchSpecificAwsIaModules* tool to check for specialized AWS-IA modules first (Bedrock, OpenSearch Serverless, SageMaker, Streamlit)
+            - *SearchUserProvidedModule* tool to analyze any Terraform Registry module provided by the user
             - *SearchAwsccProviderDocs* tool to look up specific Cloud Control API resources
             - *SearchAwsProviderDocs* tool to look up specific resource documentation
 2. Validate Code

{awslabs_terraform_mcp_server-0.0.2.dist-info → awslabs_terraform_mcp_server-0.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.terraform-mcp-server
-Version: 0.0.2
+Version: 0.0.7
 Summary: An AWS Labs Model Context Protocol (MCP) server for terraform
 Requires-Python: >=3.10
 Requires-Dist: beautifulsoup4>=4.12.0
@@ -45,6 +45,12 @@ MCP server for Terraform on AWS best practices, infrastructure as code patterns,
   - SageMaker endpoint deployment for ML model hosting
   - Serverless Streamlit application deployment for AI interfaces
 
+- **Terraform Registry Module Analysis** - Analyze Terraform Registry modules
+  - Search for modules by URL or identifier
+  - Extract input variables, output variables, and README content
+  - Understand module usage and configuration options
+  - Analyze module structure and dependencies
+
 - **Terraform Workflow Execution** - Run Terraform commands directly
   - Initialize, plan, validate, apply, and destroy operations
   - Pass variables and specify AWS regions

{awslabs_terraform_mcp_server-0.0.2.dist-info → awslabs_terraform_mcp_server-0.0.7.dist-info}/RECORD

@@ -1,32 +1,29 @@
 awslabs/__init__.py,sha256=4zfFn3N0BkvQmMTAIvV_QAbKp6GWzrwaUN17YeRoChM,115
 awslabs/terraform_mcp_server/__init__.py,sha256=a-zIkwClerA84_XGykBggO4w5kf8f85EapnWnbAH01c,58
-awslabs/terraform_mcp_server/server.py,sha256=3N5A5eP6YB0YqCw6nbekzpXctd724rgtu--6zqvFlfA,13626
+awslabs/terraform_mcp_server/server.py,sha256=ZyX_w3oath8l2T3tH63EyH5J_l93LZGCxW2HCeL4t2s,15787
 awslabs/terraform_mcp_server/impl/resources/__init__.py,sha256=bxqHGDtuwWq8w-21lT5GzOuxBqkmnUvW6cnSA36ve3A,388
 awslabs/terraform_mcp_server/impl/resources/terraform_aws_provider_resources_listing.py,sha256=t_guGQ44yvip9u2cONN2cWc3fXk6HyQGPu5DDP5YEkI,1973
 awslabs/terraform_mcp_server/impl/resources/terraform_awscc_provider_resources_listing.py,sha256=RoBSe-p5OyLYWNYa7aGx2f4NWyjap1wNrj8F4o8_PSo,2060
-awslabs/terraform_mcp_server/impl/tools/__init__.py,sha256=TRlf5cGGVKfofMGwX_T8YXvnq8Evcyl0AzabiwIZyFc,601
+awslabs/terraform_mcp_server/impl/tools/__init__.py,sha256=4D7D6cYF98qj4lltiO5UOHNFRl8Gfg7Zv9n-wRlOj3o,715
 awslabs/terraform_mcp_server/impl/tools/execute_terraform_command.py,sha256=_fPxC2wT23oWa9XaAoUEV1j5lK7Xpi7MqBAq85fLNdU,8507
 awslabs/terraform_mcp_server/impl/tools/run_checkov_scan.py,sha256=oS6qgUPPti7lpa2VNsUwPbswiq2hgpe49fggVUI_PfM,13435
 awslabs/terraform_mcp_server/impl/tools/search_aws_provider_docs.py,sha256=w7NX4oQwsCeVQPj8YZzvXvXcox8drzZimW3mRL8Kd84,29049
 awslabs/terraform_mcp_server/impl/tools/search_awscc_provider_docs.py,sha256=fY8eRQ1j6oB4XyfAoUC6mTUXNkoFaY0vsmMhV6w0NR8,27297
 awslabs/terraform_mcp_server/impl/tools/search_specific_aws_ia_modules.py,sha256=Kny5iharA1Wj-9tnhoPHntu4qM9n-2mC2dB6TXwhqfk,20148
+awslabs/terraform_mcp_server/impl/tools/search_user_provided_module.py,sha256=6d_QHnDwpGicMsmQmpfs0hzTtf-XAVXZBNUsBnkfxtw,14491
 awslabs/terraform_mcp_server/impl/tools/utils.py,sha256=GB1OuhYgrg00AzKfhgaUpQwbVBK5D56GQLcm60NHd1c,20500
-awslabs/terraform_mcp_server/models/__init__.py,sha256=L7nlJJ9g_J7tuYw_BD_XarEcwH3WonrIoOBZcD5FC7A,651
-awslabs/terraform_mcp_server/models/models.py,sha256=ZVn0gFlqjYTD0mNvlvXTilNhk2o9Mse9oUEBbl38Cbo,10685
+awslabs/terraform_mcp_server/models/__init__.py,sha256=-kCRp9lGMz0-SXv0XvXP48_dVCeMbQIJqkxLTqXlmhk,801
+awslabs/terraform_mcp_server/models/models.py,sha256=oQutL78FdbXXQgekXjp8Dhotq-A-wBJLylQAHRMyNs0,12614
 awslabs/terraform_mcp_server/scripts/generate_aws_provider_resources.py,sha256=v8t4UL8KwIoqVYz25Xd8wiJJ2cwKmHvxaF0_ZpUI0b4,56714
 awslabs/terraform_mcp_server/scripts/generate_awscc_provider_resources.py,sha256=FN2QOYefeEcyy-JXl1v3hLEz2YSBNRJB_9i182DdWiA,48722
 awslabs/terraform_mcp_server/scripts/scrape_aws_terraform_best_practices.py,sha256=t_eFbciBwZXdK6pNkkcp4X7VPR4vXyXkmE_YLhm6Xr4,3783
 awslabs/terraform_mcp_server/static/AWSCC_PROVIDER_RESOURCES.md,sha256=I_vu3dWXzd9Pxcd7tkdxFQs97wtuunyoXdJCAkCUXnE,440270
 awslabs/terraform_mcp_server/static/AWS_PROVIDER_RESOURCES.md,sha256=OMboscC0Ov6O3U3K1uqrzRzx18nq_5AsJzITc5-CA8Y,303030
 awslabs/terraform_mcp_server/static/AWS_TERRAFORM_BEST_PRACTICES.md,sha256=cftJ9y2nZ0kMendoV6WBlQFsNw-QnGnmF6dR88eoYdA,87665
-awslabs/terraform_mcp_server/static/MCP_INSTRUCTIONS.md,sha256=k6XQFgab3_W7_g0AZNXdyxyrzA2RjOzP9DbXS_9FiP4,6010
-awslabs/terraform_mcp_server/static/TERRAFORM_WORKFLOW_GUIDE.md,sha256=qvqtaikyyQ_XHL-LyU_BDos1ym2kKBR7ljMZtLn3FAs,9322
+awslabs/terraform_mcp_server/static/MCP_INSTRUCTIONS.md,sha256=P82tv9KJJxhoG5UFKaG16nQS5a0ud6Fcvg_BUypz-z0,6445
+awslabs/terraform_mcp_server/static/TERRAFORM_WORKFLOW_GUIDE.md,sha256=1RrSF7NH0BNVnQTZFwPhuaWsxElmCcwvl07iLOnO2gw,9768
 awslabs/terraform_mcp_server/static/__init__.py,sha256=J5JGKYybg48XyBi2hepC101RDNFBxSXZS5YGvj0tql8,549
-awslabs/terraform_mcp_server/tests/__init__.py,sha256=eWyaXCIPzCC8_4kXuKnK9s79gg6_J3XYoAtQzkNesWI,45
-awslabs/terraform_mcp_server/tests/run_tests.sh,sha256=u6GUSQffYJyn0RkAd4qeqkczYc_cf-lL3kC4a1YYjB4,1068
-awslabs/terraform_mcp_server/tests/test_parameter_annotations.py,sha256=7DrC-idSQmrSAAPLHQ09V3sU7a_8HkfCrOnqJn2SQeo,8361
-awslabs/terraform_mcp_server/tests/test_tool_implementations.py,sha256=Uh8vyFJJ9ti14hkmQDkWJEtGUJvFR3IZ0P90ez0LiMY,11641
-awslabs_terraform_mcp_server-0.0.2.dist-info/METADATA,sha256=wNYoFnJY-jLzLtJuCwGHQ8kxrIUC9uEp8D4BvYWxQiY,4344
-awslabs_terraform_mcp_server-0.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-awslabs_terraform_mcp_server-0.0.2.dist-info/entry_points.txt,sha256=jCTPQeUJ74jpIDcYwVHQCk-y0n0ehRFFerh_Qm4ZU1c,90
-awslabs_terraform_mcp_server-0.0.2.dist-info/RECORD,,
+awslabs_terraform_mcp_server-0.0.7.dist-info/METADATA,sha256=FelC966-qfJFl-RXXsRb-pNp8oHC1e9WUNqjtDIs4no,4633
+awslabs_terraform_mcp_server-0.0.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+awslabs_terraform_mcp_server-0.0.7.dist-info/entry_points.txt,sha256=jCTPQeUJ74jpIDcYwVHQCk-y0n0ehRFFerh_Qm4ZU1c,90
+awslabs_terraform_mcp_server-0.0.7.dist-info/RECORD,,

awslabs/terraform_mcp_server/tests/__init__.py

@@ -1 +0,0 @@
-"""Test package for terraform_mcp_server."""

awslabs/terraform_mcp_server/tests/run_tests.sh

@@ -1,35 +0,0 @@
-#!/bin/bash
-# Script to run the Terraform MCP server tests
-
-SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
-PROJECT_ROOT="$SCRIPT_DIR/../../.."
-
-# Set PYTHONPATH to include the project root
-export PYTHONPATH="$PROJECT_ROOT:$PYTHONPATH"
-
-# Function to run a test module
-run_test() {
-    echo "Running $1..."
-    cd "$PROJECT_ROOT"
-    python -m awslabs.terraform_mcp_server.tests.$1
-    echo "Test completed: $1"
-}
-
-# Get the test name from the first argument, default to all tests
-TEST_NAME=$1
-if [ -z "$TEST_NAME" ]; then
-    echo "=== Running All Tests ==="
-    run_test "test_parameter_annotations"
-    run_test "test_tool_implementations"
-elif [ "$TEST_NAME" == "params" ]; then
-    run_test "test_parameter_annotations"
-elif [ "$TEST_NAME" == "tools" ]; then
-    run_test "test_tool_implementations"
-else
-    echo "Unknown test: $TEST_NAME"
-    echo "Usage: $0 [params|tools]"
-    echo "  params - Run parameter annotation tests"
-    echo "  tools  - Run tool implementation tests"
-    echo "  (no args) - Run all tests"
-    exit 1
-fi

awslabs/terraform_mcp_server/tests/test_parameter_annotations.py

@@ -1,207 +0,0 @@
-"""Test script for verifying parameter annotations in MCP tools."""
-
-import json
-import sys
-from awslabs.terraform_mcp_server.server import mcp
-from pathlib import Path
-
-
-# Add project root to path to allow importing the server
-project_root = str(Path(__file__).parent.parent.parent.parent)
-if project_root not in sys.path:
-    sys.path.insert(0, project_root)
-
-
-def print_tool_parameters():
-    """Print the parameters for each tool after annotations are added."""
-    tool_names = [
-        'SearchAwsProviderDocs',
-        'ExecuteTerraformCommand',
-        'SearchAwsccProviderDocs',
-        'SearchSpecificAwsIaModules',
-        'RunCheckovScan',
-    ]
-
-    print('\n=== Current Tool Parameter Schemas ===\n')
-    for tool_name in tool_names:
-        try:
-            tool = mcp._tool_manager.get_tool(tool_name)
-            if tool is None:
-                print(f'Tool {tool_name} not found')
-                continue
-
-            if not hasattr(tool, 'parameters') or tool.parameters is None:
-                print(f'Tool {tool_name} has no parameters schema')
-                continue
-
-            print(f'=== {tool_name} Parameters Schema ===')
-            print(json.dumps(tool.parameters, indent=2))
-            print('\n')
-        except Exception as e:
-            print(f'Error getting tool {tool_name}: {e}')
-
-
-def add_parameter_annotations():
-    """Add parameter annotations to the MCP tools."""
-    print('Adding parameter annotations to MCP tools...\n')
-
-    # Add parameter descriptions for SearchAwsProviderDocs
-    search_tool = mcp._tool_manager.get_tool('SearchAwsProviderDocs')
-    if (
-        search_tool is not None
-        and hasattr(search_tool, 'parameters')
-        and search_tool.parameters is not None
-    ):
-        if (
-            'properties' in search_tool.parameters
-            and 'asset_name' in search_tool.parameters['properties']
-        ):
-            search_tool.parameters['properties']['asset_name']['description'] = (
-                'Name of the AWS service (asset) to look for (e.g., "aws_s3_bucket", "aws_lambda_function")'
-            )
-        if (
-            'properties' in search_tool.parameters
-            and 'asset_type' in search_tool.parameters['properties']
-        ):
-            search_tool.parameters['properties']['asset_type']['description'] = (
-                "Type of documentation to search - 'resource', 'data_source', or 'both' (default)"
-            )
-
-    # Add parameter descriptions for SearchAwsccProviderDocs
-    awscc_docs_tool = mcp._tool_manager.get_tool('SearchAwsccProviderDocs')
-    if (
-        awscc_docs_tool is not None
-        and hasattr(awscc_docs_tool, 'parameters')
-        and awscc_docs_tool.parameters is not None
-    ):
-        if (
-            'properties' in awscc_docs_tool.parameters
-            and 'asset_name' in awscc_docs_tool.parameters['properties']
-        ):
-            awscc_docs_tool.parameters['properties']['asset_name']['description'] = (
-                'Name of the AWSCC service (asset) to look for (e.g., awscc_s3_bucket, awscc_lambda_function)'
-            )
-        if (
-            'properties' in awscc_docs_tool.parameters
-            and 'asset_type' in awscc_docs_tool.parameters['properties']
-        ):
-            awscc_docs_tool.parameters['properties']['asset_type']['description'] = (
-                "Type of documentation to search - 'resource', 'data_source', or 'both' (default)"
-            )
-
-    # Add parameter descriptions for SearchSpecificAwsIaModules
-    modules_tool = mcp._tool_manager.get_tool('SearchSpecificAwsIaModules')
-    if (
-        modules_tool is not None
-        and hasattr(modules_tool, 'parameters')
-        and modules_tool.parameters is not None
-    ):
-        if (
-            'properties' in modules_tool.parameters
-            and 'query' in modules_tool.parameters['properties']
-        ):
-            modules_tool.parameters['properties']['query']['description'] = (
-                'Optional search term to filter modules (empty returns all four modules)'
-            )
-
-    # Add parameter descriptions for ExecuteTerraformCommand
-    terraform_tool = mcp._tool_manager.get_tool('ExecuteTerraformCommand')
-    if (
-        terraform_tool is not None
-        and hasattr(terraform_tool, 'parameters')
-        and terraform_tool.parameters is not None
-    ):
-        if (
-            'properties' in terraform_tool.parameters
-            and 'request' in terraform_tool.parameters['properties']
-        ):
-            terraform_tool.parameters['properties']['request']['description'] = (
-                'Details about the Terraform command to execute'
-            )
-
-            # Since request is a complex object with nested properties, update its schema
-            if (
-                'properties' in terraform_tool.parameters['properties']['request']
-                and 'properties'
-                in terraform_tool.parameters['properties']['request']['properties']
-            ):
-                props = terraform_tool.parameters['properties']['request']['properties']
-                if 'command' in props:
-                    props['command']['description'] = (
-                        'Terraform command to execute (init, plan, validate, apply, destroy)'
-                    )
-                if 'working_directory' in props:
-                    props['working_directory']['description'] = (
-                        'Directory containing Terraform files'
-                    )
-                if 'variables' in props:
-                    props['variables']['description'] = 'Terraform variables to pass'
-                if 'aws_region' in props:
-                    props['aws_region']['description'] = 'AWS region to use'
-                if 'strip_ansi' in props:
-                    props['strip_ansi']['description'] = (
-                        'Whether to strip ANSI color codes from output'
-                    )
-
-    # Add parameter descriptions for RunCheckovScan
-    checkov_scan_tool = mcp._tool_manager.get_tool('RunCheckovScan')
-    if (
-        checkov_scan_tool is not None
-        and hasattr(checkov_scan_tool, 'parameters')
-        and checkov_scan_tool.parameters is not None
-    ):
-        if (
-            'properties' in checkov_scan_tool.parameters
-            and 'request' in checkov_scan_tool.parameters['properties']
-        ):
-            checkov_scan_tool.parameters['properties']['request']['description'] = (
-                'Details about the Checkov scan to execute'
-            )
-
-            # Since request is a complex object with nested properties, update its schema
-            if (
-                'properties' in checkov_scan_tool.parameters['properties']['request']
-                and 'properties'
-                in checkov_scan_tool.parameters['properties']['request']['properties']
-            ):
-                props = checkov_scan_tool.parameters['properties']['request']['properties']
-                if 'working_directory' in props:
-                    props['working_directory']['description'] = (
-                        'Directory containing Terraform files to scan'
-                    )
-                if 'framework' in props:
-                    props['framework']['description'] = (
-                        'Framework to scan (terraform, cloudformation, etc.)'
-                    )
-                if 'check_ids' in props:
-                    props['check_ids']['description'] = (
-                        'Optional list of specific check IDs to run'
-                    )
-                if 'skip_check_ids' in props:
-                    props['skip_check_ids']['description'] = 'Optional list of check IDs to skip'
-                if 'output_format' in props:
-                    props['output_format']['description'] = (
-                        'Format for scan results (default: json)'
-                    )
-
-    print('Parameter annotations added successfully.\n')
-
-
-def main():
-    """Run the parameter annotation test."""
-    print('=== Terraform MCP Parameter Annotation Test ===\n')
-
-    # Print original parameter schemas
-    print('Original parameter schemas:')
-    print_tool_parameters()
-
-    # Add parameter annotations
-    add_parameter_annotations()
-
-    # Print updated parameter schemas
-    print('Updated parameter schemas:')
-    print_tool_parameters()
-
-
-if __name__ == '__main__':
-    main()

awslabs/terraform_mcp_server/tests/test_tool_implementations.py

@@ -1,309 +0,0 @@
-"""Test script for Terraform MCP server implementation functions."""
-
-import asyncio
-import json
-import sys
-from awslabs.terraform_mcp_server.impl.tools.search_aws_provider_docs import (
-    search_aws_provider_docs_impl,
-)
-from awslabs.terraform_mcp_server.impl.tools.search_awscc_provider_docs import (
-    search_awscc_provider_docs_impl,
-)
-from awslabs.terraform_mcp_server.impl.tools.search_specific_aws_ia_modules import (
-    search_specific_aws_ia_modules_impl,
-)
-from loguru import logger
-from typing import Any
-
-
-# Configure logger for enhanced diagnostics with stacktraces
-logger.configure(
-    handlers=[
-        {
-            'sink': sys.stderr,
-            'backtrace': True,
-            'diagnose': True,
-            'format': '<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>',
-        }
-    ]
-)
-
-
-def print_aws_provider_results(results):
-    """Print formatted results data using the provided logger.
-
-    Args:
-        results: List of result objects containing asset information
-        logger: Logger object to use for output
-    """
-    logger.info(f'Found {len(results)} results')
-
-    for i, result in enumerate(results):
-        logger.info(f'\nResult {i + 1}:')
-        logger.info(f'  Asset Name: {result.asset_name}')
-        logger.info(f'  Asset Type: {result.asset_type}')
-        logger.info(f'  URL: {result.url}')
-
-        # Handle description
-        if result.description:
-            description_preview = (
-                result.description[:50] + '...'
-                if len(result.description) > 50
-                else result.description
-            )
-            logger.info(f'  Description: {description_preview}')
-        else:
-            logger.info('  No description')
-
-        # Handle example usage
-        if result.example_usage:
-            logger.info(f'  Example Usage: {len(result.example_usage)} found')
-
-        # Handle arguments
-        if result.arguments:
-            logger.info(f'  Arguments: {len(result.arguments)} found')
-
-        # Handle attributes
-        if result.attributes:
-            logger.info(f'  Attributes: {len(result.attributes)} found')
-
-
-def print_awscc_provider_results(results):
-    """Print formatted results data using the provided logger.
-
-    Args:
-        results: List of result objects containing asset information
-        logger: Logger object to use for output
-    """
-    logger.info(f'Found {len(results)} results')
-
-    for i, result in enumerate(results):
-        logger.info(f'\nResult {i + 1}:')
-        logger.info(f'  Asset Name: {result.asset_name}')
-        logger.info(f'  Asset Type: {result.asset_type}')
-        logger.info(f'  URL: {result.url}')
-
-        # Handle description
-        if result.description:
-            description_preview = (
-                result.description[:50] + '...'
-                if len(result.description) > 50
-                else result.description
-            )
-            logger.info(f'  Description: {description_preview}')
-        else:
-            logger.info('  No description')
-
-        # Handle example usage
-        if result.example_usage:
-            logger.info(f'  Example Usage: {len(result.example_usage)} found')
-
-        # Handle schema arguments
-        if result.schema_arguments:
-            logger.info(f'  Schema arguments: {len(result.schema_arguments)} found')
-
-
-async def test_search_aws_provider_docs():
-    """Test the AWS provider docs search function."""
-    logger.info('=== Testing search_aws_provider_docs_impl ===')
-
-    # Test case 1: Common resource with just 1 example snippet
-    logger.info('**********---Test case 1: Searching for aws_s3_bucket as a resource---**********')
-    results = await search_aws_provider_docs_impl('aws_s3_bucket', 'resource')
-    print_aws_provider_results(results)
-
-    # Test case 2: Common resource with multiple example snippets
-    logger.info(
-        '**********---Test case 2: Searching for aws_api_gateway_rest_api as a resource---**********'
-    )
-    results = await search_aws_provider_docs_impl('api_gateway_rest_api', 'resource')
-    print_aws_provider_results(results)
-
-    # Test case 3: Common resource with multiple example snippets and multiple arguments in subsections
-    logger.info(
-        '**********---Test case 3: Searching for aws_lambda_function as a resource---**********'
-    )
-    results = await search_aws_provider_docs_impl('aws_lambda_function', 'resource')
-    print_aws_provider_results(results)
-
-    # Test case 4: Specifying data source as asset type
-    logger.info(
-        '**********---Test case 4: Searching for aws_lambda_function as a data source ---**********'
-    )
-    results = await search_aws_provider_docs_impl('aws_lambda_function', 'data_source')
-    print_aws_provider_results(results)
-
-    # Test case 5: Searching for both kinds
-    logger.info('**********---Test case 5: Searching for aws_dynamodb_table as both ---**********')
-    results = await search_aws_provider_docs_impl('aws_dynamodb_table', 'both')
-    print_aws_provider_results(results)
-
-    # Test case 6: Non-existent resource
-    logger.info('**********---Test case 6: Searching for non-existent resource---**********')
-    results = await search_aws_provider_docs_impl('aws_nonexistent_resource')
-    print_aws_provider_results(results)
-
-
-async def test_search_awscc_provider_docs():
-    """Test the AWSCC provider docs search function."""
-    logger.info('\n=== Testing search_awscc_provider_docs_impl ===')
-
-    # Test case 1: Common resource
-    logger.info(
-        '**********---Test case 1: Searching for awscc_apigateway_api_key as a resource---**********'
-    )
-    results = await search_awscc_provider_docs_impl('awscc_apigateway_api_key', 'resource')
-    print_awscc_provider_results(results)
-
-    # Test case 2: Resource with attribute
-    logger.info(
-        '**********---Test case 2: Searching for awscc_apigateway_api_key as a data source---**********'
-    )
-    results = await search_awscc_provider_docs_impl('awscc_apigateway_api_key', 'data_source')
-    print_awscc_provider_results(results)
-
-    # Test case 3: lambda_function resource
-    logger.info(
-        '**********---Test case 7: Searching for lambda_function as a resource---**********'
-    )
-    results = await search_awscc_provider_docs_impl('lambda_function', 'resource')
-    print_awscc_provider_results(results)
-
-    # Test case 4: Searching for both kinds
-    logger.info(
-        '**********---Test case 4: Searching for lambda_function as both kinds---**********'
-    )
-    results = await search_awscc_provider_docs_impl('awscc_lambda_function', 'both')
-    print_awscc_provider_results(results)
-
-    # Test case 5: Non-existent resource
-    logger.info('**********---Test case 5: Searching for non-existent resource---**********')
-    results = await search_awscc_provider_docs_impl('awscc_nonexistent_resource')
-    print_awscc_provider_results(results)
-
-
-async def test_search_specific_aws_ia_modules():
-    """Test the AWS IA modules search function."""
-    logger.info('\n=== Testing search_specific_aws_ia_modules_impl ===')
-
-    # Test case 1: Search all modules
-    logger.info('Test case 1: Searching all AWS IA modules')
-    results = await search_specific_aws_ia_modules_impl('')
-
-    logger.info(f'Found {len(results)} modules')
-    for i, result in enumerate(results):
-        logger.info(f'\nModule {i + 1}:')
-        logger.info(f'  Name: {result.name}')
-        logger.info(f'  Namespace: {result.namespace}')
-        logger.info(
-            f'  Description: {result.description[:100]}...'
-            if result.description
-            else '  No description'
-        )
-        logger.info(f'  URL: {result.url}')
-
-    # Test case 2: Search with query
-    logger.info("\nTest case 2: Searching for 'bedrock' modules")
-    results = await search_specific_aws_ia_modules_impl('bedrock')
-
-    logger.info(f'Found {len(results)} modules')
-    for i, result in enumerate(results):
-        logger.info(f'\nModule {i + 1}:')
-        logger.info(f'  Name: {result.name}')
-        logger.info(f'  Namespace: {result.namespace}')
-        logger.info(
-            f'  Description: {result.description[:100]}...'
-            if result.description
-            else '  No description'
-        )
-
-
-async def test_execute_terraform_command():
-    """Test the Terraform command execution function.
-
-    Note: This test requires a valid Terraform configuration in a temporary directory.
-    Skip this test if you don't have a valid Terraform configuration to test with.
-    """
-    logger.info('\n=== Testing execute_terraform_command_impl ===')
-    logger.info('Skipping actual execution as it requires a valid Terraform configuration.')
-    logger.info('To test this function, you would need to:')
-    logger.info('1. Create a temporary directory with valid Terraform files')
-    logger.info('2. Run terraform init, plan, etc. on those files')
-
-    # Example of how you would call it (commented out)
-    """
-    request = TerraformExecutionRequest(
-        command="validate",
-        working_directory="/path/to/terraform/config",
-        variables={"environment": "test"},
-        aws_region="us-west-2",
-        strip_ansi=True
-    )
-
-    result = await execute_terraform_command_impl(request)
-    logger.info(f"Command: {result.command}")
-    logger.info(f"Status: {result.status}")
-    logger.info(f"Return Code: {result.return_code}")
-    if result.stdout:
-        logger.info(f"Stdout: {result.stdout[:100]}...")
-    if result.stderr:
-        logger.info(f"Stderr: {result.stderr[:100]}...")
-    """
-
-
-async def test_run_checkov_scan():
-    """Test the Checkov scan function.
-
-    Note: This test requires a valid Terraform configuration in a temporary directory.
-    Skip this test if you don't have a valid Terraform configuration to test with.
-    """
-    logger.info('\n=== Testing run_checkov_scan_impl ===')
-    logger.info('Skipping actual execution as it requires a valid Terraform configuration.')
-    logger.info('To test this function, you would need to:')
-    logger.info('1. Create a temporary directory with valid Terraform files')
-    logger.info('2. Run Checkov on those files')
-
-    # Example of how you would call it (commented out)
-    """
-    request = CheckovScanRequest(
-        working_directory="/path/to/terraform/config",
-        framework="terraform",
-        output_format="json"
-    )
-
-    result = await run_checkov_scan_impl(request)
-    logger.info(f"Status: {result.status}")
-    logger.info(f"Return Code: {result.return_code}")
-    logger.info(f"Found {len(result.vulnerabilities)} vulnerabilities")
-    for i, vuln in enumerate(result.vulnerabilities[:3]):  # Show first 3 only
-        logger.info(f"\nVulnerability {i+1}:")
-        logger.info(f"  ID: {vuln.id}")
-        logger.info(f"  Resource: {vuln.resource}")
-        logger.info(f"  Description: {vuln.description[:100]}..." if vuln.description else "  No description")
-    """
-
-
-def format_json(obj: Any) -> str:
-    """Format an object as pretty JSON."""
-    if hasattr(obj, 'model_dump'):
-        # For Pydantic v2
-        data = obj.model_dump()
-    elif hasattr(obj, 'dict'):
-        # For Pydantic v1
-        data = obj.dict()
-    else:
-        data = obj
-    return json.dumps(data, indent=2, default=str)
-
-
-async def main():
-    """Run all tests."""
-    try:
-        await test_search_aws_provider_docs()
-        await test_search_awscc_provider_docs()
-    except Exception as e:
-        logger.exception(f'Error running tests: {e}')
-
-
-if __name__ == '__main__':
-    asyncio.run(main())