import requests

API_KEY = "sk_xxxxxxxxxxxxxxxx"   # 到 https://data.epmap.org/api_key 创建
BASE = "https://mcp.epmap.org/v1/hydrology_data/api"
# 下面两种写法等价，跑其中一种即可（两段都跑会各扣一次费）

# 方式一：GET，参数走 query string
resp = requests.get(
    f"{BASE}/get_stations",
    headers={"Authorization": f"Bearer {API_KEY}"},
    params={
        "province": "福建",
        "site_name": "龙须湖",
        "data_type": "水库",
        "page": "1",
    },
    timeout=30,
)
print(resp.status_code)
print(resp.json())

# 方式二：POST，参数走 JSON body（写 json=，不要用 data= 传表单）
resp = requests.post(
    f"{BASE}/get_stations",
    headers={"Authorization": f"Bearer {API_KEY}"},
    json={
        "province": "福建",
        "site_name": "龙须湖",
        "data_type": "水库",
        "page": "1",
    },
    timeout=30,
)
print(resp.status_code)
print(resp.json())

import asyncio
from mcp import ClientSession
from mcp.client.streamable_http import streamablehttp_client

API_KEY = "sk_xxxxxxxxxxxxxxxx"
URL = "https://mcp.epmap.org/v1/hydrology_data/mcp"

async def main():
    async with streamablehttp_client(
        URL, headers={"Authorization": f"Bearer {API_KEY}"}
    ) as (read, write, _):
        async with ClientSession(read, write) as session:
            await session.initialize()

            # 先看服务端提供了哪些工具
            listed = await session.list_tools()
            print([t.name for t in listed.tools])

            result = await session.call_tool(
                "get_stations",
                {
                    "province": "福建",
                    "site_name": "龙须湖",
                    "data_type": "水库",
                    "page": "1",
                },
            )
            print(result.structuredContent or result.content[0].text)

asyncio.run(main())

curl -G "https://mcp.epmap.org/v1/hydrology_data/api/get_stations" \
  -H "Authorization: Bearer sk_xxxxxxxxxxxxxxxx" \
  --data-urlencode 'province=福建' \
  --data-urlencode 'site_name=龙须湖' \
  --data-urlencode 'data_type=水库' \
  --data-urlencode 'page=1'

{
  "mcpServers": {
    "epmap-hydrology_data": {
      "type": "streamable-http",
      "url": "https://mcp.epmap.org/v1/hydrology_data/mcp",
      "headers": {
        "Authorization": "Bearer sk_xxxxxxxxxxxxxxxx"
      }
    }
  }
}