MCP integration

1
{
2
  "mcpServers": {
3
    "agent-handler": {
4
      "command": "npx",
5
      "args": [
6
        "-y",
7
        "mcp-remote",
8
        "https://ah-api.merge.dev/api/v1/tool-packs/<TOOL_PACK_ID>/registered-users/<REGISTERED_USER_ID>/mcp",
9
        "--header",
10
        "Authorization: Bearer ${AUTH_TOKEN}"
11
      ],
12
      "env": {
13
        "AUTH_TOKEN": "<YOUR_API_KEY>"
14
      }
15
    }
16
  }
17
}

1
import asyncio
2
from mcp.client.session import ClientSession
3
from mcp.client.streamable_http import streamablehttp_client
4
5
API_KEY = "<YOUR_API_KEY>"
6
TOOL_PACK_ID = "<TOOL_PACK_ID>"
7
REGISTERED_USER_ID = "<REGISTERED_USER_ID>"
8
9
async def run():
10
    url = (
11
        f"https://ah-api.merge.dev/api/v1/tool-packs/{TOOL_PACK_ID}"
12
        f"/registered-users/{REGISTERED_USER_ID}/mcp"
13
    )
14
    headers = {"Authorization": f"Bearer {API_KEY}"}
15
16
    async with streamablehttp_client(url, headers=headers) as (read, write, _):
17
        async with ClientSession(read, write) as session:
18
            await session.initialize()
19
            tools = await session.list_tools()
20
            print(tools)
21
22
            result = await session.call_tool(
23
                "weather__get_forecast",
24
                {"location": "Stockholm", "days": 3},
25
            )
26
            print(result)
27
28
asyncio.run(run())

1
import uuid
2
import aiohttp
3
4
class MCPClient:
5
    def __init__(self, url: str, api_key: str):
6
        self.url = url
7
        self.session_id = str(uuid.uuid4())
8
        self.request_id = 0
9
        self.headers = {
10
            "Authorization": f"Bearer {api_key}",
11
            "Content-Type": "application/json",
12
            "Accept": "application/json, text/event-stream",
13
            "Mcp-Session-Id": self.session_id,
14
        }
15
16
    async def _send(self, method: str, params: dict | None = None) -> dict:
17
        self.request_id += 1
18
        payload = {"jsonrpc": "2.0", "id": self.request_id, "method": method}
19
        if params:
20
            payload["params"] = params
21
22
        async with aiohttp.ClientSession() as http:
23
            async with http.post(self.url, headers=self.headers, json=payload) as resp:
24
                if "Mcp-Session-Id" in resp.headers:
25
                    self.session_id = resp.headers["Mcp-Session-Id"]
26
                    self.headers["Mcp-Session-Id"] = self.session_id
27
                resp.raise_for_status()
28
                return await resp.json()
29
30
    async def initialize(self):
31
        return await self._send("initialize", {"protocolVersion": "2024-11-05"})
32
33
    async def list_tools(self):
34
        return await self._send("tools/list")
35
36
    async def call_tool(self, name: str, arguments: dict):
37
        return await self._send("tools/call", {"name": name, "arguments": arguments})