runline 0.2.1 → 0.3.0

package/dist/plugins/databricks/src/index.js

@@ -11,7 +11,13 @@ async function api(host, token, method, path, body, qs) {
                 url.searchParams.set(k, v);
         }
     }
-    const init = { method, headers: { Authorization: `Bearer ${token}`, "Content-Type": "application/json" } };
+    const init = {
+        method,
+        headers: {
+            Authorization: `Bearer ${token}`,
+            "Content-Type": "application/json",
+        },
+    };
     if (body && Object.keys(body).length > 0)
         init.body = JSON.stringify(body);
     const res = await fetch(url.toString(), init);
@@ -28,35 +34,64 @@ function volumeParts(volumePath) {
         throw new Error("Volume path must be catalog.schema.volume");
     return parts;
 }
-function sleep(ms) { return new Promise((r) => setTimeout(r, ms)); }
+function sleep(ms) {
+    return new Promise((r) => setTimeout(r, ms));
+}
 export default function databricks(rl) {
     rl.setName("databricks");
     rl.setVersion("0.1.0");
     rl.setConnectionSchema({
-        host: { type: "string", required: true, description: "Databricks workspace URL, e.g. https://adb-12345.azuredatabricks.net", env: "DATABRICKS_HOST" },
-        accessToken: { type: "string", required: true, description: "Personal access token or OAuth2 token", env: "DATABRICKS_TOKEN" },
+        host: {
+            type: "string",
+            required: true,
+            description: "Databricks workspace URL, e.g. https://adb-12345.azuredatabricks.net",
+            env: "DATABRICKS_HOST",
+        },
+        accessToken: {
+            type: "string",
+            required: true,
+            description: "Personal access token or OAuth2 token",
+            env: "DATABRICKS_TOKEN",
+        },
     });
     // ── Databricks SQL ──────────────────────────────────
-    rl.registerAction("sql.executeQuery", { description: "Execute a SQL query on a warehouse and return results",
+    rl.registerAction("sql.executeQuery", {
+        description: "Execute a SQL query on a warehouse and return results",
         inputSchema: {
-            warehouseId: { type: "string", required: true, description: "SQL warehouse ID" },
+            warehouseId: {
+                type: "string",
+                required: true,
+                description: "SQL warehouse ID",
+            },
             query: { type: "string", required: true, description: "SQL statement" },
-            parameters: { type: "object", required: false, description: "Array of {name, value, type?} query parameters" },
+            parameters: {
+                type: "object",
+                required: false,
+                description: "Array of {name, value, type?} query parameters",
+            },
         },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             const params = (p.parameters ?? []);
-            const body = { warehouse_id: p.warehouseId, statement: p.query, wait_timeout: "50s", on_wait_timeout: "CONTINUE" };
+            const body = {
+                warehouse_id: p.warehouseId,
+                statement: p.query,
+                wait_timeout: "50s",
+                on_wait_timeout: "CONTINUE",
+            };
             if (params.length > 0)
                 body.parameters = params.map(({ name, value, type }) => type ? { name, value, type } : { name, value });
-            let result = await api(host, token, "POST", "/api/2.0/sql/statements", body);
+            let result = (await api(host, token, "POST", "/api/2.0/sql/statements", body));
             const statementId = result.statement_id;
             let status = result.status.state;
             let retries = 0;
-            while (status !== "SUCCEEDED" && status !== "FAILED" && status !== "CANCELED" && retries < 60) {
+            while (status !== "SUCCEEDED" &&
+                status !== "FAILED" &&
+                status !== "CANCELED" &&
+                retries < 60) {
                 await sleep(5000);
-                result = await api(host, token, "GET", `/api/2.0/sql/statements/${statementId}`);
+                result = (await api(host, token, "GET", `/api/2.0/sql/statements/${statementId}`));
                 status = result.status.state;
                 retries++;
             }
@@ -74,61 +109,119 @@ export default function databricks(rl) {
             }
             let chunkIdx = allRows.length > 0 ? 1 : 0;
             while (chunkIdx < totalChunks) {
-                const chunk = await api(host, token, "GET", `/api/2.0/sql/statements/${statementId}/result/chunks/${chunkIdx}`);
+                const chunk = (await api(host, token, "GET", `/api/2.0/sql/statements/${statementId}/result/chunks/${chunkIdx}`));
                 if (chunk.data_array)
                     allRows.push(...chunk.data_array);
                 chunkIdx++;
             }
             // Transform to objects
-            const columns = (manifest?.schema?.columns ?? []);
+            const columns = (manifest?.schema?.columns ??
+                []);
             return allRows.map((row) => {
                 const obj = {};
-                columns.forEach((col, idx) => { obj[col.name] = row[idx]; });
+                columns.forEach((col, idx) => {
+                    obj[col.name] = row[idx];
+                });
                 return obj;
             });
-        } });
+        },
+    });
     // ── Files ───────────────────────────────────────────
