@fabric-harness/databricks 0.16.1 → 0.18.0

package/dist/index.js

@@ -1,234 +1,112 @@
-import { redactText } from '@fabric-harness/sdk';
-export class DatabricksRestClient {
-    host;
-    token;
-    fetchImpl;
-    constructor(options) {
-        this.host = options.host.replace(/\/$/, '');
-        this.token = options.token;
-        this.fetchImpl = options.fetchImpl ?? fetch;
-    }
-    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 token = typeof this.token === 'function' ? await this.token() : this.token;
-        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));
-        const response = await this.fetchImpl(url, {
-            method,
-            headers: {
-                authorization: `Bearer ${token}`,
-                ...(body !== undefined ? { 'content-type': 'application/json' } : {}),
-            },
-            ...(body !== undefined ? { body: JSON.stringify(body) } : {}),
+import { createSearchTool } from '@fabric-harness/sdk';
+import { DatabricksRestClient } from './client.js';
+import { databricksSqlTool, databricksTableInfoTool, databricksWorkspaceSource, unityCatalogTablesTool } from './tools.js';
+import { databricksIdentity } from './identity.js';
+import { databricksFoundationModelProvider, registerDatabricksModelProvider } from './model.js';
+import { databricksGovernancePolicy, withGovernanceTools } from './governance.js';
+import { lakebaseClient } from './lakebase.js';
+import { databricksEmbeddings } from './embeddings.js';
+import { databricksVectorSearch } from './vector-search.js';
+import { databricksAiQueryTool } from './ai.js';
+import { databricksGenieTool } from './genie.js';
+import { databricksLakeflowTools } from './lakeflow.js';
+import { databricksConsumption, databricksConsumptionTool } from './consumption.js';
+import { databricksFeatureLookupTool } from './feature-serving.js';
+import { databricksActualCostSource, databricksTenantCostLimit } from './cost-budget-actuals.js';
+// REST client + wait helpers (client.ts) and tool factories + workspace source (tools.ts).
+export { DatabricksRestClient, createDatabricksClient, waitForDatabricksRun, waitForDatabricksStatement } from './client.js';
+export { databricksSqlTool, databricksRunJobTool, databricksNotebookTool, databricksMlflowLogMetricTool, databricksMlflowLogParamTool, unityCatalogTablesTool, databricksTableInfoTool, databricksWorkspaceSource } from './tools.js';
+export { databricksFoundationModelProvider, registerDatabricksModelProvider } from './model.js';
+export { databricksIdentity } from './identity.js';
+export { withGovernance, withGovernanceTools, databricksApprovalRules, databricksGovernancePolicy } from './governance.js';
+export { lakebaseClient } from './lakebase.js';
+export { databricksEmbeddings } from './embeddings.js';
+export { databricksVectorSearch } from './vector-search.js';
+export { databricksAiQueryTool } from './ai.js';
+export { databricksGenieTool } from './genie.js';
+export { databricksLakeflowTools, databricksPipelineListTool, databricksPipelineStatusTool, databricksPipelineStartTool, databricksPipelineStopTool } from './lakeflow.js';
+export { databricksConsumption, databricksConsumptionTool } from './consumption.js';
+export { databricksActualCostSource, databricksTenantCostLimit } from './cost-budget-actuals.js';
+export { databricksFeatureLookupTool } from './feature-serving.js';
+export { databricksAgent, defineDatabricksAgent, resolveToolRefs } from './agent.js';
+export { MockDatabricksModelProvider } from './mock-provider.js';
+/**
+ * One-call wiring for a Databricks-native agent: model serving, a UC principal threaded through
+ * model + data + state, governed tools, an approval/egress policy, and (optionally) Lakebase-backed
+ * durable state. Also registers `databricks/*` + `mosaic-ai/*` model refs. Everything is layered on
+ * Unity Catalog — never a replacement for it.
+ */
+export function databricks(config) {
+    registerDatabricksModelProvider();
+    const identity = databricksIdentity(config.principal);
+    const client = new DatabricksRestClient({ host: config.host, token: identity, ...(config.fetchImpl ? { fetchImpl: config.fetchImpl } : {}) });
+    let embeddings;
+    let retriever;
+    let searchTool;
+    if (config.vectorSearch) {
+        if (config.vectorSearch.embeddingEndpoint)
+            embeddings = databricksEmbeddings({ client, endpoint: config.vectorSearch.embeddingEndpoint });
+        retriever = databricksVectorSearch({
+            client,
+            index: config.vectorSearch.index,
+            textColumn: config.vectorSearch.textColumn,
+            ...(config.vectorSearch.columns ? { columns: config.vectorSearch.columns } : {}),
+            ...(config.vectorSearch.idColumn ? { idColumn: config.vectorSearch.idColumn } : {}),
+            ...(config.vectorSearch.queryType ? { queryType: config.vectorSearch.queryType } : {}),
+            ...(embeddings ? { embeddings } : {}),
+        });
+        searchTool = createSearchTool(retriever, {
+            ...(config.vectorSearch.searchToolName ? { name: config.vectorSearch.searchToolName } : {}),
+            metadata: { provider: 'databricks', service: 'vector-search' },
         });
-        if (!response.ok) {
-            const text = redactText(await response.text(), { additionalSecrets: [token] });
-            throw new Error(`databricks: ${response.status} ${response.statusText}: ${text.slice(0, 500)}`);
-        }
-        if (response.status === 204)
-            return undefined;
-        return await response.json();
-    }
-}
-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;
-}
-export function databricksSqlTool(client, options) {
-    return {
-        name: options.name ?? 'databricks_sql',
-        description: options.description ?? 'Run a SQL statement on a Databricks SQL Warehouse.',
-        inputSchema: {
-            type: 'object',
-            additionalProperties: false,
-            properties: {
-                statement: { type: 'string' },
-                catalog: { type: 'string' },
-                schema: { type: 'string' },
-            },
-            required: ['statement'],
-        },
-        metadata: { effect: 'execute', provider: 'databricks', service: 'sql' },
-        execute: (input) => client.post('/api/2.0/sql/statements', {
-            warehouse_id: options.warehouseId,
-            statement: input.statement,
-            wait_timeout: options.waitTimeout ?? '30s',
-            disposition: options.disposition ?? 'INLINE',
-            ...(input.catalog ? { catalog: input.catalog } : {}),
-            ...(input.schema ? { schema: input.schema } : {}),
-        }),
+    const rawTools = [
+        ...(config.warehouseId ? [databricksSqlTool(client, { warehouseId: config.warehouseId })] : []),
+        unityCatalogTablesTool(client),
+        databricksTableInfoTool(client),
+        ...(searchTool ? [searchTool] : []),
+        ...(config.aiFunctions && config.warehouseId ? [databricksAiQueryTool(client, { warehouseId: config.warehouseId })] : []),
+        ...(config.genie ? [databricksGenieTool(client, { spaceId: config.genie.spaceId })] : []),
+        ...(config.lakeflow ? databricksLakeflowTools(client) : []),
+        ...(config.consumption && config.warehouseId ? [databricksConsumptionTool(client, { warehouseId: config.warehouseId })] : []),
+        ...(config.featureServing ? [databricksFeatureLookupTool(client, { endpoint: config.featureServing.endpoint })] : []),
+        ...(config.tools ?? []),
+    ];
+    const tools = withGovernanceTools(rawTools, {
+        principal: config.governance?.principalLabel ?? principalLabel(config.principal),
+        ...(config.governance?.onLineage ? { onLineage: config.governance.onLineage } : {}),
+        ...(config.governance?.catalogs ? { catalogs: config.governance.catalogs } : {}),
+    });
+    const policy = databricksGovernancePolicy({
+        tools,
+        host: config.host,
+        ...(config.governance?.stewardAudience ? { stewardAudience: config.governance.stewardAudience } : {}),
+        ...(config.governance?.approvalServices ? { approvalServices: config.governance.approvalServices } : {}),
+        ...(config.governance?.approvalTtlSeconds !== undefined ? { ttlSeconds: config.governance.approvalTtlSeconds } : {}),
+    });
+    const bundle = {
+        identity,
+        modelProvider: databricksFoundationModelProvider({ host: config.host, token: identity, ...(config.model ? { defaultModel: config.model } : {}) }),
+        client,
+        tools,
+        policy,
+        workspaceSource: (rootPath, options) => databricksWorkspaceSource(client, rootPath, options ?? {}),
+        ...(retriever ? { retriever } : {}),
+        ...(embeddings ? { embeddings } : {}),
+        ...(config.consumption && config.warehouseId ? { consumption: databricksConsumption(client, { warehouseId: config.warehouseId }) } : {}),
     };
-}
-export function databricksRunJobTool(client, options = {}) {
-    return {
-        name: options.name ?? 'databricks_run_job',
-        description: options.description ?? 'Trigger an existing Databricks job by job id.',
-        inputSchema: {
-            type: 'object',
-            additionalProperties: false,
-            properties: {
-                jobId: { type: 'number' },
-                notebookParams: { type: 'object', additionalProperties: true },
-                pythonParams: { type: 'array', items: { type: 'string' } },
-                jarParams: { type: 'array', items: { type: 'string' } },
-            },
-            required: ['jobId'],
-        },
-        metadata: { effect: 'execute', provider: 'databricks', service: 'jobs' },
-        execute: (input) => client.post('/api/2.1/jobs/run-now', {
-            job_id: input.jobId,
-            ...(input.notebookParams ? { notebook_params: input.notebookParams } : {}),
-            ...(input.pythonParams ? { python_params: input.pythonParams } : {}),
-            ...(input.jarParams ? { jar_params: input.jarParams } : {}),
-        }),
-    };
-}
-export function databricksNotebookTool(client, options = {}) {
-    return {
-        name: options.name ?? 'databricks_run_notebook',
-        description: options.description ?? 'Submit a one-off Databricks notebook run.',
-        inputSchema: {
-            type: 'object',
-            additionalProperties: false,
-            properties: {
-                notebookPath: { type: 'string' },
-                baseParameters: { type: 'object', additionalProperties: true },
-                runName: { type: 'string' },
-            },
-            required: ['notebookPath'],
-        },
-        metadata: { effect: 'execute', provider: 'databricks', service: 'jobs' },
-        execute: (input) => client.post('/api/2.1/jobs/runs/submit', {
-            run_name: input.runName ?? `fabric-harness:${input.notebookPath}`,
-            ...(options.existingClusterId ? { existing_cluster_id: options.existingClusterId } : {}),
-            notebook_task: {
-                notebook_path: input.notebookPath,
-                ...(input.baseParameters ? { base_parameters: input.baseParameters } : {}),
-            },
-        }),
-    };
-}
-export function databricksMlflowLogMetricTool(client, options = {}) {
-    return {
-        name: options.name ?? 'databricks_mlflow_log_metric',
-        description: options.description ?? 'Log a metric to an MLflow run hosted in Databricks.',
-        inputSchema: {
-            type: 'object',
-            additionalProperties: false,
-            properties: {
-                runId: { type: 'string' },
-                key: { type: 'string' },
-                value: { type: 'number' },
-                step: { type: 'number' },
-                timestamp: { type: 'number' },
-            },
-            required: ['runId', 'key', 'value'],
-        },
-        metadata: { effect: 'write', provider: 'databricks', service: 'mlflow' },
-        execute: (input) => client.post('/api/2.0/mlflow/runs/log-metric', {
-            run_id: input.runId,
-            key: input.key,
-            value: input.value,
-            timestamp: input.timestamp ?? Date.now(),
-            ...(input.step !== undefined ? { step: input.step } : {}),
-        }),
-    };
-}
-export function databricksMlflowLogParamTool(client, options = {}) {
-    return {
-        name: options.name ?? 'databricks_mlflow_log_param',
-        description: options.description ?? 'Log a parameter to an MLflow run hosted in Databricks.',
-        inputSchema: {
-            type: 'object',
-            additionalProperties: false,
-            properties: { runId: { type: 'string' }, key: { type: 'string' }, value: { type: 'string' } },
-            required: ['runId', 'key', 'value'],
-        },
-        metadata: { effect: 'write', provider: 'databricks', service: 'mlflow' },
-        execute: (input) => client.post('/api/2.0/mlflow/runs/log-parameter', { run_id: input.runId, key: input.key, value: input.value }),
-    };
-}
-export function unityCatalogTablesTool(client, options = {}) {
-    return {
-        name: options.name ?? 'databricks_unity_catalog_tables',
-        description: options.description ?? 'List Unity Catalog tables in a catalog and schema.',
-        inputSchema: {
-            type: 'object',
-            additionalProperties: false,
-            properties: { catalog: { type: 'string' }, schema: { type: 'string' } },
-            required: ['catalog', 'schema'],
-        },
-        metadata: { effect: 'read', provider: 'databricks', service: 'unity-catalog' },
-        execute: (input) => client.get('/api/2.1/unity-catalog/tables', { catalog_name: input.catalog, schema_name: input.schema }),
-    };
-}
-export function databricksWorkspaceSource(client, rootPath, options = {}) {
-    const include = options.include ?? (() => true);
-    return {
-        name: options.name ?? `databricks-workspace:${rootPath}`,
-        async *entries() {
-            yield* walkWorkspace(client, rootPath, rootPath, include, options.recursive ?? true, options.format ?? 'SOURCE');
-        },
-    };
-}
-async function* walkWorkspace(client, root, current, include, recursive, format) {
-    const listing = await client.get('/api/2.0/workspace/list', { path: current });
-    for (const object of listing.objects ?? []) {
-        if (object.object_type === 'DIRECTORY') {
-            if (recursive)
-                yield* walkWorkspace(client, root, object.path, include, recursive, format);
-            continue;
-        }
-        const relative = object.path.slice(root.length).replace(/^\/+/, '');
-        if (!relative || !include(relative))
-            continue;
-        const exported = await client.get('/api/2.0/workspace/export', { path: object.path, format });
-        yield { path: relative, content: exported.content ? base64ToBytes(exported.content) : new Uint8Array() };
+    if (config.lakebase) {
+        const lakebase = lakebaseClient({ ...config.lakebase, token: identity });
+        bundle.lakebase = lakebase;
+        if (config.sessionStoreFactory)
+            bundle.store = config.sessionStoreFactory(lakebase);
     }
+    return bundle;
 }
-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';
-}
-function base64ToBytes(value) {
-    if (typeof Buffer !== 'undefined')
-        return new Uint8Array(Buffer.from(value, 'base64'));
-    const binary = atob(value);
-    const bytes = new Uint8Array(binary.length);
-    for (let i = 0; i < binary.length; i += 1)
-        bytes[i] = binary.charCodeAt(i);
-    return bytes;
+function principalLabel(principal) {
+    if (principal.kind === 'service-principal')
+        return `sp:${principal.clientId}`;
+    // pat / on-behalf-of: prefer an explicit label (e.g. the channel actor / user id) over the bare kind.
+    return principal.label ?? principal.kind;
 }
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAQjD,MAAM,OAAO,oBAAoB;IACd,IAAI,CAAS;IACb,KAAK,CAAmC;IACxC,SAAS,CAAe;IAEzC,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;IAC9C,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,KAAK,GAAG,OAAO,IAAI,CAAC,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;QACjF,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;QACpH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE;YACzC,MAAM;YACN,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,KAAK,EAAE;gBAChC,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACtE;YACD,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC9D,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,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,GAAG,CAAC,EAAE,CAAC,CAAC;QAClG,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG;YAAE,OAAO,SAAc,CAAC;QACnD,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAO,CAAC;IACpC,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;AAUD,MAAM,UAAU,iBAAiB,CAAC,MAA4B,EAAE,OAAiC;IAC/F,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,gBAAgB;QACtC,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,oDAAoD;QACxF,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE;gBACV,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC7B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC3B;YACD,QAAQ,EAAE,CAAC,WAAW,CAAC;SACxB;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE;QACvE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE;YACzD,YAAY,EAAE,OAAO,CAAC,WAAW;YACjC,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,YAAY,EAAE,OAAO,CAAC,WAAW,IAAI,KAAK;YAC1C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,QAAQ;YAC5C,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACpD,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAClD,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAA4B,EAAE,UAAmD,EAAE;IACtH,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,oBAAoB;QAC1C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,+CAA+C;QACnF,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE;gBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;gBAC9D,YAAY,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;gBAC1D,SAAS,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;aACxD;YACD,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE;QACxE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE;YACvD,MAAM,EAAE,KAAK,CAAC,KAAK;YACnB,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1E,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACpE,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5D,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,MAA4B,EAAE,UAA+E,EAAE;IACpJ,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,yBAAyB;QAC/C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,2CAA2C;QAC/E,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE;gBACV,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAChC,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;gBAC9D,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC5B;YACD,QAAQ,EAAE,CAAC,cAAc,CAAC;SAC3B;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE;QACxE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE;YAC3D,QAAQ,EAAE,KAAK,CAAC,OAAO,IAAI,kBAAkB,KAAK,CAAC,YAAY,EAAE;YACjE,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxF,aAAa,EAAE;gBACb,aAAa,EAAE,KAAK,CAAC,YAAY;gBACjC,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC3E;SACF,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,MAA4B,EAAE,UAAmD,EAAE;IAC/H,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,8BAA8B;QACpD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,qDAAqD;QACzF,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE;gBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACvB,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC9B;YACD,QAAQ,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC;SACpC;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,QAAQ,EAAE;QACxE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE;YACjE,MAAM,EAAE,KAAK,CAAC,KAAK;YACnB,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE;YACxC,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC1D,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,MAA4B,EAAE,UAAmD,EAAE;IAC9H,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,6BAA6B;QACnD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,wDAAwD;QAC5F,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YAC7F,QAAQ,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC;SACpC;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,QAAQ,EAAE;QACxE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,CAAC,GAAG,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;KACnI,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,MAA4B,EAAE,UAAmD,EAAE;IACxH,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,iCAAiC;QACvD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,oDAAoD;QACxF,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YACvE,QAAQ,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC;SAChC;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,eAAe,EAAE;QAC9E,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,+BAA+B,EAAE,EAAE,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;KAC5H,CAAC;AACJ,CAAC;AASD,MAAM,UAAU,yBAAyB,CAAC,MAA4B,EAAE,QAAgB,EAAE,UAA4C,EAAE;IACtI,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IAChD,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,wBAAwB,QAAQ,EAAE;QACxD,KAAK,CAAC,CAAC,OAAO;YACZ,KAAK,CAAC,CAAC,aAAa,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI,EAAE,OAAO,CAAC,MAAM,IAAI,QAAQ,CAAC,CAAC;QACnH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,aAAa,CAAC,MAA4B,EAAE,IAAY,EAAE,OAAe,EAAE,OAAkC,EAAE,SAAkB,EAAE,MAAc;IAC/J,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,GAAG,CAA8D,yBAAyB,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5I,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;QAC3C,IAAI,MAAM,CAAC,WAAW,KAAK,WAAW,EAAE,CAAC;YACvC,IAAI,SAAS;gBAAE,KAAK,CAAC,CAAC,aAAa,CAAC,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;YAC3F,SAAS;QACX,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,QAAQ,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;YAAE,SAAS;QAC9C,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,GAAG,CAAuB,2BAA2B,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QACpH,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,UAAU,EAAE,EAAE,CAAC;IAC3G,CAAC;AACH,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;AAED,SAAS,aAAa,CAAC,KAAa;IAClC,IAAI,OAAO,MAAM,KAAK,WAAW;QAAE,OAAO,IAAI,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC;IACvF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IAC3B,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC;QAAE,KAAK,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IAC3E,OAAO,KAAK,CAAC;AACf,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,EAAE,iBAAiB,EAAE,uBAAuB,EAAE,yBAAyB,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAE3H,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEnD,OAAO,EAAE,iCAAiC,EAAE,+BAA+B,EAAE,MAAM,YAAY,CAAC;AAChG,OAAO,EAAE,0BAA0B,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAElF,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE/C,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AACvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAC5D,OAAO,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AAChD,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAAE,qBAAqB,EAAE,yBAAyB,EAAE,MAAM,kBAAkB,CAAC;AAEpF,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,EAAE,0BAA0B,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AAGjG,2FAA2F;AAC3F,OAAO,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,0BAA0B,EAAE,MAAM,aAAa,CAAC;AAE7H,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,6BAA6B,EAAE,4BAA4B,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,yBAAyB,EAAE,MAAM,YAAY,CAAC;AAGtO,OAAO,EAAE,iCAAiC,EAAE,+BAA+B,EAAE,MAAM,YAAY,CAAC;AAEhG,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEnD,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAE,uBAAuB,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAE3H,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE/C,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAEvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAE5D,OAAO,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AAEhD,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAEjD,OAAO,EAAE,uBAAuB,EAAE,0BAA0B,EAAE,4BAA4B,EAAE,2BAA2B,EAAE,0BAA0B,EAAE,MAAM,eAAe,CAAC;AAC3K,OAAO,EAAE,qBAAqB,EAAE,yBAAyB,EAAE,MAAM,kBAAkB,CAAC;AAEpF,OAAO,EAAE,0BAA0B,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AAEjG,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AAGnE,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAErF,OAAO,EAAE,2BAA2B,EAAE,MAAM,oBAAoB,CAAC;AAyFjE;;;;;GAKG;AACH,MAAM,UAAU,UAAU,CAAC,MAA8B;IACvD,+BAA+B,EAAE,CAAC;IAClC,MAAM,QAAQ,GAAG,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACtD,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAE9I,IAAI,UAAyC,CAAC;IAC9C,IAAI,SAAgC,CAAC;IACrC,IAAI,UAA+B,CAAC;IACpC,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACxB,IAAI,MAAM,CAAC,YAAY,CAAC,iBAAiB;YAAE,UAAU,GAAG,oBAAoB,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,YAAY,CAAC,iBAAiB,EAAE,CAAC,CAAC;QAC1I,SAAS,GAAG,sBAAsB,CAAC;YACjC,MAAM;YACN,KAAK,EAAE,MAAM,CAAC,YAAY,CAAC,KAAK;YAChC,UAAU,EAAE,MAAM,CAAC,YAAY,CAAC,UAAU;YAC1C,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAChF,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnF,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtF,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACtC,CAAC,CAAC;QACH,UAAU,GAAG,gBAAgB,CAAC,SAAS,EAAE;YACvC,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3F,QAAQ,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,eAAe,EAAE;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,MAAM,QAAQ,GAAc;QAC1B,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1G,sBAAsB,CAAC,MAAM,CAAY;QACzC,uBAAuB,CAAC,MAAM,CAAY;QAC1C,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACnC,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACpI,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACpG,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3D,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,yBAAyB,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACxI,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,2BAA2B,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,MAAM,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAChI,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;KACxB,CAAC;IACF,MAAM,KAAK,GAAG,mBAAmB,CAAC,QAAQ,EAAE;QAC1C,SAAS,EAAE,MAAM,CAAC,UAAU,EAAE,cAAc,IAAI,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC;QAChF,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnF,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACjF,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,0BAA0B,CAAC;QACxC,KAAK;QACL,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,MAAM,CAAC,UAAU,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrG,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,gBAAgB,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,MAAM,CAAC,UAAU,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACxG,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACrH,CAAC,CAAC;IAEH,MAAM,MAAM,GAAqB;QAC/B,QAAQ;QACR,aAAa,EAAE,iCAAiC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACjJ,MAAM;QACN,KAAK;QACL,MAAM;QACN,eAAe,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,CAAC,yBAAyB,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,IAAI,EAAE,CAAC;QAClG,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrC,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,qBAAqB,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACzI,CAAC;IAEF,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpB,MAAM,QAAQ,GAAG,cAAc,CAAC,EAAE,GAAG,MAAM,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QACzE,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC3B,IAAI,MAAM,CAAC,mBAAmB;YAAE,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;IACtF,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,cAAc,CAAC,SAA8B;IACpD,IAAI,SAAS,CAAC,IAAI,KAAK,mBAAmB;QAAE,OAAO,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;IAC9E,sGAAsG;IACtG,OAAO,SAAS,CAAC,KAAK,IAAI,SAAS,CAAC,IAAI,CAAC;AAC3C,CAAC"}

package/dist/lakebase.d.ts

@@ -0,0 +1,49 @@
+import type { DatabricksTokenProvider } from './identity.js';
+/**
+ * Databricks Lakebase is managed Postgres. This builds a Postgres client whose **password is the
+ * rotating OAuth token** (from {@link databricksIdentity}) rather than a static string: `pg.Pool`
+ * evaluates the `password` function for every new connection, so a long-lived pool never pins a
+ * stale token. The returned client is structurally compatible with `PostgresClientLike` from
+ * `@fabric-harness/node`, so it drops straight into `new PostgresSessionStore({ client })` (and the
+ * Postgres cost-budget / session-memory stores) — sessions + dispatch journal live in the lakehouse.
+ *
+ * Kept decoupled from `@fabric-harness/node` on purpose: this package owns the rotating-token
+ * client; store assembly happens at the app / `databricks()` bundle layer where `node` is present.
+ */
+export interface LakebaseClient {
+    query<T = Record<string, unknown>>(sql: string, params?: unknown[]): Promise<{
+        rows: T[];
+    }>;
+    end(): Promise<void>;
+}
+export interface LakebasePoolConfig {
+    host: string;
+    port: number;
+    database: string;
+    user: string;
+    /** Static password or a resolver evaluated per new connection (the rotating token path). */
+    password: string | (() => string | Promise<string>);
+    ssl?: boolean | {
+        rejectUnauthorized?: boolean;
+    };
+    max?: number;
+}
+export type LakebasePoolFactory = (config: LakebasePoolConfig) => LakebaseClient;
+export interface LakebaseClientOptions {
+    host: string;
+    database: string;
+    user: string;
+    /** Rotating OAuth token (from {@link databricksIdentity}) or a static password. */
+    token: string | DatabricksTokenProvider;
+    /** Default 5432. */
+    port?: number;
+    /** Default `{ rejectUnauthorized: true }` — Lakebase requires TLS. */
+    ssl?: boolean | {
+        rejectUnauthorized?: boolean;
+    };
+    max?: number;
+    /** Inject a pg-like Pool factory for testing; defaults to the optional `pg` package. */
+    poolFactory?: LakebasePoolFactory;
+}
+export declare function lakebaseClient(options: LakebaseClientOptions): LakebaseClient;
+//# sourceMappingURL=lakebase.d.ts.map

package/dist/lakebase.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lakebase.d.ts","sourceRoot":"","sources":["../src/lakebase.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAI7D;;;;;;;;;;GAUG;AACH,MAAM,WAAW,cAAc;IAC7B,KAAK,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,CAAC,EAAE,CAAA;KAAE,CAAC,CAAC;IAC5F,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACtB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,4FAA4F;IAC5F,QAAQ,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IACpD,GAAG,CAAC,EAAE,OAAO,GAAG;QAAE,kBAAkB,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;IACjD,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,mBAAmB,GAAG,CAAC,MAAM,EAAE,kBAAkB,KAAK,cAAc,CAAC;AAEjF,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,mFAAmF;IACnF,KAAK,EAAE,MAAM,GAAG,uBAAuB,CAAC;IACxC,oBAAoB;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,sEAAsE;IACtE,GAAG,CAAC,EAAE,OAAO,GAAG;QAAE,kBAAkB,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;IACjD,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,wFAAwF;IACxF,WAAW,CAAC,EAAE,mBAAmB,CAAC;CACnC;AAED,wBAAgB,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,cAAc,CAW7E"}

package/dist/lakebase.js

@@ -0,0 +1,25 @@
+import { createRequire } from 'node:module';
+const require = createRequire(import.meta.url);
+export function lakebaseClient(options) {
+    const factory = options.poolFactory ?? defaultPoolFactory();
+    return factory({
+        host: options.host,
+        port: options.port ?? 5432,
+        database: options.database,
+        user: options.user,
+        password: options.token,
+        ssl: options.ssl ?? { rejectUnauthorized: true },
+        ...(options.max !== undefined ? { max: options.max } : {}),
+    });
+}
+function defaultPoolFactory() {
+    let PoolCtor;
+    try {
+        PoolCtor = require('pg').Pool;
+    }
+    catch (error) {
+        throw new Error(`Lakebase store requires the optional "pg" package to be installed by the application. Run: pnpm add pg. ${error instanceof Error ? error.message : String(error)}`);
+    }
+    return (config) => new PoolCtor(config);
+}
+//# sourceMappingURL=lakebase.js.map

package/dist/lakebase.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lakebase.js","sourceRoot":"","sources":["../src/lakebase.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAG5C,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AA8C/C,MAAM,UAAU,cAAc,CAAC,OAA8B;IAC3D,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,IAAI,kBAAkB,EAAE,CAAC;IAC5D,OAAO,OAAO,CAAC;QACb,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI;QAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,QAAQ,EAAE,OAAO,CAAC,KAAK;QACvB,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE;QAChD,GAAG,CAAC,OAAO,CAAC,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC3D,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB;IACzB,IAAI,QAA4D,CAAC;IACjE,IAAI,CAAC;QACH,QAAQ,GAAI,OAAO,CAAC,IAAI,CAAkE,CAAC,IAAI,CAAC;IAClG,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,2GAA2G,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACvL,CAAC;IACD,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;AAC1C,CAAC"}

package/dist/lakeflow.d.ts

@@ -0,0 +1,28 @@
+import type { ToolDef } from '@fabric-harness/sdk';
+import type { DatabricksRestClient } from './client.js';
+type ToolMeta = {
+    name?: string;
+    description?: string;
+};
+/**
+ * Lakeflow Declarative Pipelines (formerly DLT). These drive heavy continuous-compute consumption.
+ * `start`/`stop` mutate pipeline state (`effect: 'execute'`, governable); `list`/`status` are reads.
+ */
+export declare function databricksPipelineListTool(client: DatabricksRestClient, options?: ToolMeta): ToolDef<{
+    maxResults?: number;
+    filter?: string;
+}, unknown>;
+export declare function databricksPipelineStatusTool(client: DatabricksRestClient, options?: ToolMeta): ToolDef<{
+    pipelineId: string;
+}, unknown>;
+export declare function databricksPipelineStartTool(client: DatabricksRestClient, options?: ToolMeta): ToolDef<{
+    pipelineId: string;
+    fullRefresh?: boolean;
+}, unknown>;
+export declare function databricksPipelineStopTool(client: DatabricksRestClient, options?: ToolMeta): ToolDef<{
+    pipelineId: string;
+}, unknown>;
+/** The default Lakeflow tool set added by the bundle's `lakeflow` option. */
+export declare function databricksLakeflowTools(client: DatabricksRestClient): ToolDef[];
+export {};
+//# sourceMappingURL=lakeflow.d.ts.map

package/dist/lakeflow.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lakeflow.d.ts","sourceRoot":"","sources":["../src/lakeflow.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAc,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAGxD,KAAK,QAAQ,GAAG;IAAE,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAExD;;;GAGG;AAEH,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,GAAE,QAAa,GAAG,OAAO,CAAC;IAAE,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,EAAE,OAAO,CAAC,CAe3J;AAED,wBAAgB,4BAA4B,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,GAAE,QAAa,GAAG,OAAO,CAAC;IAAE,UAAU,EAAE,MAAM,CAAA;CAAE,EAAE,OAAO,CAAC,CAa3I;AAED,wBAAgB,2BAA2B,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,GAAE,QAAa,GAAG,OAAO,CAAC;IAAE,UAAU,EAAE,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,OAAO,CAAA;CAAE,EAAE,OAAO,CAAC,CAejK;AAED,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,GAAE,QAAa,GAAG,OAAO,CAAC;IAAE,UAAU,EAAE,MAAM,CAAA;CAAE,EAAE,OAAO,CAAC,CAazI;AAED,6EAA6E;AAC7E,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,oBAAoB,GAAG,OAAO,EAAE,CAO/E"}

package/dist/lakeflow.js

@@ -0,0 +1,75 @@
+const enc = encodeURIComponent;
+/**
+ * Lakeflow Declarative Pipelines (formerly DLT). These drive heavy continuous-compute consumption.
+ * `start`/`stop` mutate pipeline state (`effect: 'execute'`, governable); `list`/`status` are reads.
+ */
+export function databricksPipelineListTool(client, options = {}) {
+    return {
+        name: options.name ?? 'databricks_pipeline_list',
+        description: options.description ?? 'List Lakeflow Declarative Pipelines (DLT) in the workspace.',
+        inputSchema: {
+            type: 'object',
+            additionalProperties: false,
+            properties: { maxResults: { type: 'number', minimum: 1 }, filter: { type: 'string' } },
+        },
+        metadata: { effect: 'read', provider: 'databricks', service: 'lakeflow' },
+        execute: (input) => client.get('/api/2.0/pipelines', {
+            ...(input.maxResults !== undefined ? { max_results: input.maxResults } : {}),
+            ...(input.filter ? { filter: input.filter } : {}),
+        }),
+    };
+}
+export function databricksPipelineStatusTool(client, options = {}) {
+    return {
+        name: options.name ?? 'databricks_pipeline_status',
+        description: options.description ?? 'Get a Lakeflow pipeline, including its state and latest updates.',
+        inputSchema: {
+            type: 'object',
+            additionalProperties: false,
+            properties: { pipelineId: { type: 'string' } },
+            required: ['pipelineId'],
+        },
+        metadata: { effect: 'read', provider: 'databricks', service: 'lakeflow' },
+        execute: (input) => client.get(`/api/2.0/pipelines/${enc(input.pipelineId)}`),
+    };
+}
+export function databricksPipelineStartTool(client, options = {}) {
+    return {
+        name: options.name ?? 'databricks_pipeline_start',
+        description: options.description ?? 'Trigger a Lakeflow pipeline update (run). Set fullRefresh to reprocess all data.',
+        inputSchema: {
+            type: 'object',
+            additionalProperties: false,
+            properties: { pipelineId: { type: 'string' }, fullRefresh: { type: 'boolean' } },
+            required: ['pipelineId'],
+        },
+        metadata: { effect: 'execute', provider: 'databricks', service: 'lakeflow' },
+        execute: (input) => client.post(`/api/2.0/pipelines/${enc(input.pipelineId)}/updates`, {
+            ...(input.fullRefresh !== undefined ? { full_refresh: input.fullRefresh } : {}),
+        }),
+    };
+}
+export function databricksPipelineStopTool(client, options = {}) {
+    return {
+        name: options.name ?? 'databricks_pipeline_stop',
+        description: options.description ?? 'Stop the active update of a Lakeflow pipeline.',
+        inputSchema: {
+            type: 'object',
+            additionalProperties: false,
+            properties: { pipelineId: { type: 'string' } },
+            required: ['pipelineId'],
+        },
+        metadata: { effect: 'execute', provider: 'databricks', service: 'lakeflow' },
+        execute: (input) => client.post(`/api/2.0/pipelines/${enc(input.pipelineId)}/stop`),
+    };
+}
+/** The default Lakeflow tool set added by the bundle's `lakeflow` option. */
+export function databricksLakeflowTools(client) {
+    return [
+        databricksPipelineListTool(client),
+        databricksPipelineStatusTool(client),
+        databricksPipelineStartTool(client),
+        databricksPipelineStopTool(client),
+    ];
+}
+//# sourceMappingURL=lakeflow.js.map

package/dist/lakeflow.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lakeflow.js","sourceRoot":"","sources":["../src/lakeflow.ts"],"names":[],"mappings":"AAGA,MAAM,GAAG,GAAG,kBAAkB,CAAC;AAG/B;;;GAGG;AAEH,MAAM,UAAU,0BAA0B,CAAC,MAA4B,EAAE,UAAoB,EAAE;IAC7F,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,0BAA0B;QAChD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,6DAA6D;QACjG,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;SACvF;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,UAAU,EAAE;QACzE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,oBAAoB,EAAE;YACnD,GAAG,CAAC,KAAK,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAClD,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,MAA4B,EAAE,UAAoB,EAAE;IAC/F,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,4BAA4B;QAClD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,kEAAkE;QACtG,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YAC9C,QAAQ,EAAE,CAAC,YAAY,CAAC;SACzB;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,UAAU,EAAE;QACzE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,sBAAsB,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;KAC9E,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,MAA4B,EAAE,UAAoB,EAAE;IAC9F,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,2BAA2B;QACjD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,kFAAkF;QACtH,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;YAChF,QAAQ,EAAE,CAAC,YAAY,CAAC;SACzB;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,UAAU,EAAE;QAC5E,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,EAAE;YACrF,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAClE,CAAC;KACjB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,MAA4B,EAAE,UAAoB,EAAE;IAC7F,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,0BAA0B;QAChD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,gDAAgD;QACpF,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YAC9C,QAAQ,EAAE,CAAC,YAAY,CAAC;SACzB;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,UAAU,EAAE;QAC5E,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC;KACpF,CAAC;AACJ,CAAC;AAED,6EAA6E;AAC7E,MAAM,UAAU,uBAAuB,CAAC,MAA4B;IAClE,OAAO;QACL,0BAA0B,CAAC,MAAM,CAAY;QAC7C,4BAA4B,CAAC,MAAM,CAAY;QAC/C,2BAA2B,CAAC,MAAM,CAAY;QAC9C,0BAA0B,CAAC,MAAM,CAAY;KAC9C,CAAC;AACJ,CAAC"}

package/dist/mock-provider.d.ts

@@ -0,0 +1,26 @@
+import type { ModelProvider, ModelRequest, ModelResponse, ModelStreamChunk, ModelToolCall } from '@fabric-harness/sdk';
+export interface MockDatabricksModelProviderOptions {
+    /**
+     * Optional prompt-to-response map. Keys are substrings matched against the
+     * latest user message. Values are either a plain text response or a
+     * `ModelResponse` object.
+     */
+    responses?: Record<string, string | ModelResponse>;
+    /** Optional deterministic tool calls to return when no map entry matches. */
+    defaultToolCalls?: ModelToolCall[];
+    /** Optional deterministic text response when no map entry matches. */
+    defaultText?: string;
+}
+/**
+ * A deterministic `ModelProvider` for Databricks agent tests and init templates.
+ * Returns structured responses for SQL/table-info tool calls without requiring
+ * real Databricks credentials.
+ */
+export declare class MockDatabricksModelProvider implements ModelProvider {
+    readonly name = "mock-databricks";
+    private readonly options;
+    constructor(options?: MockDatabricksModelProviderOptions);
+    generate(request: ModelRequest): Promise<ModelResponse>;
+    stream(request: ModelRequest): AsyncIterable<ModelStreamChunk>;
+}
+//# sourceMappingURL=mock-provider.d.ts.map

package/dist/mock-provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-provider.d.ts","sourceRoot":"","sources":["../src/mock-provider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,aAAa,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEvH,MAAM,WAAW,kCAAkC;IACjD;;;;OAIG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa,CAAC,CAAC;IACnD,6EAA6E;IAC7E,gBAAgB,CAAC,EAAE,aAAa,EAAE,CAAC;IACnC,sEAAsE;IACtE,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAOD;;;;GAIG;AACH,qBAAa,2BAA4B,YAAW,aAAa;IAC/D,QAAQ,CAAC,IAAI,qBAAqB;IAClC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqC;gBAEjD,OAAO,GAAE,kCAAuC;IAItD,QAAQ,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;IAmCtD,MAAM,CAAC,OAAO,EAAE,YAAY,GAAG,aAAa,CAAC,gBAAgB,CAAC;CAkBtE"}