databricks-mcp

import os from databricks_mcp.databricks_config import DatabricksMcpConfig from databricks_mcp.databricks_mcp_client import DatabricksMcpClient # Ensure DATABRICKS_HOST, DATABRICKS_TOKEN, and DATABRICKS_ACCOUNT_ID # are set as environment variables for a real connection. # Token should be a Databricks Account-level token. try: config = DatabricksMcpConfig( host=os.environ.get("DATABRICKS_HOST", "https://accounts.cloud.databricks.com"), token=os.environ.get("DATABRICKS_TOKEN", "dapi-example-token"), account_id=os.environ.get("DATABRICKS_ACCOUNT_ID", "00000000-0000-0000-0000-000000000000"), ) client = DatabricksMcpClient(config) print(f"Databricks MCP Client initialized successfully for host: {client.config.host}") # The client object contains underlying databricks_sdk clients (e.g., client.account_client) # which can be used to perform actual operations. This quickstart only demonstrates initialization. except Exception as e: print(f"Failed to initialize Databricks MCP Client: {e}") print("Please ensure DATABRICKS_HOST, DATABRICKS_TOKEN, and DATABRICKS_ACCOUNT_ID") print("environment variables are correctly configured and point to your Databricks Account.")