-    rl.registerAction("files.createDirectory", { description: "Create a directory in a Unity Catalog volume",
-        inputSchema: { volumePath: { type: "string", required: true, description: "catalog.schema.volume" }, directoryPath: { type: "string", required: true } },
+    rl.registerAction("files.createDirectory", {
+        description: "Create a directory in a Unity Catalog volume",
+        inputSchema: {
+            volumePath: {
+                type: "string",
+                required: true,
+                description: "catalog.schema.volume",
+            },
+            directoryPath: { type: "string", required: true },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             const [cat, sch, vol] = volumeParts(p.volumePath);
             await api(host, token, "PUT", `/api/2.0/fs/directories/Volumes/${cat}/${sch}/${vol}/${p.directoryPath}`);
             return { success: true, directoryPath: p.directoryPath };
-        } });
-    rl.registerAction("files.deleteDirectory", { description: "Delete a directory in a volume",
-        inputSchema: { volumePath: { type: "string", required: true, description: "catalog.schema.volume" }, directoryPath: { type: "string", required: true } },
+        },
+    });
+    rl.registerAction("files.deleteDirectory", {
+        description: "Delete a directory in a volume",
+        inputSchema: {
+            volumePath: {
+                type: "string",
+                required: true,
+                description: "catalog.schema.volume",
+            },
+            directoryPath: { type: "string", required: true },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             const [cat, sch, vol] = volumeParts(p.volumePath);
             await api(host, token, "DELETE", `/api/2.0/fs/directories/Volumes/${cat}/${sch}/${vol}/${p.directoryPath}`);
             return { success: true };
-        } });
-    rl.registerAction("files.deleteFile", { description: "Delete a file in a volume",
-        inputSchema: { volumePath: { type: "string", required: true, description: "catalog.schema.volume" }, filePath: { type: "string", required: true } },
+        },
+    });
+    rl.registerAction("files.deleteFile", {
+        description: "Delete a file in a volume",
+        inputSchema: {
+            volumePath: {
+                type: "string",
+                required: true,
+                description: "catalog.schema.volume",
+            },
+            filePath: { type: "string", required: true },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             const [cat, sch, vol] = volumeParts(p.volumePath);
             await api(host, token, "DELETE", `/api/2.0/fs/files/Volumes/${cat}/${sch}/${vol}/${p.filePath}`);
             return { success: true };
-        } });
-    rl.registerAction("files.getFileInfo", { description: "Get file metadata (content-length, type, last-modified)",
-        inputSchema: { volumePath: { type: "string", required: true, description: "catalog.schema.volume" }, filePath: { type: "string", required: true } },
+        },
+    });
+    rl.registerAction("files.getFileInfo", {
+        description: "Get file metadata (content-length, type, last-modified)",
+        inputSchema: {
+            volumePath: {
+                type: "string",
+                required: true,
+                description: "catalog.schema.volume",
+            },
+            filePath: { type: "string", required: true },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             const [cat, sch, vol] = volumeParts(p.volumePath);
             const url = `${host}/api/2.0/fs/files/Volumes/${cat}/${sch}/${vol}/${p.filePath}`;
-            const res = await fetch(url, { method: "HEAD", headers: { Authorization: `Bearer ${token}` } });
+            const res = await fetch(url, {
+                method: "HEAD",
+                headers: { Authorization: `Bearer ${token}` },
+            });
             if (!res.ok)
                 throw new Error(`Databricks error ${res.status}`);
-            return { filePath: p.filePath, contentLength: res.headers.get("content-length"), contentType: res.headers.get("content-type"), lastModified: res.headers.get("last-modified") };
-        } });
-    rl.registerAction("files.listDirectory", { description: "List files in a volume directory",
-        inputSchema: { volumePath: { type: "string", required: true, description: "catalog.schema.volume" }, directoryPath: { type: "string", required: false }, pageSize: { type: "number", required: false }, pageToken: { type: "string", required: false } },
+            return {
+                filePath: p.filePath,
+                contentLength: res.headers.get("content-length"),
+                contentType: res.headers.get("content-type"),
+                lastModified: res.headers.get("last-modified"),
+            };
+        },
+    });
+    rl.registerAction("files.listDirectory", {
+        description: "List files in a volume directory",
+        inputSchema: {
+            volumePath: {
+                type: "string",
+                required: true,
+                description: "catalog.schema.volume",
+            },
+            directoryPath: { type: "string", required: false },
+            pageSize: { type: "number", required: false },
+            pageToken: { type: "string", required: false },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
@@ -140,61 +233,108 @@ export default function databricks(rl) {
             if (p.pageToken)
                 qs.page_token = p.pageToken;
             return api(host, token, "GET", `/api/2.0/fs/directories/Volumes/${cat}/${sch}/${vol}${dir}`, undefined, qs);
-        } });
+        },
+    });
     // ── Genie ───────────────────────────────────────────
