@fabric-harness/databricks 0.16.1 → 1.0.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/app.d.ts

@@ -0,0 +1,81 @@
+import type { FabricActor, FabricPrincipal, ModelProvider } from '@fabric-harness/sdk';
+import { DatabricksRestClient } from './client.js';
+import type { DatabricksPrincipal, DatabricksTokenProvider } from './identity.js';
+import type { DatabricksPersistence } from './persistence.js';
+import type { DatabricksTelemetry } from './telemetry-sink.js';
+import type { LakebasePoolFactory } from './lakebase.js';
+/**
+ * Production preset for agents hosted on Databricks Apps (v2, workstream C5).
+ *
+ * One call resolves the runtime from the standard app environment: the app's
+ * service principal (client-credentials OAuth), the workspace REST client and
+ * serving-endpoint model provider, Lakebase-backed persistence for sessions +
+ * submissions + conversation streams, and submission-keyed telemetry — all
+ * running as one governed identity that Unity Catalog enforces.
+ *
+ * ```ts
+ * const app = databricksApp();
+ * await startDevServer({ port: Number(process.env.DATABRICKS_APP_PORT ?? 8000), ...(await app.serverOptions()) });
+ * ```
+ */
+export interface DatabricksAppOptions {
+    /** Environment source. Default `process.env`. */
+    env?: Record<string, string | undefined>;
+    /** Acting identity. Default: the app service principal from the environment. */
+    principal?: DatabricksPrincipal;
+    /** Workspace host. Default `DATABRICKS_HOST`. */
+    host?: string;
+    /** Default serving endpoint for the model provider. Default `DATABRICKS_SERVING_ENDPOINT`. */
+    model?: string;
+    /**
+     * Lakebase connection. Defaults from the app's database resource
+     * environment (`DATABRICKS_LAKEBASE_HOST`/`DATABASE`/`USER`, falling back
+     * to `PGHOST`/`PGDATABASE`/`PGUSER`); the password is always the rotating
+     * OAuth token. Set `false` to run without durable persistence.
+     */
+    lakebase?: false | {
+        host?: string;
+        database?: string;
+        user?: string;
+        port?: number;
+        ssl?: boolean | {
+            rejectUnauthorized?: boolean;
+        };
+        poolFactory?: LakebasePoolFactory;
+    };
+    /** Unity Catalog principal name, when it differs from the client id. */
+    ucPrincipal?: string;
+    /** Submission-keyed telemetry into Lakebase. Default: on when Lakebase resolves. */
+    telemetry?: boolean;
+    fetchImpl?: typeof fetch;
+}
+export interface DatabricksAppRuntime {
+    identity: DatabricksTokenProvider;
+    /** Governed identity the app executes as; stamp into actors/submissions. */
+    principal: FabricPrincipal;
+    actor: FabricActor;
+    client: DatabricksRestClient;
+    modelProvider: ModelProvider;
+    persistence?: DatabricksPersistence;
+    telemetry?: DatabricksTelemetry;
+    /**
+     * The durable-store slice of `startDevServer` options (submission store +
+     * conversation stream store on Lakebase), with schema ensured. Empty when
+     * Lakebase is disabled.
+     */
+    serverOptions(): Promise<{
+        submissionStore?: import('@fabric-harness/sdk').AgentSubmissionStore;
+        conversationStreamStore?: import('@fabric-harness/sdk').ConversationStreamStore;
+    }>;
+    /**
+     * Per-request on-behalf-of identity from Databricks Apps forwarded
+     * headers, or `undefined` when the request carries no user token.
+     */
+    onBehalfOf(headers: Headers | Record<string, string | string[] | undefined>): {
+        identity: DatabricksTokenProvider;
+        actor: FabricActor;
+    } | undefined;
+    close(): Promise<void>;
+}
+export declare function databricksApp(options?: DatabricksAppOptions): DatabricksAppRuntime;
+//# sourceMappingURL=app.d.ts.map

