@fabric-harness/databricks 0.16.1 → 1.0.0

package/dist/client.js

@@ -0,0 +1,114 @@
+import { redactText } from '@fabric-harness/sdk';
+function isRetryableStatus(status) {
+    return status === 429 || (status >= 500 && status <= 599);
+}
+function retryDelayMs(response, attempt, base) {
+    const retryAfter = response.headers.get('retry-after');
+    if (retryAfter) {
+        const seconds = Number(retryAfter);
+        if (Number.isFinite(seconds))
+            return seconds * 1000;
+    }
+    return base * 2 ** attempt;
+}
+export class DatabricksRestClient {
+    host;
+    token;
+    fetchImpl;
+    maxRetries;
+    retryBaseDelayMs;
+    signal;
+    extraHeaders;
+    constructor(options) {
+        this.host = options.host.replace(/\/$/, '');
+        this.token = options.token;
+        this.fetchImpl = options.fetchImpl ?? fetch;
+        this.maxRetries = options.maxRetries ?? 3;
+        this.retryBaseDelayMs = options.retryBaseDelayMs ?? 500;
+        this.signal = options.signal;
+        this.extraHeaders = options.extraHeaders;
+    }
+    get(path, query) {
+        return this.request('GET', path, undefined, query);
+    }
+    post(path, body, query) {
+        return this.request('POST', path, body, query);
+    }
+    async request(method, path, body, query = {}) {
+        const url = new URL(`${this.host}${path.startsWith('/') ? path : `/${path}`}`);
+        for (const [key, value] of Object.entries(query))
+            if (value !== undefined)
+                url.searchParams.set(key, String(value));
+        for (let attempt = 0;; attempt += 1) {
+            // Resolve the (possibly rotating) token on every attempt.
+            const token = typeof this.token === 'function' ? await this.token() : this.token;
+            const extra = typeof this.extraHeaders === 'function' ? await this.extraHeaders() : this.extraHeaders;
+            const init = {
+                method,
+                headers: {
+                    authorization: `Bearer ${token}`,
+                    ...(body !== undefined ? { 'content-type': 'application/json' } : {}),
+                    ...extra,
+                },
+                ...(body !== undefined ? { body: JSON.stringify(body) } : {}),
+                ...(this.signal ? { signal: this.signal } : {}),
+            };
+            let response;
+            try {
+                response = await this.fetchImpl(url, init);
+            }
+            catch (error) {
+                // Network error — retry transient failures, but never swallow an abort.
+                if (this.signal?.aborted || attempt >= this.maxRetries)
+                    throw error;
+                await new Promise((resolve) => setTimeout(resolve, this.retryBaseDelayMs * 2 ** attempt));
+                continue;
+            }
+            if (response.ok) {
+                if (response.status === 204)
+                    return undefined;
+                return await response.json();
+            }
+            if (isRetryableStatus(response.status) && attempt < this.maxRetries) {
+                await new Promise((resolve) => setTimeout(resolve, retryDelayMs(response, attempt, this.retryBaseDelayMs)));
+                continue;
+            }
+            const text = redactText(await response.text(), { additionalSecrets: [token] });
+            throw new Error(`databricks: ${response.status} ${response.statusText}: ${text.slice(0, 2000)}`);
+        }
+    }
+}
+export function createDatabricksClient(options) {
+    return new DatabricksRestClient(options);
+}
+export async function waitForDatabricksRun(client, runId, options = {}) {
+    const deadline = Date.now() + (options.timeoutMs ?? 600_000);
+    const interval = options.pollIntervalMs ?? 5_000;
+    let current = await client.get('/api/2.1/jobs/runs/get', { run_id: runId });
+    while (!isTerminalDatabricksRunState(current) && Date.now() < deadline) {
+        await new Promise((resolve) => setTimeout(resolve, interval));
+        current = await client.get('/api/2.1/jobs/runs/get', { run_id: runId });
+    }
+    return current;
+}
+export async function waitForDatabricksStatement(client, statementId, options = {}) {
+    const deadline = Date.now() + (options.timeoutMs ?? 300_000);
+    const interval = options.pollIntervalMs ?? 2_000;
+    let current = await client.get(`/api/2.0/sql/statements/${encodeURIComponent(statementId)}`);
+    while (!isTerminalDatabricksStatementState(current) && Date.now() < deadline) {
+        await new Promise((resolve) => setTimeout(resolve, interval));
+        current = await client.get(`/api/2.0/sql/statements/${encodeURIComponent(statementId)}`);
+    }
+    return current;
+}
+function isTerminalDatabricksRunState(run) {
+    const state = run.state;
+    const lifeCycle = state?.life_cycle_state;
+    return lifeCycle === 'TERMINATED' || lifeCycle === 'SKIPPED' || lifeCycle === 'INTERNAL_ERROR';
+}
+function isTerminalDatabricksStatementState(statement) {
+    const status = statement.status;
+    const state = status?.state;
+    return state === 'SUCCEEDED' || state === 'FAILED' || state === 'CANCELED' || state === 'CLOSED';
+}
+//# sourceMappingURL=client.js.map