-    rl.registerAction("genie.startConversation", { description: "Start a new Genie conversation",
-        inputSchema: { spaceId: { type: "string", required: true }, initialMessage: { type: "string", required: true } },
+    rl.registerAction("genie.startConversation", {
+        description: "Start a new Genie conversation",
+        inputSchema: {
+            spaceId: { type: "string", required: true },
+            initialMessage: { type: "string", required: true },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             return api(host, token, "POST", `/api/2.0/genie/spaces/${p.spaceId}/start-conversation`, { content: p.initialMessage });
-        } });
-    rl.registerAction("genie.createMessage", { description: "Send a message in an existing Genie conversation",
-        inputSchema: { spaceId: { type: "string", required: true }, conversationId: { type: "string", required: true }, message: { type: "string", required: true } },
+        },
+    });
+    rl.registerAction("genie.createMessage", {
+        description: "Send a message in an existing Genie conversation",
+        inputSchema: {
+            spaceId: { type: "string", required: true },
+            conversationId: { type: "string", required: true },
+            message: { type: "string", required: true },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             return api(host, token, "POST", `/api/2.0/genie/spaces/${p.spaceId}/conversations/${p.conversationId}/messages`, { content: p.message });
-        } });
-    rl.registerAction("genie.getMessage", { description: "Get a specific message from a conversation",
-        inputSchema: { spaceId: { type: "string", required: true }, conversationId: { type: "string", required: true }, messageId: { type: "string", required: true } },
+        },
+    });
+    rl.registerAction("genie.getMessage", {
+        description: "Get a specific message from a conversation",
+        inputSchema: {
+            spaceId: { type: "string", required: true },
+            conversationId: { type: "string", required: true },
+            messageId: { type: "string", required: true },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             return api(host, token, "GET", `/api/2.0/genie/spaces/${p.spaceId}/conversations/${p.conversationId}/messages/${p.messageId}`);
-        } });
-    rl.registerAction("genie.getQueryResults", { description: "Get query results from a message attachment",
-        inputSchema: { spaceId: { type: "string", required: true }, conversationId: { type: "string", required: true }, messageId: { type: "string", required: true }, attachmentId: { type: "string", required: true } },
+        },
+    });
+    rl.registerAction("genie.getQueryResults", {
+        description: "Get query results from a message attachment",
+        inputSchema: {
+            spaceId: { type: "string", required: true },
+            conversationId: { type: "string", required: true },
+            messageId: { type: "string", required: true },
+            attachmentId: { type: "string", required: true },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             return api(host, token, "GET", `/api/2.0/genie/spaces/${p.spaceId}/conversations/${p.conversationId}/messages/${p.messageId}/attachments/${p.attachmentId}/query-result`);
-        } });
-    rl.registerAction("genie.executeMessageQuery", { description: "Execute a query from a message attachment",
-        inputSchema: { spaceId: { type: "string", required: true }, conversationId: { type: "string", required: true }, messageId: { type: "string", required: true }, attachmentId: { type: "string", required: true } },
+        },
+    });
+    rl.registerAction("genie.executeMessageQuery", {
+        description: "Execute a query from a message attachment",
+        inputSchema: {
+            spaceId: { type: "string", required: true },
+            conversationId: { type: "string", required: true },
+            messageId: { type: "string", required: true },
+            attachmentId: { type: "string", required: true },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             return api(host, token, "POST", `/api/2.0/genie/spaces/${p.spaceId}/conversations/${p.conversationId}/messages/${p.messageId}/attachments/${p.attachmentId}/execute-query`);
-        } });
-    rl.registerAction("genie.getSpace", { description: "Get a Genie space",
+        },
+    });
+    rl.registerAction("genie.getSpace", {
+        description: "Get a Genie space",
         inputSchema: { spaceId: { type: "string", required: true } },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             return api(host, token, "GET", `/api/2.0/genie/spaces/${input.spaceId}`);
-        } });
+        },
+    });
     // ── Model Serving ───────────────────────────────────
