@fabric-harness/databricks 0.16.1 → 0.18.0

package/dist/agent.d.ts

@@ -0,0 +1,45 @@
+import type { AgentProfile, AgentTriggers, DefinedAgent, JsonObject, ModelProvider, SandboxBackend, Schema, Skill, ToolDef } from '@fabric-harness/sdk';
+import type { CapabilityPolicy, CostLimit, ApprovalOptions } from '@fabric-harness/sdk';
+import type { DatabricksBundle, DatabricksBundleConfig } from './index.js';
+export interface DatabricksAgentOptions {
+    name?: string;
+    description?: string;
+    instructions?: string;
+    input?: Schema<JsonObject>;
+    output?: Schema<unknown>;
+    model?: string;
+    target?: 'node' | 'temporal-worker';
+    skills?: Skill[];
+    triggers?: AgentTriggers;
+    databricks: DatabricksBundleConfig;
+    modelProvider?: ModelProvider;
+    tools?: ToolDef[] | string[];
+    sandbox?: SandboxBackend;
+    profile?: AgentProfile;
+    policy?: CapabilityPolicy;
+    costBudget?: CostLimit;
+    approvals?: ApprovalOptions;
+}
+export interface DefineDatabricksAgentOptions {
+    name?: string;
+    description?: string;
+    instructions?: string;
+    input?: Schema<JsonObject>;
+    output?: Schema<unknown>;
+    model?: string;
+    target?: 'node' | 'temporal-worker';
+    skills?: Skill[];
+    triggers?: AgentTriggers;
+    databricks?: DatabricksBundleConfig;
+    modelProvider?: ModelProvider;
+    tools?: ToolDef[] | string[];
+    sandbox?: SandboxBackend;
+    profile?: AgentProfile;
+    policy?: CapabilityPolicy;
+    costBudget?: CostLimit;
+    approvals?: ApprovalOptions;
+}
+export declare function databricksAgent(options: DatabricksAgentOptions): DefinedAgent<JsonObject, unknown>;
+export declare function defineDatabricksAgent(options?: DefineDatabricksAgentOptions): DefinedAgent<JsonObject, unknown>;
+export declare function resolveToolRefs(bundle: DatabricksBundle, refs: string[]): ToolDef[];
+//# sourceMappingURL=agent.d.ts.map

package/dist/agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAmB,YAAY,EAAE,aAAa,EAAE,YAAY,EAAiB,UAAU,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AACxL,OAAO,KAAK,EAAE,gBAAgB,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAExF,OAAO,KAAK,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAI3E,MAAM,WAAW,sBAAsB;IACrC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,GAAG,iBAAiB,CAAC;IACpC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC;IACjB,QAAQ,CAAC,EAAE,aAAa,CAAC;IACzB,UAAU,EAAE,sBAAsB,CAAC;IACnC,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,KAAK,CAAC,EAAE,OAAO,EAAE,GAAG,MAAM,EAAE,CAAC;IAC7B,OAAO,CAAC,EAAE,cAAc,CAAC;IACzB,OAAO,CAAC,EAAE,YAAY,CAAC;IACvB,MAAM,CAAC,EAAE,gBAAgB,CAAC;IAC1B,UAAU,CAAC,EAAE,SAAS,CAAC;IACvB,SAAS,CAAC,EAAE,eAAe,CAAC;CAC7B;AAED,MAAM,WAAW,4BAA4B;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,GAAG,iBAAiB,CAAC;IACpC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC;IACjB,QAAQ,CAAC,EAAE,aAAa,CAAC;IACzB,UAAU,CAAC,EAAE,sBAAsB,CAAC;IACpC,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,KAAK,CAAC,EAAE,OAAO,EAAE,GAAG,MAAM,EAAE,CAAC;IAC7B,OAAO,CAAC,EAAE,cAAc,CAAC;IACzB,OAAO,CAAC,EAAE,YAAY,CAAC;IACvB,MAAM,CAAC,EAAE,gBAAgB,CAAC;IAC1B,UAAU,CAAC,EAAE,SAAS,CAAC;IACvB,SAAS,CAAC,EAAE,eAAe,CAAC;CAC7B;AAuGD,wBAAgB,eAAe,CAAC,OAAO,EAAE,sBAAsB,GAAG,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAElG;AAED,wBAAgB,qBAAqB,CAAC,OAAO,CAAC,EAAE,4BAA4B,GAAG,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAuE/G;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,OAAO,EAAE,CAgCnF"}

package/dist/agent.js

