Databricks MCP Server

Instructions

Implementation Reference

• src/server/simple_databricks_mcp_server.py:337-346 (handler)

  MCP tool handler and registration for 'get_sql_status'. Takes statement_id as input, fetches status via sql.get_statement_status helper, returns JSON result or error.

  @mcp.tool()
  async def get_sql_status(statement_id: str) -> str:
      """Get the status and results of a SQL statement by statement_id"""
      logger.info(f"Getting status for SQL statement: {statement_id}")
      try:
          result = await sql.get_statement_status(statement_id)
          return json.dumps(result)
      except Exception as e:
          logger.error(f"Error getting SQL status: {str(e)}")
          return json.dumps({"error": str(e)})

• src/api/sql.py:140-155 (helper)

  Supporting utility function that performs the actual API call to Databricks /sql/statements/{statement_id} endpoint to retrieve the SQL statement status.

  async def get_statement_status(statement_id: str) -> Dict[str, Any]:
      """
      Get the status of a SQL statement.
      
      Args:
          statement_id: ID of the statement to check
          
      Returns:
          Response containing statement status
          
      Raises:
          DatabricksAPIError: If the API request fails
      """
      logger.info(f"Getting status of SQL statement: {statement_id}")
      return make_api_request("GET", f"/api/2.0/sql/statements/{statement_id}", params={})