-    rl.registerAction("modelServing.queryEndpoint", { description: "Query a model serving endpoint",
-        inputSchema: { endpointName: { type: "string", required: true }, requestBody: { type: "object", required: true, description: "JSON body matching the endpoint schema (chat, completions, embeddings, dataframe, etc.)" } },
+    rl.registerAction("modelServing.queryEndpoint", {
+        description: "Query a model serving endpoint",
+        inputSchema: {
+            endpointName: { type: "string", required: true },
+            requestBody: {
+                type: "object",
+                required: true,
+                description: "JSON body matching the endpoint schema (chat, completions, embeddings, dataframe, etc.)",
+            },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             const body = p.requestBody;
             return api(host, token, "POST", `/serving-endpoints/${p.endpointName}/invocations`, body);
-        } });
+        },
+    });
     // ── Unity Catalog: Catalogs ─────────────────────────
-    rl.registerAction("catalog.create", { description: "Create a Unity Catalog catalog",
-        inputSchema: { name: { type: "string", required: true }, comment: { type: "string", required: false } },
+    rl.registerAction("catalog.create", {
+        description: "Create a Unity Catalog catalog",
+        inputSchema: {
+            name: { type: "string", required: true },
+            comment: { type: "string", required: false },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
@@ -202,33 +342,98 @@ export default function databricks(rl) {
             if (p.comment)
                 body.comment = p.comment;
             return api(host, token, "POST", "/api/2.1/unity-catalog/catalogs", body);
-        } });
-    rl.registerAction("catalog.get", { description: "Get a catalog", inputSchema: { name: { type: "string", required: true } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); return api(host, token, "GET", `/api/2.1/unity-catalog/catalogs/${input.name}`); } });
-    rl.registerAction("catalog.list", { description: "List all catalogs", inputSchema: {},
-        async execute(_input, ctx) { const { host, token } = getConn(ctx); return api(host, token, "GET", "/api/2.1/unity-catalog/catalogs"); } });
-    rl.registerAction("catalog.update", { description: "Update a catalog's comment",
-        inputSchema: { name: { type: "string", required: true }, comment: { type: "string", required: true } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); const p = input; return api(host, token, "PATCH", `/api/2.1/unity-catalog/catalogs/${p.name}`, { comment: p.comment }); } });
-    rl.registerAction("catalog.delete", { description: "Delete a catalog", inputSchema: { name: { type: "string", required: true } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); await api(host, token, "DELETE", `/api/2.1/unity-catalog/catalogs/${input.name}`); return { success: true }; } });
+        },
+    });
+    rl.registerAction("catalog.get", {
+        description: "Get a catalog",
+        inputSchema: { name: { type: "string", required: true } },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            return api(host, token, "GET", `/api/2.1/unity-catalog/catalogs/${input.name}`);
+        },
+    });
+    rl.registerAction("catalog.list", {
+        description: "List all catalogs",
+        inputSchema: {},
+        async execute(_input, ctx) {
+            const { host, token } = getConn(ctx);
+            return api(host, token, "GET", "/api/2.1/unity-catalog/catalogs");
+        },
+    });
+    rl.registerAction("catalog.update", {
+        description: "Update a catalog's comment",
+        inputSchema: {
+            name: { type: "string", required: true },
+            comment: { type: "string", required: true },
+        },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            const p = input;
+            return api(host, token, "PATCH", `/api/2.1/unity-catalog/catalogs/${p.name}`, { comment: p.comment });
+        },
+    });
+    rl.registerAction("catalog.delete", {
+        description: "Delete a catalog",
+        inputSchema: { name: { type: "string", required: true } },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            await api(host, token, "DELETE", `/api/2.1/unity-catalog/catalogs/${input.name}`);
+            return { success: true };
+        },
+    });
     // ── Unity Catalog: Tables ───────────────────────────