package/dist/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AA0BjD,SAAS,iBAAiB,CAAC,MAAc;IACvC,OAAO,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,IAAI,GAAG,IAAI,MAAM,IAAI,GAAG,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,YAAY,CAAC,QAAkB,EAAE,OAAe,EAAE,IAAY;IACrE,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IACvD,IAAI,UAAU,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;QACnC,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,OAAO,GAAG,IAAI,CAAC;IACtD,CAAC;IACD,OAAO,IAAI,GAAG,CAAC,IAAI,OAAO,CAAC;AAC7B,CAAC;AAED,MAAM,OAAO,oBAAoB;IACd,IAAI,CAAS;IACb,KAAK,CAAmC;IACxC,SAAS,CAAe;IACxB,UAAU,CAAS;IACnB,gBAAgB,CAAS;IACzB,MAAM,CAA0B;IAChC,YAAY,CAA0C;IAEvE,YAAY,OAAgC;QAC1C,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,KAAK,CAAC;QAC5C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,GAAG,CAAC;QACxD,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IAC3C,CAAC;IAED,GAAG,CAAI,IAAY,EAAE,KAA6D;QAChF,OAAO,IAAI,CAAC,OAAO,CAAI,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;IACxD,CAAC;IAED,IAAI,CAAI,IAAY,EAAE,IAAc,EAAE,KAA6D;QACjG,OAAO,IAAI,CAAC,OAAO,CAAI,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,OAAO,CAAI,MAAc,EAAE,IAAY,EAAE,IAAc,EAAE,QAA+D,EAAE;QAC9H,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;QAC/E,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC;YAAE,IAAI,KAAK,KAAK,SAAS;gBAAE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAEpH,KAAK,IAAI,OAAO,GAAG,CAAC,GAAI,OAAO,IAAI,CAAC,EAAE,CAAC;YACrC,0DAA0D;YAC1D,MAAM,KAAK,GAAG,OAAO,IAAI,CAAC,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;YACjF,MAAM,KAAK,GAAG,OAAO,IAAI,CAAC,YAAY,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC;YACtG,MAAM,IAAI,GAAgB;gBACxB,MAAM;gBACN,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,KAAK,EAAE;oBAChC,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACrE,GAAG,KAAK;iBACT;gBACD,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7D,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAChD,CAAC;YAEF,IAAI,QAAkB,CAAC;YACvB,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,wEAAwE;gBACxE,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,IAAI,OAAO,IAAI,IAAI,CAAC,UAAU;oBAAE,MAAM,KAAK,CAAC;gBACpE,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,gBAAgB,GAAG,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC;gBAC1F,SAAS;YACX,CAAC;YAED,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAChB,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG;oBAAE,OAAO,SAAc,CAAC;gBACnD,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAO,CAAC;YACpC,CAAC;YAED,IAAI,iBAAiB,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpE,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBAC5G,SAAS;YACX,CAAC;YAED,MAAM,IAAI,GAAG,UAAU,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC/E,MAAM,IAAI,KAAK,CAAC,eAAe,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QACnG,CAAC;IACH,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CAAC,OAAgC;IACrE,OAAO,IAAI,oBAAoB,CAAC,OAAO,CAAC,CAAC;AAC3C,CAAC;AAOD,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAA4B,EAAE,KAAa,EAAE,UAAiC,EAAE;IACzH,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,CAAC;IAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,cAAc,IAAI,KAAK,CAAC;IACjD,IAAI,OAAO,GAAG,MAAM,MAAM,CAAC,GAAG,CAA0B,wBAAwB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IACrG,OAAO,CAAC,4BAA4B,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QACvE,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;QAC9D,OAAO,GAAG,MAAM,MAAM,CAAC,GAAG,CAA0B,wBAAwB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IACnG,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,MAA4B,EAAE,WAAmB,EAAE,UAAiC,EAAE;IACrI,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,CAAC;IAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,cAAc,IAAI,KAAK,CAAC;IACjD,IAAI,OAAO,GAAG,MAAM,MAAM,CAAC,GAAG,CAA0B,2BAA2B,kBAAkB,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;IACtH,OAAO,CAAC,kCAAkC,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC7E,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;QAC9D,OAAO,GAAG,MAAM,MAAM,CAAC,GAAG,CAA0B,2BAA2B,kBAAkB,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;IACpH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,4BAA4B,CAAC,GAA4B;IAChE,MAAM,KAAK,GAAG,GAAG,CAAC,KAAyE,CAAC;IAC5F,MAAM,SAAS,GAAG,KAAK,EAAE,gBAAgB,CAAC;IAC1C,OAAO,SAAS,KAAK,YAAY,IAAI,SAAS,KAAK,SAAS,IAAI,SAAS,KAAK,gBAAgB,CAAC;AACjG,CAAC;AAED,SAAS,kCAAkC,CAAC,SAAkC;IAC5E,MAAM,MAAM,GAAG,SAAS,CAAC,MAAwC,CAAC;IAClE,MAAM,KAAK,GAAG,MAAM,EAAE,KAAK,CAAC;IAC5B,OAAO,KAAK,KAAK,WAAW,IAAI,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,UAAU,IAAI,KAAK,KAAK,QAAQ,CAAC;AACnG,CAAC"}

package/dist/consumption.d.ts

@@ -0,0 +1,45 @@
+import type { JsonObject, ToolDef } from '@fabric-harness/sdk';
+import type { DatabricksRestClient } from './client.js';
+export type ConsumptionGroupBy = 'sku' | 'day';
+export interface ConsumptionSummaryOptions {
+    /** Inclusive start date, `YYYY-MM-DD`. */
+    startDate: string;
+    /** Inclusive end date, `YYYY-MM-DD`. Defaults to `startDate`'s caller-provided window end. */
+    endDate: string;
+    groupBy?: ConsumptionGroupBy;
+}
+export interface ConsumptionSummary {
+    startDate: string;
+    endDate: string;
+    groupBy: ConsumptionGroupBy;
+    rows: JsonObject[];
+    totalDbus: number;
+    totalListCostUsd: number;
+}
+export interface DatabricksConsumptionOptions {
+    warehouseId: string;
+    pollIntervalMs?: number;
+    timeoutMs?: number;
+}
+/**
+ * System-Tables consumption reporting. Reads `system.billing.usage` joined to `list_prices` to report
+ * real DBUs + list cost — the "prove the consumption" artifact for chargeback and partner credit.
+ */
+export declare function databricksConsumption(client: DatabricksRestClient, options: DatabricksConsumptionOptions): {
+    summary(opts: ConsumptionSummaryOptions): Promise<ConsumptionSummary>;
+};
+export type DatabricksConsumption = ReturnType<typeof databricksConsumption>;
+export interface DatabricksConsumptionToolOptions extends DatabricksConsumptionOptions {
+    name?: string;
+    description?: string;
+    /** Clock injection for tests / determinism. */
+    now?: () => Date;
+}
+/** Agent-callable consumption report over the last `days` (default 30) or an explicit date window. */
+export declare function databricksConsumptionTool(client: DatabricksRestClient, options: DatabricksConsumptionToolOptions): ToolDef<{
+    days?: number;
+    startDate?: string;
+    endDate?: string;
+    groupBy?: ConsumptionGroupBy;
+}, unknown>;
+//# sourceMappingURL=consumption.d.ts.map

