t6_mem0_v2/test-mcp-tools.py

#!/usr/bin/env python3
"""Quick test of MCP server tool definitions"""

import asyncio
import sys
from mcp_server.main import T6Mem0Server

async def test_tools():
    """Test MCP server initialization and tool listing"""
    try:
        print("Initializing T6 Mem0 MCP Server...")
        server = T6Mem0Server()
        await server.initialize()

        print("\n✓ Server initialized successfully")
        print(f"✓ Memory instance: {type(server.memory).__name__}")
        print(f"✓ Tools instance: {type(server.tools).__name__}")

        # Get tool definitions
        tools = server.tools.get_tool_definitions()
        print(f"\n✓ Found {len(tools)} MCP tools:")

        for tool in tools:
            print(f"\n  • {tool.name}")
            print(f"    {tool.description}")
            required = tool.inputSchema.get('required', [])
            if required:
                print(f"    Required: {', '.join(required)}")

        print("\n✅ MCP server test passed!")
        return 0

    except Exception as e:
        print(f"\n❌ Test failed: {e}", file=sys.stderr)
        import traceback
        traceback.print_exc()
        return 1

if __name__ == "__main__":
    exit_code = asyncio.run(test_tools())
    sys.exit(exit_code)