-    rl.registerAction("table.create", { description: "Create an external Delta table",
-        inputSchema: { catalogName: { type: "string", required: true }, schemaName: { type: "string", required: true }, tableName: { type: "string", required: true }, storageLocation: { type: "string", required: true }, columns: { type: "object", required: false, description: "JSON array of column defs" }, comment: { type: "string", required: false } },
+    rl.registerAction("table.create", {
+        description: "Create an external Delta table",
+        inputSchema: {
+            catalogName: { type: "string", required: true },
+            schemaName: { type: "string", required: true },
+            tableName: { type: "string", required: true },
+            storageLocation: { type: "string", required: true },
+            columns: {
+                type: "object",
+                required: false,
+                description: "JSON array of column defs",
+            },
+            comment: { type: "string", required: false },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
-            const body = { catalog_name: p.catalogName, schema_name: p.schemaName, name: p.tableName, table_type: "EXTERNAL", data_source_format: "DELTA", storage_location: p.storageLocation };
+            const body = {
+                catalog_name: p.catalogName,
+                schema_name: p.schemaName,
+                name: p.tableName,
+                table_type: "EXTERNAL",
+                data_source_format: "DELTA",
+                storage_location: p.storageLocation,
+            };
             if (p.columns)
                 body.columns = p.columns;
             if (p.comment)
                 body.comment = p.comment;
             return api(host, token, "POST", "/api/2.1/unity-catalog/tables", body);
-        } });
-    rl.registerAction("table.get", { description: "Get table info", inputSchema: { fullName: { type: "string", required: true, description: "catalog.schema.table" } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); return api(host, token, "GET", `/api/2.1/unity-catalog/tables/${input.fullName}`); } });
-    rl.registerAction("table.list", { description: "List tables",
-        inputSchema: { catalogName: { type: "string", required: false }, schemaName: { type: "string", required: false } },
+        },
+    });
+    rl.registerAction("table.get", {
+        description: "Get table info",
+        inputSchema: {
+            fullName: {
+                type: "string",
+                required: true,
+                description: "catalog.schema.table",
+            },
+        },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            return api(host, token, "GET", `/api/2.1/unity-catalog/tables/${input.fullName}`);
+        },
+    });
+    rl.registerAction("table.list", {
+        description: "List tables",
+        inputSchema: {
+            catalogName: { type: "string", required: false },
+            schemaName: { type: "string", required: false },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = (input ?? {});
@@ -238,29 +443,79 @@ export default function databricks(rl) {
             if (p.schemaName)
                 qs.schema_name = p.schemaName;
             return api(host, token, "GET", "/api/2.1/unity-catalog/tables", undefined, qs);
-        } });
-    rl.registerAction("table.delete", { description: "Delete a table", inputSchema: { fullName: { type: "string", required: true, description: "catalog.schema.table" } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); await api(host, token, "DELETE", `/api/2.1/unity-catalog/tables/${input.fullName}`); return { success: true }; } });
+        },
+    });
+    rl.registerAction("table.delete", {
+        description: "Delete a table",
+        inputSchema: {
+            fullName: {
+                type: "string",
+                required: true,
+                description: "catalog.schema.table",
+            },
+        },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            await api(host, token, "DELETE", `/api/2.1/unity-catalog/tables/${input.fullName}`);
+            return { success: true };
+        },
+    });
     // ── Unity Catalog: Volumes ──────────────────────────
