Amap (Gaode Maps) MCP Server

import pytest import asyncio import traceback from datetime import timedelta from mcp.client.session import ClientSession from mcp.client.stdio import StdioServerParameters, stdio_client from mcp.client.streamable_http import streamablehttp_client # 编写客户端代码进行调试stdio协议的MCP服务 # @pytest.mark.asyncio async def test_mcp_server(): async with stdio_client( StdioServerParameters(command="uv", args=["run", "build_mcp"]) ) as (read, write): print("启动服务端...") async with ClientSession(read, write) as session: await session.initialize() print("初始化完成") tools = await session.list_tools() print("可用工具：", tools) assert hasattr(tools, "tools") assert isinstance(tools.tools, list) assert any(tool.name == "locate_ip" for tool in tools.tools) result = await session.call_tool("locate_ip", {"ip": None}) print("结果:", result.content[0].text) # 编写客户端代码进行调试streamablehttp_client协议的MCP服务 # @pytest.mark.asyncio async def test_streamable_http_main(): # 测试 mcp 客户端的功能 try: SERVER_URL = "http://127.0.0.1:8000/mcp" # 与服务器路径保持一致 # 1. 建立 Streamable-HTTP 长连接 async with streamablehttp_client(SERVER_URL) as (read_stream, write_stream, _): # 2. 创建会话 async with ClientSession(read_stream, write_stream) as session: # 3. 握手 await session.initialize() # 4. 查看可用工具 tools = await session.list_tools() print("可用工具:", [t.name for t in tools.tools]) # 5. 调用工具示例（假设服务端注册了名为 get_time 的工具） res = await session.call_tool("locate_ip", {}) print("工具返回 =>", res) list_prompts = await session.list_prompts() print("list_prompts:", list_prompts) list_resources = await session.list_resources() print("list_resources:", list_resources) except: print(f"traceback: {traceback.print_exc()}") async def test_streamable_http_main_ok(): # 项目中出现 tests_*.py / *_test.py 里的函数 PyCharm 默认把这类文件划进 Test Scope； # 只要函数名以 test_ 开头（或类以 Test 开头），就会被 Python Test Runner 识别,识别后自动标注为「可执行的测试用例」，于是左侧出现 ▶ Run Test await asyncio.sleep(0.1) print("test_streamable_http_main_ok") async def streamable_http_main_fail(): await asyncio.sleep(0.1) print("test_streamable_http_main_ok")