@fabric-harness/databricks 0.16.1 → 0.18.0

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"}

package/dist/feature-serving.d.ts

@@ -0,0 +1,17 @@
+import type { JsonObject, ToolDef } from '@fabric-harness/sdk';
+import type { DatabricksRestClient } from './client.js';
+export interface DatabricksFeatureLookupToolOptions {
+    /** Feature Serving endpoint name (serves features from an online table / feature spec). */
+    endpoint: string;
+    name?: string;
+    description?: string;
+}
+/**
+ * Low-latency feature lookup from a Databricks **Feature Serving** endpoint (backed by an Online
+ * Table). The model passes primary-key records; the endpoint returns the served feature values.
+ * Drives serving consumption and keeps features governed by Unity Catalog under the bundle principal.
+ */
+export declare function databricksFeatureLookupTool(client: DatabricksRestClient, options: DatabricksFeatureLookupToolOptions): ToolDef<{
+    records: JsonObject[];
+}, unknown>;
+//# sourceMappingURL=feature-serving.d.ts.map

package/dist/feature-serving.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"feature-serving.d.ts","sourceRoot":"","sources":["../src/feature-serving.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAExD,MAAM,WAAW,kCAAkC;IACjD,2FAA2F;IAC3F,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;;;GAIG;AACH,wBAAgB,2BAA2B,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,EAAE,kCAAkC,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,UAAU,EAAE,CAAA;CAAE,EAAE,OAAO,CAAC,CAsBlK"}

package/dist/feature-serving.js

@@ -0,0 +1,29 @@
+/**
+ * Low-latency feature lookup from a Databricks **Feature Serving** endpoint (backed by an Online
+ * Table). The model passes primary-key records; the endpoint returns the served feature values.
+ * Drives serving consumption and keeps features governed by Unity Catalog under the bundle principal.
+ */
+export function databricksFeatureLookupTool(client, options) {
+    return {
+        name: options.name ?? 'databricks_feature_lookup',
+        description: options.description ?? 'Look up low-latency features for primary-key records from a Databricks Feature Serving endpoint.',
+        inputSchema: {
+            type: 'object',
+            additionalProperties: false,
+            properties: {
+                records: {
+                    type: 'array',
+                    items: { type: 'object', additionalProperties: true },
+                    minItems: 1,
+                    description: 'Primary-key records, e.g. [{ "user_id": 123 }].',
+                },
+            },
+            required: ['records'],
+        },
+        metadata: { effect: 'read', provider: 'databricks', service: 'feature-serving' },
+        execute: (input) => client.post(`/serving-endpoints/${encodeURIComponent(options.endpoint)}/invocations`, {
+            dataframe_records: input.records,
+        }),
+    };
+}
+//# sourceMappingURL=feature-serving.js.map

package/dist/feature-serving.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"feature-serving.js","sourceRoot":"","sources":["../src/feature-serving.ts"],"names":[],"mappings":"AAUA;;;;GAIG;AACH,MAAM,UAAU,2BAA2B,CAAC,MAA4B,EAAE,OAA2C;IACnH,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,2BAA2B;QACjD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,kGAAkG;QACtI,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE;gBACV,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;oBACrD,QAAQ,EAAE,CAAC;oBACX,WAAW,EAAE,iDAAiD;iBAC/D;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,iBAAiB,EAAE;QAChF,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,kBAAkB,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE;YACxG,iBAAiB,EAAE,KAAK,CAAC,OAAO;SACjC,CAAC;KACH,CAAC;AACJ,CAAC"}

package/dist/genie.d.ts