package/dist/consumption.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"consumption.d.ts","sourceRoot":"","sources":["../src/consumption.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAGxD,MAAM,MAAM,kBAAkB,GAAG,KAAK,GAAG,KAAK,CAAC;AAQ/C,MAAM,WAAW,yBAAyB;IACxC,0CAA0C;IAC1C,SAAS,EAAE,MAAM,CAAC;IAClB,8FAA8F;IAC9F,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,kBAAkB,CAAC;CAC9B;AAED,MAAM,WAAW,kBAAkB;IACjC,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,kBAAkB,CAAC;IAC5B,IAAI,EAAE,UAAU,EAAE,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,4BAA4B;IAC3C,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAwBD;;;GAGG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,EAAE,4BAA4B;kBAGjF,yBAAyB,GAAG,OAAO,CAAC,kBAAkB,CAAC;EAuB9E;AAED,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAE7E,MAAM,WAAW,gCAAiC,SAAQ,4BAA4B;IACpF,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,+CAA+C;IAC/C,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC;CAClB;AAMD,sGAAsG;AACtG,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,EAAE,gCAAgC,GAAG,OAAO,CAAC;IAAE,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,CAAC;IAAC,OAAO,CAAC,EAAE,kBAAkB,CAAA;CAAE,EAAE,OAAO,CAAC,CA4B1N"}

package/dist/consumption.js

@@ -0,0 +1,85 @@
+import { parseStatementRows, runStatement } from './sql-exec.js';
+// Allowlisted group expressions — never interpolated from user/model input.
+const GROUP_COLUMNS = {
+    sku: { select: 'u.sku_name AS sku, u.usage_unit AS unit', group: 'u.sku_name, u.usage_unit' },
+    day: { select: 'u.usage_date AS day', group: 'u.usage_date' },
+};
+function buildQuery(groupBy) {
+    const g = GROUP_COLUMNS[groupBy];
+    return [
+        `SELECT ${g.select},`,
+        '       SUM(u.usage_quantity) AS quantity,',
+        '       SUM(u.usage_quantity * lp.pricing.default) AS list_cost_usd',
+        'FROM system.billing.usage u',
+        'LEFT JOIN system.billing.list_prices lp',
+        '  ON u.cloud = lp.cloud AND u.sku_name = lp.sku_name AND u.usage_unit = lp.usage_unit',
+        ' AND u.usage_end_time >= lp.price_start_time',
+        ' AND (lp.price_end_time IS NULL OR u.usage_end_time < lp.price_end_time)',
+        'WHERE u.usage_date >= :start_date AND u.usage_date <= :end_date',
+        `GROUP BY ${g.group}`,
+        'ORDER BY list_cost_usd DESC',
+    ].join('\n');
+}
+function toNumber(value) {
+    const n = typeof value === 'number' ? value : Number(value);
+    return Number.isFinite(n) ? n : 0;
+}
+/**
+ * System-Tables consumption reporting. Reads `system.billing.usage` joined to `list_prices` to report
+ * real DBUs + list cost — the "prove the consumption" artifact for chargeback and partner credit.
+ */
+export function databricksConsumption(client, options) {
+    const poll = { intervalMs: options.pollIntervalMs ?? 2000, timeoutMs: options.timeoutMs ?? 120_000 };
+    return {
+        async summary(opts) {
+            const groupBy = opts.groupBy ?? 'sku';
+            const response = await runStatement(client, options.warehouseId, buildQuery(groupBy), [
+                { name: 'start_date', value: opts.startDate, type: 'DATE' },
+                { name: 'end_date', value: opts.endDate, type: 'DATE' },
+            ], poll);
+            const rows = parseStatementRows(response);
+            return {
+                startDate: opts.startDate,
+                endDate: opts.endDate,
+                groupBy,
+                rows,
+                totalDbus: rows.reduce((sum, r) => sum + toNumber(r.quantity), 0),
+                totalListCostUsd: rows.reduce((sum, r) => sum + toNumber(r.list_cost_usd), 0),
+            };
+        },
+    };
+}
+function isoDate(date) {
+    return date.toISOString().slice(0, 10);
+}
+/** Agent-callable consumption report over the last `days` (default 30) or an explicit date window. */
+export function databricksConsumptionTool(client, options) {
+    const reporter = databricksConsumption(client, options);
+    const now = options.now ?? (() => new Date());
+    return {
+        name: options.name ?? 'databricks_consumption',
+        description: options.description ?? 'Report real Databricks consumption (DBUs and list cost) from system.billing over a date window.',
+        inputSchema: {
+            type: 'object',
+            additionalProperties: false,
+            properties: {
+                days: { type: 'number', minimum: 1 },
+                startDate: { type: 'string' },
+                endDate: { type: 'string' },
+                groupBy: { type: 'string', enum: ['sku', 'day'] },
+            },
+        },
+        metadata: { effect: 'read', provider: 'databricks', service: 'system-tables' },
+        execute: (input) => {
+            const endDate = input.endDate ?? isoDate(now());
+            let startDate = input.startDate;
+            if (!startDate) {
+                const start = now();
+                start.setUTCDate(start.getUTCDate() - (input.days ?? 30));
+                startDate = isoDate(start);
+            }
+            return reporter.summary({ startDate, endDate, ...(input.groupBy ? { groupBy: input.groupBy } : {}) });
+        },
+    };
+}
+//# sourceMappingURL=consumption.js.map

