import asyncio
import logging

from python_a2a.mcp import MCPClient

# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

async def test_mcp_tools():
    # 创建MCP客户端
    mcp_client = MCPClient(server_url="http://localhost:8010")
    try:
        # 获取可用的工具列表
        tools = await mcp_client.get_tools()
        print(f'MCP tools-->{tools}')

        # 直接调用工具
        result = await mcp_client.call_tool("query_high_frequency_question")
        print(f'MCP result-->{result}')

    except Exception as e:
        logger.error(f'执行报错，错误信息为：{e}')


if __name__ == '__main__':
    asyncio.run(test_mcp_tools())