import asyncio
import logging

from fastmcp import Client, FastMCP
from fastmcp.client.transports import StreamableHttpTransport

# External MCP server client
transport = StreamableHttpTransport(url="http://[REMOTE_IP]:8080/mcp")
client = Client(transport)


async def remote_tool() -> str:
    """Call remote MCP server tool directly"""
    async with client:
        result = await client.call_tool("weather_tool", {})
        print(result)
        return "test remote_tool"


# Local MCP server
mcp = FastMCP(name="Local Server")


@mcp.tool()
async def local_tool() -> str:
    """Local tool that attempts to call remote tool"""
    try:
        r = await remote_tool()
        print(r)
    except Exception as e:
        logging.exception("call exception")
    return "done"


async def run_http_server():
    """Run local MCP server"""
    await mcp.run_async(
        transport="http", 
        show_banner=False, 
        host="0.0.0.0", 
        port=8300
    )


if __name__ == "__main__":
    # This works fine
    asyncio.run(remote_tool())
    
    # This fails with 400 Bad Request when local_tool is called
    asyncio.run(run_http_server())