@@ -0,0 +1,206 @@
+import { agent } from '@fabric-harness/sdk/strict';
+import { databricks } from './index.js';
+function resolveTools(bundle, tools) {
+    if (!tools)
+        return [];
+    const stringTools = tools.filter((t) => typeof t === 'string');
+    const toolDefs = tools.filter((t) => typeof t !== 'string');
+    if (stringTools.length > 0) {
+        const resolved = resolveToolRefs(bundle, stringTools);
+        return [...resolved, ...toolDefs];
+    }
+    return toolDefs;
+}
+function mergeProfile(options) {
+    const profile = options?.profile;
+    if (!profile)
+        return {};
+    return {
+        ...(profile.name !== undefined ? { name: profile.name } : {}),
+        ...(profile.description !== undefined ? { description: profile.description } : {}),
+        ...(profile.instructions !== undefined ? { instructions: profile.instructions } : {}),
+        ...(profile.model !== undefined ? { model: profile.model } : {}),
+        ...(profile.skills !== undefined ? { skills: profile.skills } : {}),
+        ...(profile.tools !== undefined ? { tools: profile.tools } : {}),
+        ...(profile.policy !== undefined ? { policy: profile.policy } : {}),
+        ...(profile.costBudget !== undefined ? { costBudget: profile.costBudget } : {}),
+        ...(profile.approvals !== undefined ? { approvals: profile.approvals } : {}),
+        ...(profile.sandbox !== undefined ? { sandbox: profile.sandbox } : {}),
+    };
+}
+function mergeOptions(options) {
+    const profile = mergeProfile(options);
+    return {
+        ...profile,
+        ...options,
+        ...(profile.skills || options.skills ? { skills: [...(profile.skills ?? []), ...(options.skills ?? [])] } : {}),
+        ...(profile.tools || options.tools ? { tools: [...(profile.tools ?? []), ...(options.tools ?? [])] } : {}),
+    };
+}
+/**
+ * Shared run body for both `databricksAgent` and `defineDatabricksAgent`. Threads the bundle's
+ * UC-principal model provider through by default — env-driven agents get the Databricks
+ * foundation model unless an explicit provider is supplied (or `--mock` overrides it upstream).
+ */
+async function runWithBundle(context, bundle, options) {
+    const extraTools = resolveTools(bundle, options.tools);
+    const agent = await context.init({
+        modelProvider: options.modelProvider ?? bundle.modelProvider,
+        tools: [...bundle.tools, ...extraTools],
+        policy: bundle.policy,
+        ...(bundle.store ? { store: bundle.store } : {}),
+        ...(options.skills ? { skills: options.skills } : {}),
+        ...(options.sandbox ? { sandbox: options.sandbox } : {}),
+    });
+    const session = await agent.session();
+    const question = context.input?.question ?? 'What would you like to know?';
+    return session.prompt(question);
+}
+function buildAgentDefinition(options) {
+    const merged = mergeOptions(options);
+    const bundle = databricks(merged.databricks);
+    const def = {
+        run: (context) => runWithBundle(context, bundle, {
+            ...(merged.modelProvider ? { modelProvider: merged.modelProvider } : {}),
+            ...(merged.tools ? { tools: merged.tools } : {}),
+            ...(merged.skills ? { skills: merged.skills } : {}),
+            ...(merged.sandbox ? { sandbox: merged.sandbox } : {}),
+        }),
+    };
+    if (merged.name !== undefined)
+        def.name = merged.name;
+    if (merged.description !== undefined)
+        def.description = merged.description;
+    if (merged.instructions !== undefined)
+        def.instructions = merged.instructions;
+    if (merged.input !== undefined)
+        def.input = merged.input;
+    if (merged.output !== undefined)
+        def.output = merged.output;
+    if (merged.model !== undefined)
+        def.model = merged.model;
+    if (merged.target !== undefined)
+        def.target = merged.target;
+    if (merged.skills !== undefined)
+        def.skills = merged.skills;
+    if (merged.triggers !== undefined)
+        def.triggers = merged.triggers;
+    if (merged.policy !== undefined)
+        def.policy = merged.policy;
+    if (merged.costBudget !== undefined)
+        def.costBudget = merged.costBudget;
+    if (merged.approvals !== undefined)
+        def.approvals = merged.approvals;
+    return def;
+}
+export function databricksAgent(options) {
+    return agent(buildAgentDefinition(options));
+}
+export function defineDatabricksAgent(options) {
+    const merged = options ? mergeOptions(options) : {};
+    const def = {
+        run: async (context) => {
+            // If explicit databricks config is provided, use it directly
+            if (merged.databricks) {
+                const bundle = databricks(merged.databricks);
+                return runWithBundle(context, bundle, {
+                    ...(merged.modelProvider ? { modelProvider: merged.modelProvider } : {}),
+                    ...(merged.tools ? { tools: merged.tools } : {}),
+                    ...(merged.skills ? { skills: merged.skills } : {}),
+                    ...(merged.sandbox ? { sandbox: merged.sandbox } : {}),
+                });
+            }
+            // Otherwise, read from env (or use a dummy config in mock mode)
+            const env = context.env ?? {};
+            const host = env.DATABRICKS_HOST;
+            if (!host && env.FABRIC_MOCK !== '1') {
+                throw new Error('Missing DATABRICKS_HOST environment variable. Set it or pass an explicit databricks config.');
+            }
+            const token = env.DATABRICKS_TOKEN;
+            const clientId = env.DATABRICKS_CLIENT_ID;
+            const clientSecret = env.DATABRICKS_CLIENT_SECRET;
+            if (!token && !(clientId && clientSecret) && env.FABRIC_MOCK !== '1') {
+                throw new Error('Missing Databricks credentials. Set DATABRICKS_TOKEN or both DATABRICKS_CLIENT_ID and DATABRICKS_CLIENT_SECRET.');
+            }
+            // In real mode the relevant fields are guaranteed by the guards above; in mock mode this
+            // principal is discarded (FABRIC_MOCK overrides it below), so empty-string fallbacks are safe.
+            const principal = token
+                ? { kind: 'pat', token }
+                : { kind: 'service-principal', host: host ?? '', clientId: clientId ?? '', clientSecret: clientSecret ?? '' };
+            const config = {
+                host: host ?? 'https://mock.databricks.com',
+                principal: env.FABRIC_MOCK === '1' ? { kind: 'pat', token: 'mock-token' } : principal,
+                ...(env.DATABRICKS_MODEL ? { model: env.DATABRICKS_MODEL } : {}),
+                ...(env.DATABRICKS_WAREHOUSE_ID ? { warehouseId: env.DATABRICKS_WAREHOUSE_ID } : {}),
+                ...(env.FABRIC_MOCK === '1' ? { warehouseId: 'mock-warehouse-id' } : {}),
+            };
+            const bundle = databricks(config);
+            // Default to the bundle's UC-principal Databricks foundation model so env-driven agents
+            // actually consume the Databricks serving endpoint. `--mock` overrides this upstream
+            // (see packages/node/src/run.ts), and an explicit modelProvider still wins.
+            return runWithBundle(context, bundle, {
+                ...(merged.modelProvider ? { modelProvider: merged.modelProvider } : {}),
+                ...(merged.tools ? { tools: merged.tools } : {}),
+                ...(merged.skills ? { skills: merged.skills } : {}),
+                ...(merged.sandbox ? { sandbox: merged.sandbox } : {}),
+            });
+        },
+    };
+    if (merged.name !== undefined)
+        def.name = merged.name;
+    if (merged.description !== undefined)
+        def.description = merged.description;
+    if (merged.instructions !== undefined)
+        def.instructions = merged.instructions;
+    if (merged.input !== undefined)
+        def.input = merged.input;
+    if (merged.output !== undefined)
+        def.output = merged.output;
+    if (merged.model !== undefined)
+        def.model = merged.model;
+    if (merged.target !== undefined)
+        def.target = merged.target;
+    if (merged.skills !== undefined)
+        def.skills = merged.skills;
+    if (merged.triggers !== undefined)
+        def.triggers = merged.triggers;
+    if (merged.policy !== undefined)
+        def.policy = merged.policy;
+    if (merged.costBudget !== undefined)
+        def.costBudget = merged.costBudget;
+    if (merged.approvals !== undefined)
+        def.approvals = merged.approvals;
+    return agent(def);
+}
+export function resolveToolRefs(bundle, refs) {
+    const toolMap = new Map();
+    for (const tool of bundle.tools) {
+        toolMap.set(tool.name, tool);
+    }
+    const shortNameMap = {
+        'sql': 'databricks_sql',
+        'tables': 'databricks_unity_catalog_tables',
+        'table-info': 'databricks_table_info',
+        'search': 'search',
+        'ai-query': 'databricks_ai_query',
+        'genie': 'databricks_genie_ask',
+        'lakeflow-list': 'databricks_pipeline_list',
+        'lakeflow-status': 'databricks_pipeline_status',
+        'lakeflow-start': 'databricks_pipeline_start',
+        'lakeflow-stop': 'databricks_pipeline_stop',
+        'consumption': 'databricks_consumption',
+        'feature-lookup': 'databricks_feature_lookup',
+    };
+    const result = [];
+    for (const ref of refs) {
+        const fullName = shortNameMap[ref] ?? ref;
+        const tool = toolMap.get(fullName);
+        if (!tool) {
+            const available = [...toolMap.keys()].join(', ');
+            throw new Error(`Unknown tool reference "${ref}". Available tools: ${available}`);
+        }
+        result.push(tool);
+    }
+    return result;
+}
+//# sourceMappingURL=agent.js.map