-    rl.registerAction("volume.create", { description: "Create a Unity Catalog volume",
-        inputSchema: { catalogName: { type: "string", required: true }, schemaName: { type: "string", required: true }, volumeName: { type: "string", required: true }, volumeType: { type: "string", required: true, description: "MANAGED or EXTERNAL" }, storageLocation: { type: "string", required: false, description: "Required for EXTERNAL volumes" }, comment: { type: "string", required: false } },
+    rl.registerAction("volume.create", {
+        description: "Create a Unity Catalog volume",
+        inputSchema: {
+            catalogName: { type: "string", required: true },
+            schemaName: { type: "string", required: true },
+            volumeName: { type: "string", required: true },
+            volumeType: {
+                type: "string",
+                required: true,
+                description: "MANAGED or EXTERNAL",
+            },
+            storageLocation: {
+                type: "string",
+                required: false,
+                description: "Required for EXTERNAL volumes",
+            },
+            comment: { type: "string", required: false },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
             if (p.volumeType === "EXTERNAL" && !p.storageLocation)
                 throw new Error("storageLocation required for EXTERNAL volumes");
-            const body = { catalog_name: p.catalogName, schema_name: p.schemaName, name: p.volumeName, volume_type: p.volumeType };
+            const body = {
+                catalog_name: p.catalogName,
+                schema_name: p.schemaName,
+                name: p.volumeName,
+                volume_type: p.volumeType,
+            };
             if (p.storageLocation)
                 body.storage_location = p.storageLocation;
             if (p.comment)
                 body.comment = p.comment;
             return api(host, token, "POST", "/api/2.1/unity-catalog/volumes", body);
-        } });
-    rl.registerAction("volume.get", { description: "Get a volume",
-        inputSchema: { catalogName: { type: "string", required: true }, schemaName: { type: "string", required: true }, volumeName: { type: "string", required: true } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); const p = input; return api(host, token, "GET", `/api/2.1/unity-catalog/volumes/${p.catalogName}.${p.schemaName}.${p.volumeName}`); } });
-    rl.registerAction("volume.list", { description: "List volumes",
-        inputSchema: { catalogName: { type: "string", required: false }, schemaName: { type: "string", required: false } },
+        },
+    });
+    rl.registerAction("volume.get", {
+        description: "Get a volume",
+        inputSchema: {
+            catalogName: { type: "string", required: true },
+            schemaName: { type: "string", required: true },
+            volumeName: { type: "string", required: true },
+        },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            const p = input;
+            return api(host, token, "GET", `/api/2.1/unity-catalog/volumes/${p.catalogName}.${p.schemaName}.${p.volumeName}`);
+        },
+    });
+    rl.registerAction("volume.list", {
+        description: "List volumes",
+        inputSchema: {
+            catalogName: { type: "string", required: false },
+            schemaName: { type: "string", required: false },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = (input ?? {});
@@ -270,26 +525,101 @@ export default function databricks(rl) {
             if (p.schemaName)
                 qs.schema_name = p.schemaName;
             return api(host, token, "GET", "/api/2.1/unity-catalog/volumes", undefined, qs);
-        } });
-    rl.registerAction("volume.delete", { description: "Delete a volume",
-        inputSchema: { catalogName: { type: "string", required: true }, schemaName: { type: "string", required: true }, volumeName: { type: "string", required: true } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); const p = input; await api(host, token, "DELETE", `/api/2.1/unity-catalog/volumes/${p.catalogName}.${p.schemaName}.${p.volumeName}`); return { success: true }; } });
+        },
+    });
+    rl.registerAction("volume.delete", {
+        description: "Delete a volume",
+        inputSchema: {
+            catalogName: { type: "string", required: true },
+            schemaName: { type: "string", required: true },
+            volumeName: { type: "string", required: true },
+        },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            const p = input;
+            await api(host, token, "DELETE", `/api/2.1/unity-catalog/volumes/${p.catalogName}.${p.schemaName}.${p.volumeName}`);
+            return { success: true };
+        },
+    });
     // ── Unity Catalog: Functions ────────────────────────