package/dist/app.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"app.d.ts","sourceRoot":"","sources":["../src/app.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACvF,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAOnD,OAAO,KAAK,EAAE,mBAAmB,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAGlF,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAE9D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAEzD;;;;;;;;;;;;;GAaG;AACH,MAAM,WAAW,oBAAoB;IACnC,iDAAiD;IACjD,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IACzC,gFAAgF;IAChF,SAAS,CAAC,EAAE,mBAAmB,CAAC;IAChC,iDAAiD;IACjD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,8FAA8F;IAC9F,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;OAKG;IACH,QAAQ,CAAC,EAAE,KAAK,GAAG;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,OAAO,GAAG;YAAE,kBAAkB,CAAC,EAAE,OAAO,CAAA;SAAE,CAAC;QAAC,WAAW,CAAC,EAAE,mBAAmB,CAAA;KAAE,CAAC;IAC3K,wEAAwE;IACxE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,oFAAoF;IACpF,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,SAAS,CAAC,EAAE,OAAO,KAAK,CAAC;CAC1B;AAED,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,uBAAuB,CAAC;IAClC,4EAA4E;IAC5E,SAAS,EAAE,eAAe,CAAC;IAC3B,KAAK,EAAE,WAAW,CAAC;IACnB,MAAM,EAAE,oBAAoB,CAAC;IAC7B,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,CAAC,EAAE,qBAAqB,CAAC;IACpC,SAAS,CAAC,EAAE,mBAAmB,CAAC;IAChC;;;;OAIG;IACH,aAAa,IAAI,OAAO,CAAC;QACvB,eAAe,CAAC,EAAE,OAAO,qBAAqB,EAAE,oBAAoB,CAAC;QACrE,uBAAuB,CAAC,EAAE,OAAO,qBAAqB,EAAE,uBAAuB,CAAC;KACjF,CAAC,CAAC;IACH;;;OAGG;IACH,UAAU,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,GAAG;QAAE,QAAQ,EAAE,uBAAuB,CAAC;QAAC,KAAK,EAAE,WAAW,CAAA;KAAE,GAAG,SAAS,CAAC;IACpJ,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AAED,wBAAgB,aAAa,CAAC,OAAO,GAAE,oBAAyB,GAAG,oBAAoB,CA2FtF"}

package/dist/app.js