package/dist/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAC;AAEnD,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AA2CxC,SAAS,YAAY,CAAC,MAAwB,EAAE,KAA4B;IAC1E,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IACtB,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC;IAC5E,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAgB,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC;IAC1E,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,eAAe,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACtD,OAAO,CAAC,GAAG,QAAQ,EAAE,GAAG,QAAQ,CAAC,CAAC;IACpC,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AASD,SAAS,YAAY,CAAC,OAA0E;IAC9F,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,CAAC;IACjC,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAC;IACxB,OAAO;QACL,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7D,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,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrF,GAAG,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAChE,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnE,GAAG,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAChE,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnE,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,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvE,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CACnB,OAAU;IAEV,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;IACtC,OAAO;QACL,GAAG,OAAO;QACV,GAAG,OAAO;QACV,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/G,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACtG,CAAC;AACT,CAAC;AAED;;;;GAIG;AACH,KAAK,UAAU,aAAa,CAC1B,OAA0D,EAC1D,MAAwB,EACxB,OAA6B;IAE7B,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IACvD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;QAC/B,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,MAAM,CAAC,aAAa;QAC5D,KAAK,EAAE,CAAC,GAAG,MAAM,CAAC,KAAK,EAAE,GAAG,UAAU,CAAC;QACvC,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAChD,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrD,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACzD,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACtC,MAAM,QAAQ,GAAI,OAAO,CAAC,KAA+B,EAAE,QAAQ,IAAI,8BAA8B,CAAC;IACtG,OAAO,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;AAClC,CAAC;AAED,SAAS,oBAAoB,CAAC,OAA+B;IAC3D,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAE7C,MAAM,GAAG,GAAyC;QAChD,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,CACf,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE;YAC7B,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxE,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAChD,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnD,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACvD,CAAC;KACL,CAAC;IAEF,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS;QAAE,GAAG,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;IACtD,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS;QAAE,GAAG,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;IAC3E,IAAI,MAAM,CAAC,YAAY,KAAK,SAAS;QAAE,GAAG,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IAC9E,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS;QAAE,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IACzD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5D,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS;QAAE,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IACzD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5D,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5D,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS;QAAE,GAAG,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;IAClE,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5D,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS;QAAE,GAAG,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;IACxE,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS;QAAE,GAAG,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;IAErE,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,OAA+B;IAC7D,OAAO,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,OAAsC;IAC1E,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACpD,MAAM,GAAG,GAAyC;QAChD,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACrB,6DAA6D;YAC7D,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;gBACtB,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBAC7C,OAAO,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE;oBACpC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACxE,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAChD,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACnD,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACvD,CAAC,CAAC;YACL,CAAC;YAED,gEAAgE;YAChE,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,EAAE,CAAC;YAC9B,MAAM,IAAI,GAAG,GAAG,CAAC,eAAe,CAAC;YACjC,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC,WAAW,KAAK,GAAG,EAAE,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,6FAA6F,CAAC,CAAC;YACjH,CAAC;YAED,MAAM,KAAK,GAAG,GAAG,CAAC,gBAAgB,CAAC;YACnC,MAAM,QAAQ,GAAG,GAAG,CAAC,oBAAoB,CAAC;YAC1C,MAAM,YAAY,GAAG,GAAG,CAAC,wBAAwB,CAAC;YAElD,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,QAAQ,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,WAAW,KAAK,GAAG,EAAE,CAAC;gBACrE,MAAM,IAAI,KAAK,CAAC,iHAAiH,CAAC,CAAC;YACrI,CAAC;YAED,yFAAyF;YACzF,+FAA+F;YAC/F,MAAM,SAAS,GAAwB,KAAK;gBAC1C,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;gBACxB,CAAC,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,IAAI,IAAI,EAAE,EAAE,QAAQ,EAAE,QAAQ,IAAI,EAAE,EAAE,YAAY,EAAE,YAAY,IAAI,EAAE,EAAE,CAAC;YAEhH,MAAM,MAAM,GAA2B;gBACrC,IAAI,EAAE,IAAI,IAAI,6BAA6B;gBAC3C,SAAS,EAAE,GAAG,CAAC,WAAW,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAc,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS;gBAC9F,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,GAAG,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACpF,GAAG,CAAC,GAAG,CAAC,WAAW,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACzE,CAAC;YAEF,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC;YAClC,wFAAwF;YACxF,qFAAqF;YACrF,4EAA4E;YAC5E,OAAO,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE;gBACpC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxE,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChD,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnD,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACvD,CAAC,CAAC;QACL,CAAC;KACF,CAAC;IAEF,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS;QAAE,GAAG,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;IACtD,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS;QAAE,GAAG,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;IAC3E,IAAI,MAAM,CAAC,YAAY,KAAK,SAAS;QAAE,GAAG,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IAC9E,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS;QAAE,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IACzD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5D,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS;QAAE,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IACzD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5D,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5D,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS;QAAE,GAAG,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;IAClE,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5D,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS;QAAE,GAAG,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;IACxE,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS;QAAE,GAAG,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;IAErE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,MAAwB,EAAE,IAAc;IACtE,MAAM,OAAO,GAAG,IAAI,GAAG,EAAmB,CAAC;IAC3C,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,YAAY,GAA2B;QAC3C,KAAK,EAAE,gBAAgB;QACvB,QAAQ,EAAE,iCAAiC;QAC3C,YAAY,EAAE,uBAAuB;QACrC,QAAQ,EAAE,QAAQ;QAClB,UAAU,EAAE,qBAAqB;QACjC,OAAO,EAAE,sBAAsB;QAC/B,eAAe,EAAE,0BAA0B;QAC3C,iBAAiB,EAAE,4BAA4B;QAC/C,gBAAgB,EAAE,2BAA2B;QAC7C,eAAe,EAAE,0BAA0B;QAC3C,aAAa,EAAE,wBAAwB;QACvC,gBAAgB,EAAE,2BAA2B;KAC9C,CAAC;IAEF,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC;QAC1C,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,SAAS,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,uBAAuB,SAAS,EAAE,CAAC,CAAC;QACpF,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/ai.d.ts

