print_tools.py

# 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.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Script to print out all registered tools and their signatures."""

from awslabs.amazon_sns_sqs_mcp_server.sns import register_sns_tools
from awslabs.amazon_sns_sqs_mcp_server.sqs import register_sqs_tools
from mcp.server.fastmcp import FastMCP


async def print_tool_info(mcp):
    """Print information about all tools registered with the MCP server."""
    print('\n=== Registered Tools ===\n')

    # Use the list_tools method to get all registered tools
    tools = await mcp.list_tools()

    for tool in tools:
        print(f'Tool: {tool.name}')
        print(f'  Description: {tool.description}')

        # Print input schema if available
        if tool.inputSchema:
            print('  Input Schema:')
            for prop_name, prop_info in tool.inputSchema.get('properties', {}).items():
                prop_type = prop_info.get('type', 'any')
                required = prop_name in tool.inputSchema.get('required', [])
                req_str = ' (required)' if required else ''
                print(f'    - {prop_name}: {prop_type}{req_str}')
                if 'description' in prop_info:
                    print(f'      {prop_info["description"]}')

        print()  # Empty line for readability


async def main():
    """Register tools and print their information."""
    # Create a FastMCP instance
    mcp = FastMCP()

    # Register tools
    print('Registering SNS tools...')
    register_sns_tools(mcp)

    print('Registering SQS tools...')
    register_sqs_tools(mcp)

    # Print tool information
    await print_tool_info(mcp)


if __name__ == '__main__':
    import asyncio

    asyncio.run(main())