databricks_mcp_server_quickstart_with_stdio_transport.py

1import asyncio
2from databricks.mcp.server import DatabricksPCPServer
3from mcp.server.models import InitializationOptions
4import mcp.types as types
5
6async def main():
7    # Initialize the Databricks MCP Server
8    # Note: Ensure DATABRICKS_HOST and DATABRICKS_TOKEN are set in your environment
9    server = DatabricksPCPServer("databricks-mcp-server")
10
11    # Run the server using Standard Input/Output as the transport layer
12    async with server.run_stdio_transport() as (read_stream, write_stream):
13        await server.mcp_server.run(
14            read_stream,
15            write_stream,
16            InitializationOptions(
17                server_name="databricks-mcp-server",
18                server_version="0.1.0",
19                capabilities=server.mcp_server.get_capabilities(),
20            ),
21        )
22
23if __name__ == "__main__":
24    asyncio.run(main())