@@ -0,0 +1,22 @@
+import type { ToolDef } from '@fabric-harness/sdk';
+import type { DatabricksRestClient } from './client.js';
+export interface DatabricksGenieToolOptions {
+    /** AI/BI Genie space id. */
+    spaceId: string;
+    name?: string;
+    description?: string;
+    pollIntervalMs?: number;
+    timeoutMs?: number;
+}
+/**
+ * AI/BI Genie — delegate a natural-language analytics question to a governed Genie space. Genie
+ * generates and runs SQL on the warehouse and returns an answer; this tool returns the answer text,
+ * the generated SQL, and the result rows. Pass `conversationId` to continue a multi-turn thread.
+ *
+ * Genie response field names vary by API version — parsed defensively; smoke-test against a live space.
+ */
+export declare function databricksGenieTool(client: DatabricksRestClient, options: DatabricksGenieToolOptions): ToolDef<{
+    question: string;
+    conversationId?: string;
+}, unknown>;
+//# sourceMappingURL=genie.d.ts.map

package/dist/genie.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"genie.d.ts","sourceRoot":"","sources":["../src/genie.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAc,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAExD,MAAM,WAAW,0BAA0B;IACzC,4BAA4B;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAgBD;;;;;;GAMG;AACH,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,EAAE,0BAA0B,GAAG,OAAO,CAAC;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,cAAc,CAAC,EAAE,MAAM,CAAA;CAAE,EAAE,OAAO,CAAC,CAoDtK"}

package/dist/genie.js

@@ -0,0 +1,60 @@
+const TERMINAL_STATES = new Set(['COMPLETED', 'FAILED', 'CANCELLED', 'QUERY_RESULT_EXPIRED']);
+const enc = encodeURIComponent;
+/**
+ * AI/BI Genie — delegate a natural-language analytics question to a governed Genie space. Genie
+ * generates and runs SQL on the warehouse and returns an answer; this tool returns the answer text,
+ * the generated SQL, and the result rows. Pass `conversationId` to continue a multi-turn thread.
+ *
+ * Genie response field names vary by API version — parsed defensively; smoke-test against a live space.
+ */
+export function databricksGenieTool(client, options) {
+    const intervalMs = options.pollIntervalMs ?? 2000;
+    const timeoutMs = options.timeoutMs ?? 120_000;
+    const space = `/api/2.0/genie/spaces/${enc(options.spaceId)}`;
+    return {
+        name: options.name ?? 'databricks_genie_ask',
+        description: options.description ?? 'Ask a natural-language analytics question in a Databricks AI/BI Genie space; returns the answer, generated SQL, and result rows.',
+        inputSchema: {
+            type: 'object',
+            additionalProperties: false,
+            properties: { question: { type: 'string' }, conversationId: { type: 'string' } },
+            required: ['question'],
+        },
+        metadata: { effect: 'execute', provider: 'databricks', service: 'genie' },
+        async execute(input) {
+            let conversationId = input.conversationId;
+            let messageId;
+            if (conversationId) {
+                const res = await client.post(`${space}/conversations/${enc(conversationId)}/messages`, { content: input.question });
+                messageId = String(res.message_id ?? res.id ?? '');
+            }
+            else {
+                const res = await client.post(`${space}/start-conversation`, { content: input.question });
+                conversationId = String(res.conversation_id ?? res.conversation?.id ?? '');
+                messageId = String(res.message_id ?? res.message?.id ?? '');
+            }
+            const messagePath = `${space}/conversations/${enc(conversationId)}/messages/${enc(messageId)}`;
+            const deadline = Date.now() + timeoutMs;
+            let message = await client.get(messagePath);
+            while (!TERMINAL_STATES.has(message.status ?? '') && Date.now() < deadline) {
+                await new Promise((resolve) => setTimeout(resolve, intervalMs));
+                message = await client.get(messagePath);
+            }
+            const text = message.attachments?.map((a) => a.text?.content).filter(Boolean).join('\n') || undefined;
+            const sql = message.attachments?.find((a) => a.query?.query)?.query?.query;
+            let result;
+            if (sql) {
+                result = await client.get(`${messagePath}/query-result`).catch(() => undefined);
+            }
+            return {
+                conversationId,
+                messageId,
+                status: message.status,
+                ...(text ? { text } : {}),
+                ...(sql ? { sql } : {}),
+                ...(result !== undefined ? { result } : {}),
+            };
+        },
+    };
+}
+//# sourceMappingURL=genie.js.map

