# /// script
# requires-python = ">=3.13"
# dependencies = [
#     "mcp",
#     "python-dotenv",
# ]
# ///
import asyncio
import os

from dotenv import load_dotenv
from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client

load_dotenv()

server_params = StdioServerParameters(
    command='uvx', args=['logfire-mcp@latest', '--read-token', os.environ['LOGFIRE_READ_TOKEN']]
)


async def main():
    async with stdio_client(server_params) as (read, write):
        async with ClientSession(read, write) as session:
            await session.initialize()
            tools = await session.list_tools()
            print('tools:', [tool.name for tool in tools.tools])
            reference = await session.call_tool('schema_reference')
            print(reference.content[0].text)


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