Databricks Workspace connector
Service Principal (OAuth 2.0)AnalyticsAutomationDatabasesConnect to Databricks Workspace APIs using a Service Principal with OAuth 2.0 client credentials to manage clusters, jobs, notebooks, SQL, and more.
Databricks Workspace connector
-
Install the SDK
Section titled “Install the SDK”Terminal window npm install @scalekit-sdk/nodeTerminal window pip install scalekit -
Set your credentials
Section titled “Set your credentials”Add your Scalekit credentials to your
.envfile. Find values in app.scalekit.com > Developers > API Credentials..env SCALEKIT_ENVIRONMENT_URL=<your-environment-url>SCALEKIT_CLIENT_ID=<your-client-id>SCALEKIT_CLIENT_SECRET=<your-client-secret>
What you can do
Section titled “What you can do”Connect this agent connector to let your agent:
- Schemata information schema — List all schemas within a catalog using INFORMATION_SCHEMA.SCHEMATA
- Constraints information schema table — List PRIMARY KEY and FOREIGN KEY constraints for tables in a schema using INFORMATION_SCHEMA.TABLE_CONSTRAINTS
- List unity catalog schemas, unity catalog catalogs, unity catalog tables — List all schemas within a Unity Catalog in the Databricks workspace
- Get sql statement result chunk, sql warehouse, sql statement — Fetch a specific result chunk for a paginated SQL statement result
- Tables information schema — List tables and views in a schema using INFORMATION_SCHEMA.TABLES
- Columns information schema — List columns for a table using INFORMATION_SCHEMA.COLUMNS
Tool list
Section titled “Tool list”Use the exact tool names from the Tool list below when you call execute_tool. If you’re not sure which name to use, list the tools available for the current user first.
databricksworkspace_cluster_get#Get details of a specific Databricks cluster by cluster ID.1 param
Get details of a specific Databricks cluster by cluster ID.
cluster_idstringrequiredThe unique identifier of the cluster.databricksworkspace_cluster_start#Start a terminated Databricks cluster by cluster ID.1 param
Start a terminated Databricks cluster by cluster ID.
cluster_idstringrequiredThe unique identifier of the cluster to start.databricksworkspace_cluster_terminate#Terminate a Databricks cluster by cluster ID. The cluster will be deleted and all its associated resources released.1 param
Terminate a Databricks cluster by cluster ID. The cluster will be deleted and all its associated resources released.
cluster_idstringrequiredThe unique identifier of the cluster to terminate.databricksworkspace_clusters_list#List all clusters in the Databricks workspace.0 params
List all clusters in the Databricks workspace.
databricksworkspace_information_schema_columns#List columns for a table using INFORMATION_SCHEMA.COLUMNS. Returns column name, data type, nullability, numeric precision/scale, max char length, and comment.4 params
List columns for a table using INFORMATION_SCHEMA.COLUMNS. Returns column name, data type, nullability, numeric precision/scale, max char length, and comment.
catalogstringrequiredThe catalog containing the table.schemastringrequiredThe schema containing the table.tablestringrequiredThe table to list columns for.warehouse_idstringrequiredThe ID of the SQL warehouse to run the query on.databricksworkspace_information_schema_schemata#List all schemas within a catalog using INFORMATION_SCHEMA.SCHEMATA. Used for schema discovery during setup.2 params
List all schemas within a catalog using INFORMATION_SCHEMA.SCHEMATA. Used for schema discovery during setup.
catalogstringrequiredThe catalog to list schemas from.warehouse_idstringrequiredThe ID of the SQL warehouse to run the query on.databricksworkspace_information_schema_table_constraints#List PRIMARY KEY and FOREIGN KEY constraints for tables in a schema using INFORMATION_SCHEMA.TABLE_CONSTRAINTS. Used to auto-detect join keys.3 params
List PRIMARY KEY and FOREIGN KEY constraints for tables in a schema using INFORMATION_SCHEMA.TABLE_CONSTRAINTS. Used to auto-detect join keys.
catalogstringrequiredThe catalog containing the schema.schemastringrequiredThe schema to list constraints from.warehouse_idstringrequiredThe ID of the SQL warehouse to run the query on.databricksworkspace_information_schema_tables#List tables and views in a schema using INFORMATION_SCHEMA.TABLES. Returns table name, type (MANAGED, EXTERNAL, VIEW, etc.), and comment for schema discovery.3 params
List tables and views in a schema using INFORMATION_SCHEMA.TABLES. Returns table name, type (MANAGED, EXTERNAL, VIEW, etc.), and comment for schema discovery.
catalogstringrequiredThe catalog to query INFORMATION_SCHEMA from.schemastringrequiredThe schema to list tables from.warehouse_idstringrequiredThe ID of the SQL warehouse to run the query on.databricksworkspace_job_get#Get details of a specific Databricks job by job ID.1 param
Get details of a specific Databricks job by job ID.
job_idintegerrequiredThe unique identifier of the job.databricksworkspace_job_run_now#Trigger an immediate run of a Databricks job by job ID.1 param
Trigger an immediate run of a Databricks job by job ID.
job_idintegerrequiredThe unique identifier of the job to run.databricksworkspace_job_runs_list#List all job runs in the Databricks workspace, optionally filtered by job ID.3 params
List all job runs in the Databricks workspace, optionally filtered by job ID.
job_idintegeroptionalFilter runs by a specific job ID. If omitted, returns runs for all jobs.limitintegeroptionalThe number of runs to return. Defaults to 20. Maximum is 1000.offsetintegeroptionalThe offset of the first run to return.databricksworkspace_jobs_list#List all jobs in the Databricks workspace.2 params
List all jobs in the Databricks workspace.
limitintegeroptionalThe number of jobs to return. Defaults to 20. Maximum is 100.offsetintegeroptionalThe offset of the first job to return.databricksworkspace_scim_me_get#Retrieve information about the currently authenticated service principal in the Databricks workspace.0 params
Retrieve information about the currently authenticated service principal in the Databricks workspace.
databricksworkspace_scim_users_list#List all users in the Databricks workspace using the SCIM v2 API.3 params
List all users in the Databricks workspace using the SCIM v2 API.
countintegeroptionalMaximum number of results to return per page.filterstringoptionalSCIM filter expression to narrow results (e.g. userName eq "user@example.com").startIndexintegeroptional1-based index of the first result to return. Used for pagination.databricksworkspace_secrets_scopes_list#List all secret scopes available in the Databricks workspace.0 params
List all secret scopes available in the Databricks workspace.
databricksworkspace_sql_statement_cancel#Cancel a running SQL statement by its statement ID.1 param
Cancel a running SQL statement by its statement ID.
statement_idstringrequiredThe ID of the SQL statement to cancel.databricksworkspace_sql_statement_execute#Execute a SQL statement on a Databricks SQL warehouse and return the results.4 params
Execute a SQL statement on a Databricks SQL warehouse and return the results.
statementstringrequiredThe SQL statement to execute.warehouse_idstringrequiredThe ID of the SQL warehouse to execute the statement on.catalogstringoptionalThe catalog to use for the statement execution.schemastringoptionalThe schema to use for the statement execution.databricksworkspace_sql_statement_get#Get the status and results of a previously executed SQL statement by its statement ID.1 param
Get the status and results of a previously executed SQL statement by its statement ID.
statement_idstringrequiredThe ID of the SQL statement to retrieve.databricksworkspace_sql_statement_result_chunk_get#Fetch a specific result chunk for a paginated SQL statement result. Use when a statement result has multiple chunks (large result sets).2 params
Fetch a specific result chunk for a paginated SQL statement result. Use when a statement result has multiple chunks (large result sets).
chunk_indexintegerrequiredThe index of the result chunk to fetch (0-based).statement_idstringrequiredThe ID of the SQL statement.databricksworkspace_sql_warehouse_get#Get details of a specific Databricks SQL warehouse by its ID.1 param
Get details of a specific Databricks SQL warehouse by its ID.
warehouse_idstringrequiredThe ID of the SQL warehouse to retrieve.databricksworkspace_sql_warehouse_start#Start a stopped Databricks SQL warehouse by its ID.1 param
Start a stopped Databricks SQL warehouse by its ID.
warehouse_idstringrequiredThe ID of the SQL warehouse to start.databricksworkspace_sql_warehouse_stop#Stop a running Databricks SQL warehouse by its ID.1 param
Stop a running Databricks SQL warehouse by its ID.
warehouse_idstringrequiredThe ID of the SQL warehouse to stop.databricksworkspace_sql_warehouses_list#List all SQL warehouses available in the Databricks workspace.0 params
List all SQL warehouses available in the Databricks workspace.
databricksworkspace_unity_catalog_catalogs_list#List all Unity Catalogs accessible to the service principal in the Databricks workspace.0 params
List all Unity Catalogs accessible to the service principal in the Databricks workspace.
databricksworkspace_unity_catalog_schemas_list#List all schemas within a Unity Catalog in the Databricks workspace.1 param
List all schemas within a Unity Catalog in the Databricks workspace.
catalog_namestringrequiredThe name of the catalog to list schemas from.databricksworkspace_unity_catalog_tables_list#List all tables and views within a schema in a Unity Catalog in the Databricks workspace.2 params
List all tables and views within a schema in a Unity Catalog in the Databricks workspace.
catalog_namestringrequiredThe name of the catalog containing the schema.schema_namestringrequiredThe name of the schema to list tables from.