@@ -0,0 +1,87 @@
+import { DatabricksRestClient } from './client.js';
+import { appServicePrincipalFromEnv, databricksIdentity, fabricPrincipalFor, onBehalfOfFromHeaders, } from './identity.js';
+import { databricksFoundationModelProvider } from './model.js';
+import { databricksPersistence } from './persistence.js';
+import { databricksTelemetry } from './telemetry-sink.js';
+export function databricksApp(options = {}) {
+    const env = options.env ?? (typeof process !== 'undefined' ? process.env : {});
+    const principal = options.principal ?? appServicePrincipalFromEnv(env);
+    if (!principal) {
+        throw new Error('databricksApp() could not resolve an identity: pass `principal` or set DATABRICKS_HOST + DATABRICKS_CLIENT_ID + DATABRICKS_CLIENT_SECRET (the Databricks Apps runtime provides these).');
+    }
+    const host = options.host ??
+        (principal.kind === 'service-principal' ? principal.host : undefined) ??
+        (env.DATABRICKS_HOST ? (env.DATABRICKS_HOST.includes('://') ? env.DATABRICKS_HOST : `https://${env.DATABRICKS_HOST}`) : undefined);
+    if (!host) {
+        throw new Error('databricksApp() requires a workspace host: pass `host` or set DATABRICKS_HOST.');
+    }
+    const identity = databricksIdentity(principal);
+    const fabricPrincipal = fabricPrincipalFor(principal, {
+        asAppPrincipal: true,
+        ...(options.ucPrincipal ? { ucPrincipal: options.ucPrincipal } : {}),
+    });
+    const actor = { principal: fabricPrincipal };
+    const client = new DatabricksRestClient({ host, token: identity, ...(options.fetchImpl ? { fetchImpl: options.fetchImpl } : {}) });
+    const model = options.model ?? env.DATABRICKS_SERVING_ENDPOINT;
+    const modelProvider = databricksFoundationModelProvider({
+        host,
+        token: identity,
+        ...(model ? { defaultModel: model } : {}),
+        ...(options.fetchImpl ? { fetchImpl: options.fetchImpl } : {}),
+    });
+    let persistence;
+    if (options.lakebase !== false) {
+        const lakebase = options.lakebase === undefined ? {} : options.lakebase;
+        const lakebaseHost = lakebase.host ?? env.DATABRICKS_LAKEBASE_HOST ?? env.PGHOST;
+        const database = lakebase.database ?? env.DATABRICKS_LAKEBASE_DATABASE ?? env.PGDATABASE;
+        const user = lakebase.user ??
+            env.DATABRICKS_LAKEBASE_USER ??
+            env.PGUSER ??
+            (principal.kind === 'service-principal' ? principal.clientId : undefined);
+        if (lakebaseHost && database && user) {
+            persistence = databricksPersistence({
+                lakebase: {
+                    host: lakebaseHost,
+                    database,
+                    user,
+                    token: identity,
+                    ...(lakebase.port !== undefined ? { port: lakebase.port } : env.PGPORT ? { port: Number(env.PGPORT) } : {}),
+                    ...(lakebase.ssl !== undefined ? { ssl: lakebase.ssl } : {}),
+                    ...(lakebase.poolFactory ? { poolFactory: lakebase.poolFactory } : {}),
+                },
+            });
+        }
+        else if (options.lakebase !== undefined) {
+            throw new Error('databricksApp({ lakebase }) is missing connection details: provide host/database/user or the DATABRICKS_LAKEBASE_* (or PG*) environment variables from the app database resource.');
+        }
+    }
+    const telemetry = persistence && options.telemetry !== false ? databricksTelemetry({ client: persistence.client }) : undefined;
+    return {
+        identity,
+        principal: fabricPrincipal,
+        actor,
+        client,
+        modelProvider,
+        ...(persistence ? { persistence } : {}),
+        ...(telemetry ? { telemetry } : {}),
+        async serverOptions() {
+            if (!persistence)
+                return {};
+            const { submissionStore, conversationStreamStore } = await persistence.connectAll();
+            return { submissionStore, conversationStreamStore };
+        },
+        onBehalfOf(headers) {
+            const user = onBehalfOfFromHeaders(headers);
+            if (!user)
+                return undefined;
+            return {
+                identity: databricksIdentity(user),
+                actor: { principal: fabricPrincipalFor(user), onBehalfOf: user.label ?? 'user' },
+            };
+        },
+        async close() {
+            await persistence?.close();
+        },
+    };
+}
+//# sourceMappingURL=app.js.map