-    rl.registerAction("function.create", { description: "Create a Unity Catalog function",
-        inputSchema: { catalogName: { type: "string", required: true }, schemaName: { type: "string", required: true }, functionName: { type: "string", required: true }, inputParams: { type: "object", required: true, description: "Array of {name, type_name, type_text?} parameter definitions" }, returnType: { type: "string", required: true, description: "e.g. STRING, INT" }, routineBody: { type: "string", required: true, description: "SQL or EXTERNAL" }, routineDefinition: { type: "string", required: true, description: "The function body" } },
+    rl.registerAction("function.create", {
+        description: "Create a Unity Catalog function",
+        inputSchema: {
+            catalogName: { type: "string", required: true },
+            schemaName: { type: "string", required: true },
+            functionName: { type: "string", required: true },
+            inputParams: {
+                type: "object",
+                required: true,
+                description: "Array of {name, type_name, type_text?} parameter definitions",
+            },
+            returnType: {
+                type: "string",
+                required: true,
+                description: "e.g. STRING, INT",
+            },
+            routineBody: {
+                type: "string",
+                required: true,
+                description: "SQL or EXTERNAL",
+            },
+            routineDefinition: {
+                type: "string",
+                required: true,
+                description: "The function body",
+            },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
-            const params = (Array.isArray(p.inputParams) ? p.inputParams : p.inputParams?.parameters ?? []);
-            const normalizedParams = params.map((param) => ({ ...param, type_text: param.type_text ?? param.type_name, type_json: param.type_json ?? JSON.stringify({ name: param.type_name }) }));
+            const params = (Array.isArray(p.inputParams)
+                ? p.inputParams
+                : (p.inputParams?.parameters ?? []));
+            const normalizedParams = params.map((param) => ({
+                ...param,
+                type_text: param.type_text ?? param.type_name,
+                type_json: param.type_json ?? JSON.stringify({ name: param.type_name }),
+            }));
             return api(host, token, "POST", "/api/2.1/unity-catalog/functions", {
-                function_info: { name: p.functionName, catalog_name: p.catalogName, schema_name: p.schemaName, input_params: { parameters: normalizedParams }, data_type: p.returnType, full_data_type: p.returnType, specific_name: p.functionName, parameter_style: "S", security_type: "DEFINER", sql_data_access: "CONTAINS_SQL", is_deterministic: false, is_null_call: true, routine_body: p.routineBody, routine_definition: p.routineDefinition },
+                function_info: {
+                    name: p.functionName,
+                    catalog_name: p.catalogName,
+                    schema_name: p.schemaName,
+                    input_params: { parameters: normalizedParams },
+                    data_type: p.returnType,
+                    full_data_type: p.returnType,
+                    specific_name: p.functionName,
+                    parameter_style: "S",
+                    security_type: "DEFINER",
+                    sql_data_access: "CONTAINS_SQL",
+                    is_deterministic: false,
+                    is_null_call: true,
+                    routine_body: p.routineBody,
+                    routine_definition: p.routineDefinition,
+                },
             });
-        } });
-    rl.registerAction("function.get", { description: "Get a function", inputSchema: { fullName: { type: "string", required: true, description: "catalog.schema.function_name" } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); return api(host, token, "GET", `/api/2.1/unity-catalog/functions/${input.fullName}`); } });
-    rl.registerAction("function.list", { description: "List functions",
-        inputSchema: { catalogName: { type: "string", required: false }, schemaName: { type: "string", required: false } },
+        },
+    });
+    rl.registerAction("function.get", {
+        description: "Get a function",
+        inputSchema: {
+            fullName: {
+                type: "string",
+                required: true,
+                description: "catalog.schema.function_name",
+            },
+        },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            return api(host, token, "GET", `/api/2.1/unity-catalog/functions/${input.fullName}`);
+        },
+    });
+    rl.registerAction("function.list", {
+        description: "List functions",
+        inputSchema: {
+            catalogName: { type: "string", required: false },
+            schemaName: { type: "string", required: false },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = (input ?? {});
@@ -299,44 +629,148 @@ export default function databricks(rl) {
             if (p.schemaName)
                 qs.schema_name = p.schemaName;
             return api(host, token, "GET", "/api/2.1/unity-catalog/functions", undefined, qs);
-        } });
-    rl.registerAction("function.delete", { description: "Delete a function", inputSchema: { fullName: { type: "string", required: true, description: "catalog.schema.function_name" } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); await api(host, token, "DELETE", `/api/2.1/unity-catalog/functions/${input.fullName}`); return { success: true }; } });
+        },
+    });
+    rl.registerAction("function.delete", {
+        description: "Delete a function",
+        inputSchema: {
+            fullName: {
+                type: "string",
+                required: true,
+                description: "catalog.schema.function_name",
+            },
+        },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            await api(host, token, "DELETE", `/api/2.1/unity-catalog/functions/${input.fullName}`);
+            return { success: true };
+        },
+    });
     // ── Vector Search ───────────────────────────────────
