38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
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():
|
||
# 连接到服务端，端口 8010
|
||
client = MCPClient("http://localhost:8010")
|
||
try:
|
||
# 步骤 1：获取可用工具列表
|
||
tools = await client.get_tools()
|
||
logger.info("可用工具列表：")
|
||
for tool in tools:
|
||
print(tool)
|
||
logger.info(f"- {tool.get('name', '未知')}: {tool.get('description', '无描述')}")
|
||
|
||
# 步骤 2：调用查询高频问题工具
|
||
result_qhf = await client.call_tool("query_high_frequency_question")
|
||
logger.info(f"高频问题查询结果：{result_qhf}")
|
||
|
||
# 步骤 3：调用查询天气工具
|
||
result_weather = await client.call_tool("get_weather")
|
||
logger.info(f"天气查询结果：{result_weather}")
|
||
|
||
except Exception as e:
|
||
logger.error(f"MCP 客户端出错：{str(e)}", exc_info=True)
|
||
finally:
|
||
await client.close()
|
||
|
||
async def main():
|
||
await test_mcp_tools()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main()) |