package/dist/genie.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"genie.js","sourceRoot":"","sources":["../src/genie.ts"],"names":[],"mappings":"AAYA,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,CAAC,WAAW,EAAE,QAAQ,EAAE,WAAW,EAAE,sBAAsB,CAAC,CAAC,CAAC;AAY9F,MAAM,GAAG,GAAG,kBAAkB,CAAC;AAE/B;;;;;;GAMG;AACH,MAAM,UAAU,mBAAmB,CAAC,MAA4B,EAAE,OAAmC;IACnG,MAAM,UAAU,GAAG,OAAO,CAAC,cAAc,IAAI,IAAI,CAAC;IAClD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC;IAC/C,MAAM,KAAK,GAAG,yBAAyB,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAE9D,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,sBAAsB;QAC5C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,kIAAkI;QACtK,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YAChF,QAAQ,EAAE,CAAC,UAAU,CAAC;SACvB;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,EAAE;QACzE,KAAK,CAAC,OAAO,CAAC,KAAK;YACjB,IAAI,cAAc,GAAG,KAAK,CAAC,cAAc,CAAC;YAC1C,IAAI,SAAiB,CAAC;YACtB,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,IAAI,CAAa,GAAG,KAAK,kBAAkB,GAAG,CAAC,cAAc,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACjI,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;YACrD,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,IAAI,CAAa,GAAG,KAAK,qBAAqB,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACtG,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,eAAe,IAAK,GAAG,CAAC,YAAuC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvG,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,IAAK,GAAG,CAAC,OAAkC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1F,CAAC;YAED,MAAM,WAAW,GAAG,GAAG,KAAK,kBAAkB,GAAG,CAAC,cAAc,CAAC,aAAa,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/F,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxC,IAAI,OAAO,GAAG,MAAM,MAAM,CAAC,GAAG,CAAe,WAAW,CAAC,CAAC;YAC1D,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;gBAC3E,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC;gBAChE,OAAO,GAAG,MAAM,MAAM,CAAC,GAAG,CAAe,WAAW,CAAC,CAAC;YACxD,CAAC;YAED,MAAM,IAAI,GAAG,OAAO,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC;YACtG,MAAM,GAAG,GAAG,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC;YAC3E,IAAI,MAAe,CAAC;YACpB,IAAI,GAAG,EAAE,CAAC;gBACR,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,GAAG,WAAW,eAAe,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAClF,CAAC;YAED,OAAO;gBACL,cAAc;gBACd,SAAS;gBACT,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzB,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvB,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5C,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/governance.d.ts