-    rl.registerAction("vectorSearch.createIndex", { description: "Create a vector search index",
-        inputSchema: { indexName: { type: "string", required: true }, endpointName: { type: "string", required: true }, primaryKey: { type: "string", required: true }, indexType: { type: "string", required: true, description: "DELTA_SYNC or DIRECT_ACCESS" }, deltaSyncIndexSpec: { type: "object", required: false, description: "Spec for DELTA_SYNC type" }, directAccessIndexSpec: { type: "object", required: false, description: "Spec for DIRECT_ACCESS type" } },
+    rl.registerAction("vectorSearch.createIndex", {
+        description: "Create a vector search index",
+        inputSchema: {
+            indexName: { type: "string", required: true },
+            endpointName: { type: "string", required: true },
+            primaryKey: { type: "string", required: true },
+            indexType: {
+                type: "string",
+                required: true,
+                description: "DELTA_SYNC or DIRECT_ACCESS",
+            },
+            deltaSyncIndexSpec: {
+                type: "object",
+                required: false,
+                description: "Spec for DELTA_SYNC type",
+            },
+            directAccessIndexSpec: {
+                type: "object",
+                required: false,
+                description: "Spec for DIRECT_ACCESS type",
+            },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
-            const body = { name: p.indexName, endpoint_name: p.endpointName, primary_key: p.primaryKey, index_type: p.indexType };
+            const body = {
+                name: p.indexName,
+                endpoint_name: p.endpointName,
+                primary_key: p.primaryKey,
+                index_type: p.indexType,
+            };
             if (p.indexType === "DELTA_SYNC" && p.deltaSyncIndexSpec)
                 body.delta_sync_index_spec = p.deltaSyncIndexSpec;
             if (p.indexType === "DIRECT_ACCESS" && p.directAccessIndexSpec)
                 body.direct_access_index_spec = p.directAccessIndexSpec;
             return api(host, token, "POST", "/api/2.0/vector-search/indexes", body);
-        } });
-    rl.registerAction("vectorSearch.getIndex", { description: "Get a vector search index", inputSchema: { indexName: { type: "string", required: true } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); return api(host, token, "GET", `/api/2.0/vector-search/indexes/${input.indexName}`); } });
-    rl.registerAction("vectorSearch.listIndexes", { description: "List vector search indexes for an endpoint", inputSchema: { endpointName: { type: "string", required: true } },
-        async execute(input, ctx) { const { host, token } = getConn(ctx); return api(host, token, "GET", "/api/2.0/vector-search/indexes", undefined, { endpoint_name: input.endpointName }); } });
-    rl.registerAction("vectorSearch.queryIndex", { description: "Query a vector search index",
-        inputSchema: { indexName: { type: "string", required: true }, queryType: { type: "string", required: true, description: "text or vector" }, queryText: { type: "string", required: false, description: "For text queries" }, queryVector: { type: "object", required: false, description: "For vector queries — array of numbers" }, numResults: { type: "number", required: false }, columns: { type: "string", required: true, description: "Comma-separated column names to return" }, searchMode: { type: "string", required: false, description: "HYBRID (default), ANN, or EXACT" }, filterExpression: { type: "string", required: false }, scoreThreshold: { type: "number", required: false }, enableReranking: { type: "boolean", required: false }, rerankerModel: { type: "string", required: false }, columnsToRerank: { type: "string", required: false } },
+        },
+    });
+    rl.registerAction("vectorSearch.getIndex", {
+        description: "Get a vector search index",
+        inputSchema: { indexName: { type: "string", required: true } },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            return api(host, token, "GET", `/api/2.0/vector-search/indexes/${input.indexName}`);
+        },
+    });
+    rl.registerAction("vectorSearch.listIndexes", {
+        description: "List vector search indexes for an endpoint",
+        inputSchema: { endpointName: { type: "string", required: true } },
+        async execute(input, ctx) {
+            const { host, token } = getConn(ctx);
+            return api(host, token, "GET", "/api/2.0/vector-search/indexes", undefined, {
+                endpoint_name: input
+                    .endpointName,
+            });
+        },
+    });
+    rl.registerAction("vectorSearch.queryIndex", {
+        description: "Query a vector search index",
+        inputSchema: {
+            indexName: { type: "string", required: true },
+            queryType: {
+                type: "string",
+                required: true,
+                description: "text or vector",
+            },
+            queryText: {
+                type: "string",
+                required: false,
+                description: "For text queries",
+            },
+            queryVector: {
+                type: "object",
+                required: false,
+                description: "For vector queries — array of numbers",
+            },
+            numResults: { type: "number", required: false },
+            columns: {
+                type: "string",
+                required: true,
+                description: "Comma-separated column names to return",
+            },
+            searchMode: {
+                type: "string",
+                required: false,
+                description: "HYBRID (default), ANN, or EXACT",
+            },
+            filterExpression: { type: "string", required: false },
+            scoreThreshold: { type: "number", required: false },
+            enableReranking: { type: "boolean", required: false },
+            rerankerModel: { type: "string", required: false },
+            columnsToRerank: { type: "string", required: false },
+        },
         async execute(input, ctx) {
             const { host, token } = getConn(ctx);
             const p = input;
-            const body = { num_results: p.numResults ?? 10, query_type: p.searchMode ?? "HYBRID" };
+            const body = {
+                num_results: p.numResults ?? 10,
+                query_type: p.searchMode ?? "HYBRID",
+            };
             if (p.queryType === "text")
                 body.query_text = p.queryText;
             else
                 body.query_vector = p.queryVector;
-            body.columns = p.columns.split(",").map((c) => c.trim()).filter(Boolean);
+            body.columns = p.columns
+                .split(",")
+                .map((c) => c.trim())
+                .filter(Boolean);
             if (p.filterExpression)
                 body.filters_json = p.filterExpression;
             if (p.scoreThreshold)
                 body.score_threshold = p.scoreThreshold;
             if (p.enableReranking) {
-                body.reranker = { model: p.rerankerModel ?? "databricks_reranker", parameters: { columns_to_rerank: (p.columnsToRerank || "").split(",").map((c) => c.trim()).filter(Boolean) } };
+                body.reranker = {
+                    model: p.rerankerModel ?? "databricks_reranker",
+                    parameters: {
+                        columns_to_rerank: (p.columnsToRerank || "")
+                            .split(",")
+                            .map((c) => c.trim())
+                            .filter(Boolean),
+                    },
+                };
             }
             return api(host, token, "POST", `/api/2.0/vector-search/indexes/${p.indexName}/query`, body);
-        } });
+        },
+    });
 }