"""Thin MCP SSE client helper.

Uses the `mcp` Python SDK's sse_client transport to connect to an SSE MCP server,

call one tool, and return the raw text result. Falls back gracefully on any failure.

"""

from __future__ import annotations

import asyncio

import json

from contextlib import asynccontextmanager

from typing import Any, AsyncGenerator

import httpx

from mcp.client.sse import sse_client

from mcp import ClientSession

async def call_mcp_tool(

    base_url: str,

    tool_name: str,

    arguments: dict[str, Any],

    timeout: float = 3.5,

) -> list[Any]:

    """Call an MCP tool over SSE and return parsed content blocks.

    Each content block's text is JSON-parsed when possible.  The result may be:

      - A dict  (e.g. {"total": N, "results": [...]} from CBM search_graph)

      - A list  (e.g. [...entities...] from memory-graph query_entities)

      - A dict with "text" key (plain-text fallback when JSON parse fails)

    Returns empty list on any failure (timeout, connection error, parse error).

    """

    try:

        async with asyncio.timeout(timeout):

            async with sse_client(base_url) as (read, write):

                async with ClientSession(read, write) as session:

                    await session.initialize()

                    result = await session.call_tool(tool_name, arguments=arguments)

                    out: list[Any] = []

                    for block in (result.content or []):

                        if hasattr(block, "text"):

                            try:

                                out.append(json.loads(block.text))

                            except (json.JSONDecodeError, TypeError):

                                out.append({"text": block.text})

                    return out

    except Exception:

        return []