package/dist/app.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"app.js","sourceRoot":"","sources":["../src/app.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,EACL,0BAA0B,EAC1B,kBAAkB,EAClB,kBAAkB,EAClB,qBAAqB,GACtB,MAAM,eAAe,CAAC;AAEvB,OAAO,EAAE,iCAAiC,EAAE,MAAM,YAAY,CAAC;AAC/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAEzD,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAmE1D,MAAM,UAAU,aAAa,CAAC,UAAgC,EAAE;IAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,OAAO,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC/E,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,0BAA0B,CAAC,GAAG,CAAC,CAAC;IACvE,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,wLAAwL,CACzL,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,GACR,OAAO,CAAC,IAAI;QACZ,CAAC,SAAS,CAAC,IAAI,KAAK,mBAAmB,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QACrE,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACrI,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,gFAAgF,CAAC,CAAC;IACpG,CAAC;IAED,MAAM,QAAQ,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAC/C,MAAM,eAAe,GAAG,kBAAkB,CAAC,SAAS,EAAE;QACpD,cAAc,EAAE,IAAI;QACpB,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACrE,CAAC,CAAC;IACH,MAAM,KAAK,GAAgB,EAAE,SAAS,EAAE,eAAe,EAAE,CAAC;IAC1D,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACnI,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,GAAG,CAAC,2BAA2B,CAAC;IAC/D,MAAM,aAAa,GAAG,iCAAiC,CAAC;QACtD,IAAI;QACJ,KAAK,EAAE,QAAQ;QACf,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC/D,CAAC,CAAC;IAEH,IAAI,WAA8C,CAAC;IACnD,IAAI,OAAO,CAAC,QAAQ,KAAK,KAAK,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;QACxE,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,wBAAwB,IAAI,GAAG,CAAC,MAAM,CAAC;QACjF,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,IAAI,GAAG,CAAC,4BAA4B,IAAI,GAAG,CAAC,UAAU,CAAC;QACzF,MAAM,IAAI,GACR,QAAQ,CAAC,IAAI;YACb,GAAG,CAAC,wBAAwB;YAC5B,GAAG,CAAC,MAAM;YACV,CAAC,SAAS,CAAC,IAAI,KAAK,mBAAmB,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC5E,IAAI,YAAY,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;YACrC,WAAW,GAAG,qBAAqB,CAAC;gBAClC,QAAQ,EAAE;oBACR,IAAI,EAAE,YAAY;oBAClB,QAAQ;oBACR,IAAI;oBACJ,KAAK,EAAE,QAAQ;oBACf,GAAG,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC3G,GAAG,CAAC,QAAQ,CAAC,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5D,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACvE;aACF,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CACb,mLAAmL,CACpL,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GACb,WAAW,IAAI,OAAO,CAAC,SAAS,KAAK,KAAK,CAAC,CAAC,CAAC,mBAAmB,CAAC,EAAE,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAE/G,OAAO;QACL,QAAQ;QACR,SAAS,EAAE,eAAe;QAC1B,KAAK;QACL,MAAM;QACN,aAAa;QACb,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAEnC,KAAK,CAAC,aAAa;YACjB,IAAI,CAAC,WAAW;gBAAE,OAAO,EAAE,CAAC;YAC5B,MAAM,EAAE,eAAe,EAAE,uBAAuB,EAAE,GAAG,MAAM,WAAW,CAAC,UAAU,EAAE,CAAC;YACpF,OAAO,EAAE,eAAe,EAAE,uBAAuB,EAAE,CAAC;QACtD,CAAC;QAED,UAAU,CAAC,OAAO;YAChB,MAAM,IAAI,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,CAAC,IAAI;gBAAE,OAAO,SAAS,CAAC;YAC5B,OAAO;gBACL,QAAQ,EAAE,kBAAkB,CAAC,IAAI,CAAC;gBAClC,KAAK,EAAE,EAAE,SAAS,EAAE,kBAAkB,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,KAAK,IAAI,MAAM,EAAE;aACjF,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,KAAK;YACT,MAAM,WAAW,EAAE,KAAK,EAAE,CAAC;QAC7B,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/client.d.ts

@@ -0,0 +1,44 @@
+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;
+    /**
+     * Extra headers resolved per attempt — e.g. on-behalf-of propagation
+     * (forwarding a user's access token or workspace OBO headers) without
+     * rebuilding the client. Never logged.
+     */
+    extraHeaders?: Record<string, string> | (() => Record<string, string> | Promise<Record<string, string>>);
+}
+export declare class DatabricksRestClient {
+    private readonly host;
+    private readonly token;
+    private readonly fetchImpl;
+    private readonly maxRetries;
+    private readonly retryBaseDelayMs;
+    private readonly signal;
+    private readonly extraHeaders;
+    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;IACrB;;;;OAIG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;CAC1G;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;IACjD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA0C;gBAE3D,OAAO,EAAE,uBAAuB;IAU5C,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;CA2C9I;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"}