package/dist/consumption.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"consumption.js","sourceRoot":"","sources":["../src/consumption.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAIjE,4EAA4E;AAC5E,MAAM,aAAa,GAAkE;IACnF,GAAG,EAAE,EAAE,MAAM,EAAE,yCAAyC,EAAE,KAAK,EAAE,0BAA0B,EAAE;IAC7F,GAAG,EAAE,EAAE,MAAM,EAAE,qBAAqB,EAAE,KAAK,EAAE,cAAc,EAAE;CAC9D,CAAC;AAyBF,SAAS,UAAU,CAAC,OAA2B;IAC7C,MAAM,CAAC,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACjC,OAAO;QACL,UAAU,CAAC,CAAC,MAAM,GAAG;QACrB,2CAA2C;QAC3C,oEAAoE;QACpE,6BAA6B;QAC7B,yCAAyC;QACzC,uFAAuF;QACvF,8CAA8C;QAC9C,0EAA0E;QAC1E,iEAAiE;QACjE,YAAY,CAAC,CAAC,KAAK,EAAE;QACrB,6BAA6B;KAC9B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,MAAM,CAAC,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC5D,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACpC,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,qBAAqB,CAAC,MAA4B,EAAE,OAAqC;IACvG,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,OAAO,CAAC,cAAc,IAAI,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,EAAE,CAAC;IACrG,OAAO;QACL,KAAK,CAAC,OAAO,CAAC,IAA+B;YAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,KAAK,CAAC;YACtC,MAAM,QAAQ,GAAG,MAAM,YAAY,CACjC,MAAM,EACN,OAAO,CAAC,WAAW,EACnB,UAAU,CAAC,OAAO,CAAC,EACnB;gBACE,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC3D,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE;aACxD,EACD,IAAI,CACL,CAAC;YACF,MAAM,IAAI,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAC1C,OAAO;gBACL,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,OAAO;gBACP,IAAI;gBACJ,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACjE,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;aAC9E,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAWD,SAAS,OAAO,CAAC,IAAU;IACzB,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACzC,CAAC;AAED,sGAAsG;AACtG,MAAM,UAAU,yBAAyB,CAAC,MAA4B,EAAE,OAAyC;IAC/G,MAAM,QAAQ,GAAG,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACxD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IAC9C,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,wBAAwB;QAC9C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,iGAAiG;QACrI,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE;gBACpC,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC7B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE;aAClD;SACF;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,eAAe,EAAE;QAC9E,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACjB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;YAChD,IAAI,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;YAChC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC;gBACpB,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,UAAU,EAAE,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC1D,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;YAC7B,CAAC;YACD,OAAO,QAAQ,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACxG,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/cost-budget-actuals.d.ts

@@ -0,0 +1,27 @@
+import type { ActualCostSource, CostLimit } from '@fabric-harness/sdk';
+import type { DatabricksRestClient } from './client.js';
+import type { StatementPoll } from './sql-exec.js';
+export interface DatabricksActualCostSourceOptions {
+    /** Cache TTL in milliseconds. Default 60_000. */
+    cacheTtlMs?: number;
+    /** Tag key prefix for filtering usage rows. Default `'fabric'`. */
+    tagPrefix?: string;
+    /** Polling configuration for statement execution. */
+    poll?: StatementPoll;
+}
+/**
+ * Create a Databricks-backed `ActualCostSource` that queries `system.billing.usage`
+ * joined to `system.billing.list_prices` for real spend data.
+ *
+ * Results are cached per-scope-key for `cacheTtlMs` (default 60s) to avoid
+ * hammering the SQL warehouse.
+ */
+export declare function databricksActualCostSource(client: DatabricksRestClient, warehouseId: string, options?: DatabricksActualCostSourceOptions): ActualCostSource;
+export declare function databricksTenantCostLimit(client: DatabricksRestClient, warehouseId: string, tenantId: string, options: {
+    perHourUsd?: number;
+    perDayUsd?: number;
+    perMonthUsd?: number;
+    cacheTtlMs?: number;
+    onExceed?: 'throw' | 'approve';
+}): CostLimit;
+//# sourceMappingURL=cost-budget-actuals.d.ts.map

package/dist/cost-budget-actuals.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-budget-actuals.d.ts","sourceRoot":"","sources":["../src/cost-budget-actuals.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,gBAAgB,EAKhB,SAAS,EACV,MAAM,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAExD,OAAO,KAAK,EAAgB,aAAa,EAAE,MAAM,eAAe,CAAC;AAEjE,MAAM,WAAW,iCAAiC;IAChD,iDAAiD;IACjD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,mEAAmE;IACnE,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,qDAAqD;IACrD,IAAI,CAAC,EAAE,aAAa,CAAC;CACtB;AAiLD;;;;;;GAMG;AACH,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,oBAAoB,EAC5B,WAAW,EAAE,MAAM,EACnB,OAAO,GAAE,iCAAsC,GAC9C,gBAAgB,CAoHlB;AAED,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,oBAAoB,EAC5B,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE;IACP,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CAChC,GACA,SAAS,CAcX"}

package/dist/cost-budget-actuals.js

@@ -0,0 +1,265 @@
+import { parseScopeKey, tenantCostLimit } from '@fabric-harness/sdk';
+import { runStatement, parseStatementRows } from './sql-exec.js';
+function toNumber(value) {
+    const n = typeof value === 'number' ? value : Number(value);
+    return Number.isFinite(n) ? n : 0;
+}
+function deriveDateWindow(parsed) {
+    const period = parsed.period;
+    if (!period)
+        return undefined;
+    const parts = period.split(':');
+    const periodType = parts[0];
+    const periodValue = parts.slice(1).join(':');
+    if (!periodValue)
+        return undefined;
+    switch (periodType) {
+        case 'day': {
+            // day:YYYY-MM-DD
+            return { startDate: periodValue, endDate: periodValue };
+        }
+        case 'hour': {
+            // hour:YYYY-MM-DDTHH:00Z → use the date portion
+            const date = periodValue.match(/^(\d{4}-\d{2}-\d{2})/)?.[1];
+            return date ? { startDate: date, endDate: date } : undefined;
+        }
+        case 'month': {
+            // month:YYYY-MM
+            const [year, month] = periodValue.split('-');
+            if (!year || !month)
+                return undefined;
+            const lastDay = new Date(Number(year), Number(month), 0).getDate();
+            return {
+                startDate: `${year}-${month}-01`,
+                endDate: `${year}-${month}-${String(lastDay).padStart(2, '0')}`,
+            };
+        }
+        default:
+            return undefined;
+    }
+}
+function buildCostQuery(tagPrefix, attribution) {
+    const tagAgent = `${tagPrefix}.agent`;
+    const tagTenant = `${tagPrefix}.tenant`;
+    const tagUser = `${tagPrefix}.user`;
+    const conditions = [
+        'u.usage_date >= :start_date AND u.usage_date <= :end_date',
+    ];
+    if (attribution?.agentId) {
+        conditions.push(`u.tags['${tagAgent}'] = :agent_id`);
+    }
+    if (attribution?.tenantId) {
+        conditions.push(`u.tags['${tagTenant}'] = :tenant_id`);
+    }
+    if (attribution?.userId) {
+        conditions.push(`u.tags['${tagUser}'] = :user_id`);
+    }
+    return [
+        'SELECT',
+        '  SUM(u.usage_quantity * lp.pricing.default) AS list_cost_usd,',
+        '  SUM(u.usage_quantity) AS quantity',
+        'FROM system.billing.usage u',
+        'LEFT JOIN system.billing.list_prices lp',
+        '  ON u.cloud = lp.cloud AND u.sku_name = lp.sku_name AND u.usage_unit = lp.usage_unit',
+        ' AND u.usage_end_time >= lp.price_start_time',
+        ' AND (lp.price_end_time IS NULL OR u.usage_end_time < lp.price_end_time)',
+        `WHERE ${conditions.join(' AND ')}`,
+    ].join('\n');
+}
+function buildAttributionQuery(tagPrefix, filters) {
+    const tagAgent = `${tagPrefix}.agent`;
+    const tagTenant = `${tagPrefix}.tenant`;
+    const tagUser = `${tagPrefix}.user`;
+    const conditions = [
+        'u.usage_date >= :start_date AND u.usage_date <= :end_date',
+    ];
+    const parameters = [
+        { name: 'start_date', value: filters.startDate ?? '', type: 'DATE' },
+        { name: 'end_date', value: filters.endDate ?? '', type: 'DATE' },
+    ];
+    if (filters.agentId) {
+        conditions.push(`u.tags['${tagAgent}'] = :agent_id`);
+        parameters.push({ name: 'agent_id', value: filters.agentId });
+    }
+    if (filters.tenantId) {
+        conditions.push(`u.tags['${tagTenant}'] = :tenant_id`);
+        parameters.push({ name: 'tenant_id', value: filters.tenantId });
+    }
+    if (filters.userId) {
+        conditions.push(`u.tags['${tagUser}'] = :user_id`);
+        parameters.push({ name: 'user_id', value: filters.userId });
+    }
+    if (filters.model) {
+        conditions.push('u.sku_name = :model');
+        parameters.push({ name: 'model', value: filters.model });
+    }
+    if (filters.provider) {
+        conditions.push('u.cloud = :provider');
+        parameters.push({ name: 'provider', value: filters.provider });
+    }
+    const groupBy = filters.groupBy ?? ['agentId', 'tenantId', 'userId', 'model', 'provider'];
+    const groupColumns = [];
+    const selectColumns = [];
+    for (const dim of groupBy) {
+        switch (dim) {
+            case 'agentId':
+                selectColumns.push(`u.tags['${tagAgent}'] AS agentId`);
+                groupColumns.push(`u.tags['${tagAgent}']`);
+                break;
+            case 'tenantId':
+                selectColumns.push(`u.tags['${tagTenant}'] AS tenantId`);
+                groupColumns.push(`u.tags['${tagTenant}']`);
+                break;
+            case 'userId':
+                selectColumns.push(`u.tags['${tagUser}'] AS userId`);
+                groupColumns.push(`u.tags['${tagUser}']`);
+                break;
+            case 'model':
+                selectColumns.push('u.sku_name AS model');
+                groupColumns.push('u.sku_name');
+                break;
+            case 'provider':
+                selectColumns.push('u.cloud AS provider');
+                groupColumns.push('u.cloud');
+                break;
+            case 'sessionId':
+                // sessionId is not stored in billing tags; skip as a group dimension
+                break;
+            case 'turnId':
+                // turnId is not stored in billing tags; skip as a group dimension
+                break;
+        }
+    }
+    selectColumns.push("CONCAT('day:', u.usage_date) AS period", 'SUM(u.usage_quantity * lp.pricing.default) AS actualUsd', 'SUM(u.usage_quantity) AS actualDbus', 'COUNT(*) AS calls');
+    groupColumns.push("CONCAT('day:', u.usage_date)");
+    const query = [
+        'SELECT',
+        `  ${selectColumns.join(',\n  ')}`,
+        'FROM system.billing.usage u',
+        'LEFT JOIN system.billing.list_prices lp',
+        '  ON u.cloud = lp.cloud AND u.sku_name = lp.sku_name AND u.usage_unit = lp.usage_unit',
+        ' AND u.usage_end_time >= lp.price_start_time',
+        ' AND (lp.price_end_time IS NULL OR u.usage_end_time < lp.price_end_time)',
+        `WHERE ${conditions.join(' AND ')}`,
+        `GROUP BY ${groupColumns.join(', ')}`,
+        'ORDER BY actualUsd DESC',
+    ].join('\n');
+    return { query, parameters };
+}
+/**
+ * Create a Databricks-backed `ActualCostSource` that queries `system.billing.usage`
+ * joined to `system.billing.list_prices` for real spend data.
+ *
+ * Results are cached per-scope-key for `cacheTtlMs` (default 60s) to avoid
+ * hammering the SQL warehouse.
+ */
+export function databricksActualCostSource(client, warehouseId, options = {}) {
+    const cacheTtlMs = options.cacheTtlMs ?? 60_000;
+    const tagPrefix = options.tagPrefix ?? 'fabric';
+    const poll = options.poll ?? { intervalMs: 2000, timeoutMs: 120_000 };
+    const cache = new Map();
+    return {
+        async getUsd(scopeKey, attribution, queryOptions) {
+            const ttl = queryOptions?.cacheTtlMs ?? cacheTtlMs;
+            const now = Date.now();
+            // Check cache first
+            const cached = cache.get(scopeKey);
+            if (cached && cached.expiresAt > now) {
+                const result = {
+                    usd: cached.usd,
+                };
+                if (cached.dbus !== undefined)
+                    result.dbus = cached.dbus;
+                result.actualsFetchedAt = cached.fetchedAt;
+                result.cacheTtlMs = ttl;
+                return result;
+            }
+            const parsed = parseScopeKey(scopeKey);
+            if (!parsed) {
+                return { usd: 0, cacheTtlMs: ttl };
+            }
+            const dateWindow = deriveDateWindow(parsed);
+            if (!dateWindow) {
+                return { usd: 0, cacheTtlMs: ttl };
+            }
+            const parameters = [
+                { name: 'start_date', value: dateWindow.startDate, type: 'DATE' },
+                { name: 'end_date', value: dateWindow.endDate, type: 'DATE' },
+            ];
+            if (attribution?.agentId) {
+                parameters.push({ name: 'agent_id', value: attribution.agentId });
+            }
+            if (attribution?.tenantId) {
+                parameters.push({ name: 'tenant_id', value: attribution.tenantId });
+            }
+            if (attribution?.userId) {
+                parameters.push({ name: 'user_id', value: attribution.userId });
+            }
+            const response = await runStatement(client, warehouseId, buildCostQuery(tagPrefix, attribution), parameters, poll);
+            const rows = parseStatementRows(response);
+            const usd = rows.reduce((sum, r) => sum + toNumber(r.list_cost_usd), 0);
+            const dbus = rows.reduce((sum, r) => sum + toNumber(r.quantity), 0);
+            const fetchedAt = new Date().toISOString();
+            cache.set(scopeKey, {
+                usd,
+                dbus: dbus || undefined,
+                fetchedAt,
+                expiresAt: now + ttl,
+            });
+            const result = {
+                usd,
+            };
+            if (dbus)
+                result.dbus = dbus;
+            result.actualsFetchedAt = fetchedAt;
+            result.cacheTtlMs = ttl;
+            return result;
+        },
+        async queryAttribution(filters) {
+            if (!filters.startDate || !filters.endDate) {
+                throw new Error('queryAttribution requires startDate and endDate');
+            }
+            const { query, parameters } = buildAttributionQuery(tagPrefix, filters);
+            const response = await runStatement(client, warehouseId, query, parameters, poll);
+            const rows = parseStatementRows(response);
+            return rows.map((r) => {
+                const row = {
+                    period: String(r.period ?? ''),
+                    estimatedUsd: 0,
+                    actualUsd: toNumber(r.actualUsd),
+                    actualDbus: toNumber(r.actualDbus),
+                    deltaUsd: toNumber(r.actualUsd),
+                    calls: toNumber(r.calls),
+                };
+                if (r.agentId)
+                    row.agentId = String(r.agentId);
+                if (r.userId)
+                    row.userId = String(r.userId);
+                if (r.tenantId)
+                    row.tenantId = String(r.tenantId);
+                if (r.model)
+                    row.model = String(r.model);
+                if (r.provider)
+                    row.provider = String(r.provider);
+                return row;
+            });
+        },
+    };
+}
+export function databricksTenantCostLimit(client, warehouseId, tenantId, options) {
+    const base = tenantCostLimit(tenantId, {
+        ...(options.perHourUsd !== undefined ? { perHourUsd: options.perHourUsd } : {}),
+        ...(options.perDayUsd !== undefined ? { perDayUsd: options.perDayUsd } : {}),
+        ...(options.perMonthUsd !== undefined ? { perMonthUsd: options.perMonthUsd } : {}),
+        ...(options.onExceed ? { onExceed: options.onExceed } : {}),
+    });
+    const opts = {};
+    if (options.cacheTtlMs !== undefined)
+        opts.cacheTtlMs = options.cacheTtlMs;
+    return {
+        ...base,
+        scopeSource: 'external',
+        actualSource: databricksActualCostSource(client, warehouseId, opts),
+    };
+}
+//# sourceMappingURL=cost-budget-actuals.js.map

package/dist/cost-budget-actuals.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-budget-actuals.js","sourceRoot":"","sources":["../src/cost-budget-actuals.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAErE,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAmBjE,SAAS,QAAQ,CAAC,KAAc;IAC9B,MAAM,CAAC,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC5D,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAsB;IAC9C,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC7B,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAE9B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAChC,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAC5B,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAE7C,IAAI,CAAC,WAAW;QAAE,OAAO,SAAS,CAAC;IAEnC,QAAQ,UAAU,EAAE,CAAC;QACnB,KAAK,KAAK,CAAC,CAAC,CAAC;YACX,iBAAiB;YACjB,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;QAC1D,CAAC;QACD,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,gDAAgD;YAChD,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC5D,OAAO,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/D,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,gBAAgB;YAChB,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC7C,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,SAAS,CAAC;YACtC,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACnE,OAAO;gBACL,SAAS,EAAE,GAAG,IAAI,IAAI,KAAK,KAAK;gBAChC,OAAO,EAAE,GAAG,IAAI,IAAI,KAAK,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE;aAChE,CAAC;QACJ,CAAC;QACD;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,SAAiB,EAAE,WAA6B;IACtE,MAAM,QAAQ,GAAG,GAAG,SAAS,QAAQ,CAAC;IACtC,MAAM,SAAS,GAAG,GAAG,SAAS,SAAS,CAAC;IACxC,MAAM,OAAO,GAAG,GAAG,SAAS,OAAO,CAAC;IAEpC,MAAM,UAAU,GAAa;QAC3B,2DAA2D;KAC5D,CAAC;IAEF,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;QACzB,UAAU,CAAC,IAAI,CAAC,WAAW,QAAQ,gBAAgB,CAAC,CAAC;IACvD,CAAC;IACD,IAAI,WAAW,EAAE,QAAQ,EAAE,CAAC;QAC1B,UAAU,CAAC,IAAI,CAAC,WAAW,SAAS,iBAAiB,CAAC,CAAC;IACzD,CAAC;IACD,IAAI,WAAW,EAAE,MAAM,EAAE,CAAC;QACxB,UAAU,CAAC,IAAI,CAAC,WAAW,OAAO,eAAe,CAAC,CAAC;IACrD,CAAC;IAED,OAAO;QACL,QAAQ;QACR,gEAAgE;QAChE,qCAAqC;QACrC,6BAA6B;QAC7B,yCAAyC;QACzC,uFAAuF;QACvF,8CAA8C;QAC9C,0EAA0E;QAC1E,SAAS,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;KACpC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,qBAAqB,CAC5B,SAAiB,EACjB,OAAyB;IAEzB,MAAM,QAAQ,GAAG,GAAG,SAAS,QAAQ,CAAC;IACtC,MAAM,SAAS,GAAG,GAAG,SAAS,SAAS,CAAC;IACxC,MAAM,OAAO,GAAG,GAAG,SAAS,OAAO,CAAC;IAEpC,MAAM,UAAU,GAAa;QAC3B,2DAA2D;KAC5D,CAAC;IACF,MAAM,UAAU,GAAmB;QACjC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,OAAO,CAAC,SAAS,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;QACpE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;KACjE,CAAC;IAEF,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,UAAU,CAAC,IAAI,CAAC,WAAW,QAAQ,gBAAgB,CAAC,CAAC;QACrD,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAChE,CAAC;IACD,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACrB,UAAU,CAAC,IAAI,CAAC,WAAW,SAAS,iBAAiB,CAAC,CAAC;QACvD,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;IAClE,CAAC;IACD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,UAAU,CAAC,IAAI,CAAC,WAAW,OAAO,eAAe,CAAC,CAAC;QACnD,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC9D,CAAC;IACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QAClB,UAAU,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACvC,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;IAC3D,CAAC;IACD,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACrB,UAAU,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACvC,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;IAC1F,MAAM,YAAY,GAAa,EAAE,CAAC;IAClC,MAAM,aAAa,GAAa,EAAE,CAAC;IAEnC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;QAC1B,QAAQ,GAAG,EAAE,CAAC;YACZ,KAAK,SAAS;gBACZ,aAAa,CAAC,IAAI,CAAC,WAAW,QAAQ,eAAe,CAAC,CAAC;gBACvD,YAAY,CAAC,IAAI,CAAC,WAAW,QAAQ,IAAI,CAAC,CAAC;gBAC3C,MAAM;YACR,KAAK,UAAU;gBACb,aAAa,CAAC,IAAI,CAAC,WAAW,SAAS,gBAAgB,CAAC,CAAC;gBACzD,YAAY,CAAC,IAAI,CAAC,WAAW,SAAS,IAAI,CAAC,CAAC;gBAC5C,MAAM;YACR,KAAK,QAAQ;gBACX,aAAa,CAAC,IAAI,CAAC,WAAW,OAAO,cAAc,CAAC,CAAC;gBACrD,YAAY,CAAC,IAAI,CAAC,WAAW,OAAO,IAAI,CAAC,CAAC;gBAC1C,MAAM;YACR,KAAK,OAAO;gBACV,aAAa,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;gBAC1C,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAChC,MAAM;YACR,KAAK,UAAU;gBACb,aAAa,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;gBAC1C,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC7B,MAAM;YACR,KAAK,WAAW;gBACd,qEAAqE;gBACrE,MAAM;YACR,KAAK,QAAQ;gBACX,kEAAkE;gBAClE,MAAM;QACV,CAAC;IACH,CAAC;IAED,aAAa,CAAC,IAAI,CAChB,wCAAwC,EACxC,yDAAyD,EACzD,qCAAqC,EACrC,mBAAmB,CACpB,CAAC;IACF,YAAY,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;IAElD,MAAM,KAAK,GAAG;QACZ,QAAQ;QACR,KAAK,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;QAClC,6BAA6B;QAC7B,yCAAyC;QACzC,uFAAuF;QACvF,8CAA8C;QAC9C,0EAA0E;QAC1E,SAAS,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;QACnC,YAAY,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;QACrC,yBAAyB;KAC1B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEb,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;AAC/B,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,0BAA0B,CACxC,MAA4B,EAC5B,WAAmB,EACnB,UAA6C,EAAE;IAE/C,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,MAAM,CAAC;IAChD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,QAAQ,CAAC;IAChD,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;IACtE,MAAM,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;IAE5C,OAAO;QACL,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,WAAW,EAAE,YAAY;YAC9C,MAAM,GAAG,GAAG,YAAY,EAAE,UAAU,IAAI,UAAU,CAAC;YACnD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEvB,oBAAoB;YACpB,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACnC,IAAI,MAAM,IAAI,MAAM,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;gBACrC,MAAM,MAAM,GAAmF;oBAC7F,GAAG,EAAE,MAAM,CAAC,GAAG;iBAChB,CAAC;gBACF,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS;oBAAE,MAAM,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;gBACzD,MAAM,CAAC,gBAAgB,GAAG,MAAM,CAAC,SAAS,CAAC;gBAC3C,MAAM,CAAC,UAAU,GAAG,GAAG,CAAC;gBACxB,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;YACrC,CAAC;YAED,MAAM,UAAU,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC5C,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;YACrC,CAAC;YAED,MAAM,UAAU,GAAmB;gBACjC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,CAAC,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE;gBACjE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE;aAC9D,CAAC;YAEF,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;gBACzB,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC;YACpE,CAAC;YACD,IAAI,WAAW,EAAE,QAAQ,EAAE,CAAC;gBAC1B,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,CAAC,QAAQ,EAAE,CAAC,CAAC;YACtE,CAAC;YACD,IAAI,WAAW,EAAE,MAAM,EAAE,CAAC;gBACxB,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;YAClE,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,YAAY,CACjC,MAAM,EACN,WAAW,EACX,cAAc,CAAC,SAAS,EAAE,WAAW,CAAC,EACtC,UAAU,EACV,IAAI,CACL,CAAC;YAEF,MAAM,IAAI,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;YACxE,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;YACpE,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAE3C,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE;gBAClB,GAAG;gBACH,IAAI,EAAE,IAAI,IAAI,SAAS;gBACvB,SAAS;gBACT,SAAS,EAAE,GAAG,GAAG,GAAG;aACrB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAmF;gBAC7F,GAAG;aACJ,CAAC;YACF,IAAI,IAAI;gBAAE,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;YAC7B,MAAM,CAAC,gBAAgB,GAAG,SAAS,CAAC;YACpC,MAAM,CAAC,UAAU,GAAG,GAAG,CAAC;YACxB,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,KAAK,CAAC,gBAAgB,CAAC,OAAO;YAC5B,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACrE,CAAC;YAED,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,qBAAqB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACxE,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;YAClF,MAAM,IAAI,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAE1C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACpB,MAAM,GAAG,GAYL;oBACF,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,IAAI,EAAE,CAAC;oBAC9B,YAAY,EAAE,CAAC;oBACf,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;oBAChC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC;oBAClC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;oBAC/B,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC;iBACzB,CAAC;gBACF,IAAI,CAAC,CAAC,OAAO;oBAAE,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAC/C,IAAI,CAAC,CAAC,MAAM;oBAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;gBAC5C,IAAI,CAAC,CAAC,QAAQ;oBAAE,GAAG,CAAC,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;gBAClD,IAAI,CAAC,CAAC,KAAK;oBAAE,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACzC,IAAI,CAAC,CAAC,QAAQ;oBAAE,GAAG,CAAC,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;gBAClD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,MAA4B,EAC5B,WAAmB,EACnB,QAAgB,EAChB,OAMC;IAED,MAAM,IAAI,GAAG,eAAe,CAAC,QAAQ,EAAE;QACrC,GAAG,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/E,GAAG,CAAC,OAAO,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5E,GAAG,CAAC,OAAO,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAClF,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC5D,CAAC,CAAC;IACH,MAAM,IAAI,GAAsC,EAAE,CAAC;IACnD,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS;QAAE,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IAC3E,OAAO;QACL,GAAG,IAAI;QACP,WAAW,EAAE,UAAU;QACvB,YAAY,EAAE,0BAA0B,CAAC,MAAM,EAAE,WAAW,EAAE,IAAI,CAAC;KACpE,CAAC;AACJ,CAAC"}

package/dist/embeddings.d.ts

@@ -0,0 +1,18 @@
+import type { EmbeddingProvider } from '@fabric-harness/sdk';
+import type { DatabricksRestClient } from './client.js';
+export interface DatabricksEmbeddingsOptions {
+    client: DatabricksRestClient;
+    /** Serving-endpoint name of the embedding model. */
+    endpoint: string;
+    name?: string;
+}
+/**
+ * Embeddings via a Mosaic AI serving **embedding** endpoint. Databricks exposes the OpenAI-compatible
+ * embeddings surface at `<host>/serving-endpoints/embeddings` with the endpoint name as `model`.
+ * Reuses the bundle's REST client, so it inherits the rotating UC-principal token.
+ *
+ * (URL/response schema worth a smoke test against a live embedding endpoint — chat and embeddings
+ * share the OpenAI-compatible convention but endpoint families can differ.)
+ */
+export declare function databricksEmbeddings(options: DatabricksEmbeddingsOptions): EmbeddingProvider;
+//# sourceMappingURL=embeddings.d.ts.map

package/dist/embeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.d.ts","sourceRoot":"","sources":["../src/embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAExD,MAAM,WAAW,2BAA2B;IAC1C,MAAM,EAAE,oBAAoB,CAAC;IAC7B,oDAAoD;IACpD,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAMD;;;;;;;GAOG;AACH,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,2BAA2B,GAAG,iBAAiB,CAa5F"}

package/dist/embeddings.js

@@ -0,0 +1,24 @@
+/**
+ * Embeddings via a Mosaic AI serving **embedding** endpoint. Databricks exposes the OpenAI-compatible
+ * embeddings surface at `<host>/serving-endpoints/embeddings` with the endpoint name as `model`.
+ * Reuses the bundle's REST client, so it inherits the rotating UC-principal token.
+ *
+ * (URL/response schema worth a smoke test against a live embedding endpoint — chat and embeddings
+ * share the OpenAI-compatible convention but endpoint families can differ.)
+ */
+export function databricksEmbeddings(options) {
+    return {
+        name: options.name ?? 'databricks',
+        async embed(texts) {
+            if (texts.length === 0)
+                return [];
+            const response = await options.client.post('/serving-endpoints/embeddings', {
+                model: options.endpoint,
+                input: texts,
+            });
+            const data = [...(response.data ?? [])].sort((a, b) => (a.index ?? 0) - (b.index ?? 0));
+            return data.map((entry) => entry.embedding);
+        },
+    };
+}
+//# sourceMappingURL=embeddings.js.map

package/dist/embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../src/embeddings.ts"],"names":[],"mappings":"AAcA;;;;;;;GAOG;AACH,MAAM,UAAU,oBAAoB,CAAC,OAAoC;IACvE,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,YAAY;QAClC,KAAK,CAAC,KAAK,CAAC,KAAe;YACzB,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAC;YAClC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,IAAI,CAA0B,+BAA+B,EAAE;gBACnG,KAAK,EAAE,OAAO,CAAC,QAAQ;gBACvB,KAAK,EAAE,KAAK;aACb,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC;YACxF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC;KACF,CAAC;AACJ,CAAC"}