import asyncio

from fastmcp import FastMCP
from fastmcp.client import Client
from fastmcp.client.transports import FastMCPTransport
from fastmcp.server.proxy import ProxyClient
from fastmcp.tools.tool import ToolResult


async def main():
    # Create the FastMCP server
    fastmcp_server = FastMCP("TestServer")

    # Add a tool that returns metadata
    @fastmcp_server.tool
    def tool_with_meta(value: str) -> ToolResult:
        """A tool that returns metadata in its result."""
        return ToolResult(
            content=f"Result: {value}",
            meta={"custom_key": "custom_value", "processed": True},
        )

    # Create the proxy server
    proxy_server = FastMCP.as_proxy(
        ProxyClient(transport=FastMCPTransport(fastmcp_server))
    )

    # Call the tool through the proxy
    async with Client(proxy_server) as client:
        result = await client.call_tool("tool_with_meta", {"value": "test"})

    # Verify the results
    print(f"Content: {result.content[0].text}")  # type: ignore[attr-defined]
    print(f"Meta: {result.meta}"). # prints None



if __name__ == "__main__":
    asyncio.run(main())