@@ -0,0 +1,66 @@
+import type { ApprovalPolicyRule, CapabilityPolicy, JsonObject, ToolDef } from '@fabric-harness/sdk';
+/**
+ * Governance for Databricks tools is **layered on top of Unity Catalog, never a replacement.** UC
+ * enforces table/row/column ACLs natively via the acting principal (see identity.ts). These helpers
+ * add the things UC doesn't: an **audit/lineage** stamp on every tool call, **approval routing** for
+ * sensitive operations, and an **egress allowlist** pinned to the workspace — all expressed through
+ * the SDK's existing `CapabilityPolicy`, so the loop enforces them without a bespoke engine.
+ */
+export interface DatabricksLineageRecord {
+    tool: string;
+    /** From the tool's `metadata.service` (e.g. 'sql', 'unity-catalog', 'jobs'). */
+    service?: string;
+    provider?: string;
+    /** Stable label for the acting principal — never the token. */
+    principal?: string;
+    catalog?: string;
+    schema?: string;
+    input: JsonObject;
+    startedAt: string;
+    finishedAt: string;
+    ok: boolean;
+    error?: string;
+}
+export interface DatabricksGovernanceOptions {
+    /** Stable label for the acting principal, recorded in lineage. Never pass a token here. */
+    principal?: string;
+    /** Sink for lineage/audit records. Defaults to no-op; wire to OTel / `onEvent` at the app layer. */
+    onLineage?: (record: DatabricksLineageRecord) => void;
+    /**
+     * Optional belt-and-suspenders catalog allowlist. Calls whose input references a catalog outside
+     * the list are rejected before execution. UC remains the source of truth; this is defense in depth.
+     */
+    catalogs?: string[];
+    /** Clock injection for tests. */
+    now?: () => Date;
+}
+/** Wraps one Databricks tool to stamp lineage/audit and enforce the optional catalog allowlist. */
+export declare function withGovernance<I, O>(tool: ToolDef<I, O>, options?: DatabricksGovernanceOptions): ToolDef<I, O>;
+/** Wraps a set of Databricks tools with {@link withGovernance}. */
+export declare function withGovernanceTools(tools: ToolDef[], options?: DatabricksGovernanceOptions): ToolDef[];
+export interface DatabricksApprovalRuleOptions {
+    /** Opaque approval audience (e.g. 'data-steward'). */
+    audience: string;
+    /** Restrict to these `metadata.service` values. Default: gate every write/execute tool. */
+    services?: string[];
+    reason?: string;
+    ttlSeconds?: number;
+}
+/** Builds approval rules (one per gated tool) for `CapabilityPolicy.toolPolicy.approvalRules`. */
+export declare function databricksApprovalRules(tools: ToolDef[], options: DatabricksApprovalRuleOptions): ApprovalPolicyRule[];
+export interface DatabricksGovernancePolicyOptions {
+    tools: ToolDef[];
+    /** Workspace host(s) to pin outbound egress to (URL or bare host). */
+    host?: string | string[];
+    /** When set, sensitive tools route to this approval audience. */
+    stewardAudience?: string;
+    approvalServices?: string[];
+    ttlSeconds?: number;
+}
+/**
+ * Assembles a `CapabilityPolicy` for a Databricks tool set: approval routing for sensitive
+ * operations (if `stewardAudience` is set) plus an egress allowlist pinned to the workspace host(s).
+ * Used by the `databricks()` bundle; can also be merged into an agent's own policy.
+ */
+export declare function databricksGovernancePolicy(options: DatabricksGovernancePolicyOptions): CapabilityPolicy;
+//# sourceMappingURL=governance.d.ts.map

package/dist/governance.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"governance.d.ts","sourceRoot":"","sources":["../src/governance.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAErG;;;;;;GAMG;AAEH,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,gFAAgF;IAChF,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,+DAA+D;IAC/D,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,UAAU,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,EAAE,EAAE,OAAO,CAAC;IACZ,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,2BAA2B;IAC1C,2FAA2F;IAC3F,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,oGAAoG;IACpG,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,uBAAuB,KAAK,IAAI,CAAC;IACtD;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,iCAAiC;IACjC,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC;CAClB;AAeD,mGAAmG;AACnG,wBAAgB,cAAc,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,OAAO,GAAE,2BAAgC,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAsClH;AAED,mEAAmE;AACnE,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,OAAO,GAAE,2BAAgC,GAAG,OAAO,EAAE,CAE1G;AAED,MAAM,WAAW,6BAA6B;IAC5C,sDAAsD;IACtD,QAAQ,EAAE,MAAM,CAAC;IACjB,2FAA2F;IAC3F,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAQD,kGAAkG;AAClG,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,6BAA6B,GAAG,kBAAkB,EAAE,CAUtH;AAED,MAAM,WAAW,iCAAiC;IAChD,KAAK,EAAE,OAAO,EAAE,CAAC;IACjB,sEAAsE;IACtE,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IACzB,iEAAiE;IACjE,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAcD;;;;GAIG;AACH,wBAAgB,0BAA0B,CAAC,OAAO,EAAE,iCAAiC,GAAG,gBAAgB,CAcvG"}