@@ -0,0 +1,19 @@
+import type { ToolDef } from '@fabric-harness/sdk';
+import type { DatabricksRestClient } from './client.js';
+export interface DatabricksAiQueryToolOptions {
+    warehouseId: string;
+    name?: string;
+    description?: string;
+    pollIntervalMs?: number;
+    timeoutMs?: number;
+}
+/**
+ * `ai_query(endpoint, request)` — run model inference inside the SQL warehouse against a serving
+ * endpoint. Drives both serving and SQL-warehouse consumption. Uses named SQL parameters, so the
+ * endpoint and request are never string-interpolated into SQL.
+ */
+export declare function databricksAiQueryTool(client: DatabricksRestClient, options: DatabricksAiQueryToolOptions): ToolDef<{
+    endpoint: string;
+    request: string;
+}, unknown>;
+//# sourceMappingURL=ai.d.ts.map

package/dist/ai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.d.ts","sourceRoot":"","sources":["../src/ai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAGxD,MAAM,WAAW,4BAA4B;IAC3C,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,EAAE,4BAA4B,GAAG,OAAO,CAAC;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,EAAE,OAAO,CAAC,CAoBlK"}

package/dist/ai.js

@@ -0,0 +1,22 @@
+import { runStatement } from './sql-exec.js';
+/**
+ * `ai_query(endpoint, request)` — run model inference inside the SQL warehouse against a serving
+ * endpoint. Drives both serving and SQL-warehouse consumption. Uses named SQL parameters, so the
+ * endpoint and request are never string-interpolated into SQL.
+ */
+export function databricksAiQueryTool(client, options) {
+    const poll = { intervalMs: options.pollIntervalMs ?? 2000, timeoutMs: options.timeoutMs ?? 120_000 };
+    return {
+        name: options.name ?? 'databricks_ai_query',
+        description: options.description ?? 'Run model inference inside the SQL warehouse via ai_query() against a Databricks serving endpoint.',
+        inputSchema: {
+            type: 'object',
+            additionalProperties: false,
+            properties: { endpoint: { type: 'string' }, request: { type: 'string' } },
+            required: ['endpoint', 'request'],
+        },
+        metadata: { effect: 'execute', provider: 'databricks', service: 'ai-functions' },
+        execute: (input) => runStatement(client, options.warehouseId, 'SELECT ai_query(:endpoint, :request) AS result', [{ name: 'endpoint', value: input.endpoint }, { name: 'request', value: input.request }], poll),
+    };
+}
+//# sourceMappingURL=ai.js.map

