| """Test script for CMBAgent MCP Server"""
|
| import asyncio
|
| from mcp import ClientSession, StdioServerParameters
|
| from mcp.client.stdio import stdio_client
|
|
|
|
|
| async def test_mcp_server():
|
| """Test the MCP server by connecting and listing available tools."""
|
| print("🧪 Testing CMBAgent MCP Server...\n")
|
|
|
|
|
| server_params = StdioServerParameters(
|
| command="python",
|
| args=["-m", "cmbagent_mcp.server", "stdio"]
|
| )
|
|
|
|
|
| async with stdio_client(server_params) as (read, write):
|
| async with ClientSession(read, write) as session:
|
|
|
| await session.initialize()
|
| print("✅ Connected to MCP server\n")
|
|
|
|
|
| tools_result = await session.list_tools()
|
| print(f"📋 Available tools ({len(tools_result.tools)}):")
|
| for tool in tools_result.tools:
|
| print(f"\n Tool: {tool.name}")
|
| print(f" Description: {tool.description}")
|
| if tool.inputSchema:
|
| print(f" Parameters:")
|
| props = tool.inputSchema.get("properties", {})
|
| for param_name, param_info in props.items():
|
| param_type = param_info.get("type", "unknown")
|
| param_desc = param_info.get("description", "")
|
| print(f" - {param_name} ({param_type}): {param_desc}")
|
|
|
| print("\n✅ MCP Server test completed successfully!")
|
|
|
|
|
| if __name__ == "__main__":
|
| asyncio.run(test_mcp_server())
|
|
|