package/dist/ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.js","sourceRoot":"","sources":["../src/ai.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAU7C;;;;GAIG;AACH,MAAM,UAAU,qBAAqB,CAAC,MAA4B,EAAE,OAAqC;IACvG,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,OAAO,CAAC,cAAc,IAAI,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,EAAE,CAAC;IACrG,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,qBAAqB;QAC3C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,oGAAoG;QACxI,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YACzE,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC;SAClC;QACD,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE;QAChF,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,YAAY,CAC9B,MAAM,EACN,OAAO,CAAC,WAAW,EACnB,gDAAgD,EAChD,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,EACxF,IAAI,CACL;KACF,CAAC;AACJ,CAAC"}

package/dist/client.d.ts

@@ -0,0 +1,37 @@
+export interface DatabricksClientOptions {
+    host: string;
+    token: string | (() => string | Promise<string>);
+    fetchImpl?: typeof fetch;
+    /**
+     * Retries *after* the initial request for transient failures (429/5xx/network). Default 3, i.e.
+     * up to 4 total attempts. This is a best-effort transport-level retry, NOT an idempotency
+     * mechanism: non-idempotent POSTs (e.g. `/sql/statements`, `/jobs/run-now`) can duplicate work if
+     * a request actually reached the server before failing. Set `maxRetries: 0` to disable, or use
+     * Databricks' own idempotency tokens where the API supports them.
+     */
+    maxRetries?: number;
+    /** Base backoff in ms (exponential per attempt; `Retry-After` honored when present). Default 500. */
+    retryBaseDelayMs?: number;
+    /** Abort signal forwarded to every request. */
+    signal?: AbortSignal;
+}
+export declare class DatabricksRestClient {
+    private readonly host;
+    private readonly token;
+    private readonly fetchImpl;
+    private readonly maxRetries;
+    private readonly retryBaseDelayMs;
+    private readonly signal;
+    constructor(options: DatabricksClientOptions);
+    get<T>(path: string, query?: Record<string, string | number | boolean | undefined>): Promise<T>;
+    post<T>(path: string, body?: unknown, query?: Record<string, string | number | boolean | undefined>): Promise<T>;
+    request<T>(method: string, path: string, body?: unknown, query?: Record<string, string | number | boolean | undefined>): Promise<T>;
+}
+export declare function createDatabricksClient(options: DatabricksClientOptions): DatabricksRestClient;
+export interface DatabricksWaitOptions {
+    pollIntervalMs?: number;
+    timeoutMs?: number;
+}
+export declare function waitForDatabricksRun(client: DatabricksRestClient, runId: number, options?: DatabricksWaitOptions): Promise<Record<string, unknown>>;
+export declare function waitForDatabricksStatement(client: DatabricksRestClient, statementId: string, options?: DatabricksWaitOptions): Promise<Record<string, unknown>>;
+//# sourceMappingURL=client.d.ts.map

package/dist/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IACjD,SAAS,CAAC,EAAE,OAAO,KAAK,CAAC;IACzB;;;;;;OAMG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,qGAAqG;IACrG,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,+CAA+C;IAC/C,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAeD,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAS;IAC9B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAmC;IACzD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAe;IACzC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA0B;gBAErC,OAAO,EAAE,uBAAuB;IAS5C,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAI/F,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAI1G,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,SAAS,CAAM,GAAG,OAAO,CAAC,CAAC,CAAC;CAyC9I;AAED,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,uBAAuB,GAAG,oBAAoB,CAE7F;AAED,MAAM,WAAW,qBAAqB;IACpC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,wBAAsB,oBAAoB,CAAC,MAAM,EAAE,oBAAoB,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,qBAA0B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAS7J;AAED,wBAAsB,0BAA0B,CAAC,MAAM,EAAE,oBAAoB,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,GAAE,qBAA0B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CASzK"}

package/dist/client.js

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

package/dist/client.js.map

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

package/dist/consumption.d.ts

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

package/dist/consumption.d.ts.map

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

package/dist/consumption.js

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

package/dist/consumption.js.map

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