@mothership/rebill-model-api 0.2.1

package/dist/rebill-model.client.d.ts

@@ -0,0 +1,111 @@
+/**
+ * Rebill Model Client
+ *
+ * TypeScript client for querying rebill predictions from Databricks.
+ */
+import { QueryOptions, RebillPrediction } from './types';
+/**
+ * Configuration for RebillModelClient
+ */
+export interface RebillModelClientConfig {
+    workspaceUrl: string;
+    clientId: string;
+    clientSecret: string;
+    warehouseId: string;
+}
+export declare class RebillModelClient {
+    private _client;
+    private _session;
+    private _connectionPromise;
+    private readonly _workspaceUrl;
+    private readonly _clientId;
+    private readonly _clientSecret;
+    private readonly _warehouseId;
+    constructor(config: RebillModelClientConfig);
+    private _validate;
+    /**
+     * Close the connection to Databricks
+     * Call this when you're done using the client to release resources
+     */
+    close(): Promise<void>;
+    /**
+     * Ensure connection is established (lazy connection)
+     * Called automatically before queries
+     */
+    private _ensureConnected;
+    /**
+     * Initialize the connection to Databricks
+     */
+    private _connect;
+    /**
+     * Close the connection to Databricks
+     */
+    private _close;
+    /**
+     * Wrap a promise with a client-side timeout
+     * This ensures reliable timeout behavior regardless of API/server behavior
+     */
+    private _withTimeout;
+    /**
+     * Execute a raw SQL query (private method for internal use)
+     *
+     * @param query SQL query string
+     * @param parameters Query parameters (ordinal/positional parameters)
+     * @param timeoutMs Timeout in milliseconds
+     * @returns Query results
+     */
+    private _executeQuery;
+    /**
+     * Get predictions by EDI invoice inbound IDs
+     * Makes a single batched query to Databricks for optimal performance
+     *
+     * @param ediInboundIds Array of EDI invoice inbound IDs
+     * @param options Query options
+     * @returns Array of predictions in the same order as input IDs (missing IDs excluded)
+     *
+     * @example
+     * ```typescript
+     * const ids = ['12345', '67890', '11111'];
+     * const predictions = await rebillModelClient.getByEdiInboundIds(ids);
+     * console.log(`Found ${predictions.length} predictions`);
+     * ```
+     */
+    getByEdiInboundIds(ediInboundIds: string[], options?: QueryOptions): Promise<RebillPrediction[]>;
+    /**
+     * Get the latest prediction for each EDI invoice inbound ID
+     * Returns only the most recent prediction per EDI ID (using window function)
+     *
+     * @param ediInboundIds Array of EDI invoice inbound IDs
+     * @param options Query options
+     * @returns Array of predictions (latest per EDI ID) in the same order as input IDs (missing IDs excluded)
+     *
+     * @example
+     * ```typescript
+     * const ids = ['12345', '67890', '11111'];
+     * const predictions = await rebillModelClient.getLatestByEdiInboundIds(ids);
+     * console.log(`Found ${predictions.length} latest predictions`);
+     * ```
+     */
+    getLatestByEdiInboundIds(ediInboundIds: string[], options?: QueryOptions): Promise<RebillPrediction[]>;
+    /**
+     * Get predictions by inference IDs
+     * Makes a single batched query to Databricks for optimal performance
+     *
+     * @param inferenceIds Array of inference IDs
+     * @param options Query options
+     * @returns Array of predictions in the same order as input IDs (missing IDs excluded)
+     *
+     * @example
+     * ```typescript
+     * const ids = ['inf-12345', 'inf-67890', 'inf-11111'];
+     * const predictions = await rebillModelClient.getByInferenceIds(ids);
+     * console.log(`Found ${predictions.length} predictions`);
+     * ```
+     */
+    getByInferenceIds(inferenceIds: string[], options?: QueryOptions): Promise<RebillPrediction[]>;
+    /**
+     * Parse a database row into a RebillPrediction object
+     */
+    private _parseRow;
+}
+//# sourceMappingURL=rebill-model.client.d.ts.map

package/dist/rebill-model.client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rebill-model.client.d.ts","sourceRoot":"","sources":["../src/rebill-model.client.ts"],"names":[],"mappings":"AACA;;;;GAIG;AAIH,OAAO,EAAE,YAAY,EAAoD,gBAAgB,EAAE,MAAM,SAAS,CAAA;AAoB1G;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,YAAY,EAAE,MAAM,CAAA;IACpB,QAAQ,EAAE,MAAM,CAAA;IAChB,YAAY,EAAE,MAAM,CAAA;IACpB,WAAW,EAAE,MAAM,CAAA;CACpB;AAED,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,OAAO,CAA2B;IAC1C,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,kBAAkB,CAA6B;IACvD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAQ;IACtC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAQ;IAClC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAQ;IACtC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAQ;gBAEzB,MAAM,EAAE,uBAAuB;IAO3C,OAAO,CAAC,SAAS;IAWjB;;;OAGG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B;;;OAGG;YACW,gBAAgB;IAY9B;;OAEG;YACW,QAAQ;IA8DtB;;OAEG;YACW,MAAM;IAgBpB;;;OAGG;IACH,OAAO,CAAC,YAAY;IAkBpB;;;;;;;OAOG;YACW,aAAa;IAoC3B;;;;;;;;;;;;;;OAcG;IACG,kBAAkB,CAAC,aAAa,EAAE,MAAM,EAAE,EAAE,OAAO,GAAE,YAAiB,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAwC1G;;;;;;;;;;;;;;OAcG;IACG,wBAAwB,CAAC,aAAa,EAAE,MAAM,EAAE,EAAE,OAAO,GAAE,YAAiB,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAgDhH;;;;;;;;;;;;;;OAcG;IACG,iBAAiB,CAAC,YAAY,EAAE,MAAM,EAAE,EAAE,OAAO,GAAE,YAAiB,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAuCxG;;OAEG;IACH,OAAO,CAAC,SAAS;CAkBlB"}

package/dist/rebill-model.client.js

@@ -0,0 +1,348 @@
+"use strict";
+/* istanbul ignore file */
+/**
+ * Rebill Model Client
+ *
+ * TypeScript client for querying rebill predictions from Databricks.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RebillModelClient = void 0;
+const sql_1 = require("@databricks/sql");
+const types_1 = require("./types");
+class RebillModelClient {
+    constructor(config) {
+        this._client = null;
+        this._session = null;
+        this._connectionPromise = null;
+        this._workspaceUrl = this._validate(config.workspaceUrl, 'workspaceUrl');
+        this._clientId = this._validate(config.clientId, 'clientId');
+        this._clientSecret = this._validate(config.clientSecret, 'clientSecret');
+        this._warehouseId = this._validate(config.warehouseId, 'warehouseId');
+    }
+    _validate(value, name) {
+        if (!value) {
+            throw new Error(`Missing required config: ${name}`);
+        }
+        const trimmed = value.trim();
+        if (!trimmed) {
+            throw new Error(`Config ${name} is empty or whitespace`);
+        }
+        return trimmed;
+    }
+    /**
+     * Close the connection to Databricks
+     * Call this when you're done using the client to release resources
+     */
+    async close() {
+        await this._close();
+    }
+    /**
+     * Ensure connection is established (lazy connection)
+     * Called automatically before queries
+     */
+    async _ensureConnected() {
+        if (this._session) {
+            return;
+        }
+        if (this._connectionPromise) {
+            await this._connectionPromise;
+            return;
+        }
+        this._connectionPromise = this._connect();
+        await this._connectionPromise;
+    }
+    /**
+     * Initialize the connection to Databricks
+     */
+    async _connect() {
+        if (this._client && this._session) {
+            return;
+        }
+        try {
+            this._client = new sql_1.DBSQLClient();
+            // Normalize the workspace URL: remove protocol, trailing slashes, and any path
+            let host = this._workspaceUrl.trim();
+            // Remove protocol
+            host = host.replace(/^https?:\/\//, '');
+            // Remove trailing slash
+            host = host.replace(/\/$/, '');
+            // Remove any path component (everything after the first /)
+            const pathIndex = host.indexOf('/');
+            if (pathIndex !== -1) {
+                host = host.substring(0, pathIndex);
+            }
+            // Normalize warehouse ID: remove any leading/trailing slashes and path prefixes
+            let warehouseId = this._warehouseId.trim();
+            // Remove leading slash
+            warehouseId = warehouseId.replace(/^\//, '');
+            // Remove path prefix if present (e.g., "/sql/1.0/warehouses/")
+            warehouseId = warehouseId.replace(/^sql\/1\.0\/warehouses\//, '');
+            // Remove any remaining path components
+            const warehouseIdIndex = warehouseId.indexOf('/');
+            if (warehouseIdIndex !== -1) {
+                warehouseId = warehouseId.substring(0, warehouseIdIndex);
+            }
+            const connectionOptions = {
+                authType: 'databricks-oauth',
+                host,
+                oauthClientId: this._clientId,
+                oauthClientSecret: this._clientSecret,
+                path: `/sql/1.0/warehouses/${warehouseId}`,
+            };
+            await this._client.connect(connectionOptions);
+            this._session = await this._client.openSession();
+        }
+        catch (error) {
+            // Convert error to Error instance if it's not already
+            const errorInstance = error instanceof Error ? error : new Error(String(error));
+            // Extract more detailed error information if available
+            let errorMessage = 'Failed to connect to Databricks';
+            if (errorInstance.message) {
+                errorMessage += `: ${errorInstance.message}`;
+            }
+            // Check if it's a known error type with additional info
+            if (error && typeof error === 'object' && 'statusCode' in error) {
+                const statusCode = error.statusCode;
+                errorMessage += ` (HTTP ${statusCode})`;
+            }
+            throw new types_1.RebillException(errorMessage, types_1.RebillErrorCode.DATABRICKS_CONNECTION_FAILED, errorInstance);
+        }
+    }
+    /**
+     * Close the connection to Databricks
+     */
+    async _close() {
+        try {
+            if (this._session) {
+                await this._session.close();
+                this._session = null;
+            }
+            if (this._client) {
+                await this._client.close();
+                this._client = null;
+            }
+        }
+        catch {
+            // Silently ignore close errors - these are cleanup operations
+            // and not actionable by API consumers
+        }
+    }
+    /**
+     * Wrap a promise with a client-side timeout
+     * This ensures reliable timeout behavior regardless of API/server behavior
+     */
+    _withTimeout(promise, timeoutMs) {
+        return new Promise((resolve, reject) => {
+            const timer = setTimeout(() => {
+                reject(new types_1.RebillException(`Query timed out after ${timeoutMs}ms`, types_1.RebillErrorCode.QUERY_TIMEOUT));
+            }, timeoutMs);
+            promise
+                .then((result) => {
+                clearTimeout(timer);
+                resolve(result);
+            })
+                .catch((error) => {
+                clearTimeout(timer);
+                reject(error);
+            });
+        });
+    }
+    /**
+     * Execute a raw SQL query (private method for internal use)
+     *
+     * @param query SQL query string
+     * @param parameters Query parameters (ordinal/positional parameters)
+     * @param timeoutMs Timeout in milliseconds
+     * @returns Query results
+     */
+    async _executeQuery(query, parameters = [], timeoutMs = 30000) {
+        await this._ensureConnected();
+        const session = this._session;
+        if (!session) {
+            throw new types_1.RebillException('Client not connected. Call connect() first.', types_1.RebillErrorCode.SESSION_NOT_INITIALIZED);
+        }
+        try {
+            const queryPromise = (async () => {
+                const queryOperation = await session.executeStatement(query, {
+                    ordinalParameters: parameters,
+                    queryTimeout: timeoutMs, // Still pass to server as a hint
+                });
+                const result = await queryOperation.fetchAll();
+                await queryOperation.close();
+                return result;
+            })();
+            return await this._withTimeout(queryPromise, timeoutMs);
+        }
+        catch (error) {
+            if (error instanceof types_1.RebillException) {
+                throw error; // Re-throw timeout or other RebillExceptions
+            }
+            throw new types_1.RebillException(`Failed to execute query: ${error.message}`, types_1.RebillErrorCode.QUERY_EXECUTION_FAILED, error);
+        }
+    }
+    /**
+     * Get predictions by EDI invoice inbound IDs
+     * Makes a single batched query to Databricks for optimal performance
+     *
+     * @param ediInboundIds Array of EDI invoice inbound IDs
+     * @param options Query options
+     * @returns Array of predictions in the same order as input IDs (missing IDs excluded)
+     *
+     * @example
+     * ```typescript
+     * const ids = ['12345', '67890', '11111'];
+     * const predictions = await rebillModelClient.getByEdiInboundIds(ids);
+     * console.log(`Found ${predictions.length} predictions`);
+     * ```
+     */
+    async getByEdiInboundIds(ediInboundIds, options = {}) {
+        if (ediInboundIds.length === 0) {
+            return [];
+        }
+        const { timeoutMs = 30000 } = options;
+        try {
+            const placeholders = ediInboundIds.map(() => '?').join(', ');
+            const query = `
+        SELECT 
+          inference_id,
+          edi_invoice_inbound_id,
+          prediction_json,
+          processed_at
+        FROM rebill_model.v1.predictions
+        WHERE edi_invoice_inbound_id IN (${placeholders})
+        ORDER BY processed_at DESC
+      `;
+            const results = await this._executeQuery(query, ediInboundIds, timeoutMs);
+            const predictions = results.map((row) => this._parseRow(row));
+            // Create map for O(1) lookup and preserve input order
+            const resultMap = new Map(predictions.map((p) => [p.edi_invoice_inbound_id, p]));
+            // Return results in the same order as input IDs, filtering out missing
+            return ediInboundIds.map((id) => resultMap.get(id)).filter((p) => p !== undefined);
+        }
+        catch (error) {
+            if (error instanceof types_1.RebillException) {
+                throw error;
+            }
+            throw new types_1.RebillException(`Failed to query records: ${error.message}`, types_1.RebillErrorCode.QUERY_EXECUTION_FAILED, error);
+        }
+    }
+    /**
+     * Get the latest prediction for each EDI invoice inbound ID
+     * Returns only the most recent prediction per EDI ID (using window function)
+     *
+     * @param ediInboundIds Array of EDI invoice inbound IDs
+     * @param options Query options
+     * @returns Array of predictions (latest per EDI ID) in the same order as input IDs (missing IDs excluded)
+     *
+     * @example
+     * ```typescript
+     * const ids = ['12345', '67890', '11111'];
+     * const predictions = await rebillModelClient.getLatestByEdiInboundIds(ids);
+     * console.log(`Found ${predictions.length} latest predictions`);
+     * ```
+     */
+    async getLatestByEdiInboundIds(ediInboundIds, options = {}) {
+        if (ediInboundIds.length === 0) {
+            return [];
+        }
+        const { timeoutMs = 30000 } = options;
+        try {
+            const placeholders = ediInboundIds.map(() => '?').join(', ');
+            const query = `
+        SELECT 
+          inference_id,
+          edi_invoice_inbound_id,
+          prediction_json,
+          processed_at
+        FROM (
+          SELECT 
+            inference_id,
+            edi_invoice_inbound_id,
+            prediction_json,
+            processed_at,
+            ROW_NUMBER() OVER (PARTITION BY edi_invoice_inbound_id ORDER BY processed_at DESC) as rn
+          FROM rebill_model.v1.predictions
+          WHERE edi_invoice_inbound_id IN (${placeholders})
+        )
+        WHERE rn = 1
+      `;
+            const results = await this._executeQuery(query, ediInboundIds, timeoutMs);
+            const predictions = results.map((row) => this._parseRow(row));
+            // Create map for O(1) lookup and preserve input order
+            const resultMap = new Map(predictions.map((p) => [p.edi_invoice_inbound_id, p]));
+            // Return results in the same order as input IDs, filtering out missing
+            return ediInboundIds.map((id) => resultMap.get(id)).filter((p) => p !== undefined);
+        }
+        catch (error) {
+            if (error instanceof types_1.RebillException) {
+                throw error;
+            }
+            throw new types_1.RebillException(`Failed to query records: ${error.message}`, types_1.RebillErrorCode.QUERY_EXECUTION_FAILED, error);
+        }
+    }
+    /**
+     * Get predictions by inference IDs
+     * Makes a single batched query to Databricks for optimal performance
+     *
+     * @param inferenceIds Array of inference IDs
+     * @param options Query options
+     * @returns Array of predictions in the same order as input IDs (missing IDs excluded)
+     *
+     * @example
+     * ```typescript
+     * const ids = ['inf-12345', 'inf-67890', 'inf-11111'];
+     * const predictions = await rebillModelClient.getByInferenceIds(ids);
+     * console.log(`Found ${predictions.length} predictions`);
+     * ```
+     */
+    async getByInferenceIds(inferenceIds, options = {}) {
+        if (inferenceIds.length === 0) {
+            return [];
+        }
+        const { timeoutMs = 30000 } = options;
+        try {
+            const placeholders = inferenceIds.map(() => '?').join(', ');
+            const query = `
+        SELECT 
+          inference_id,
+          edi_invoice_inbound_id,
+          prediction_json,
+          processed_at
+        FROM rebill_model.v1.predictions
+        WHERE inference_id IN (${placeholders})
+      `;
+            const results = await this._executeQuery(query, inferenceIds, timeoutMs);
+            const predictions = results.map((row) => this._parseRow(row));
+            // Create map for O(1) lookup and preserve input order
+            const resultMap = new Map(predictions.map((p) => [p.inference_id, p]));
+            // Return results in the same order as input IDs, filtering out missing
+            return inferenceIds.map((id) => resultMap.get(id)).filter((p) => p !== undefined);
+        }
+        catch (error) {
+            if (error instanceof types_1.RebillException) {
+                throw error;
+            }
+            throw new types_1.RebillException(`Failed to query records: ${error.message}`, types_1.RebillErrorCode.QUERY_EXECUTION_FAILED, error);
+        }
+    }
+    /**
+     * Parse a database row into a RebillPrediction object
+     */
+    _parseRow(row) {
+        try {
+            const parsed = JSON.parse(row.prediction_json);
+            return {
+                customer_summary: parsed.customer_summary,
+                edi_invoice_inbound_id: row.edi_invoice_inbound_id,
+                inference_id: row.inference_id,
+                processed_at: new Date(row.processed_at),
+                rebill_line_items: parsed.rebill_line_items,
+            };
+        }
+        catch (error) {
+            throw new types_1.RebillException('Failed to parse prediction JSON', types_1.RebillErrorCode.RESPONSE_PARSE_FAILED, error);
+        }
+    }
+}
+exports.RebillModelClient = RebillModelClient;
+//# sourceMappingURL=rebill-model.client.js.map

package/dist/rebill-model.client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rebill-model.client.js","sourceRoot":"","sources":["../src/rebill-model.client.ts"],"names":[],"mappings":";AAAA,0BAA0B;AAC1B;;;;GAIG;;;AAEH,yCAA6C;AAE7C,mCAA0G;AA8B1G,MAAa,iBAAiB;IAS5B,YAAY,MAA+B;QARnC,YAAO,GAAuB,IAAI,CAAA;QAClC,aAAQ,GAAyB,IAAI,CAAA;QACrC,uBAAkB,GAAyB,IAAI,CAAA;QAOrD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAA;QACxE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAA;QAC5D,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAA;QACxE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,WAAW,EAAE,aAAa,CAAC,CAAA;IACvE,CAAC;IAEO,SAAS,CAAC,KAAa,EAAE,IAAY;QAC3C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAA;QACrD,CAAC;QACD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAA;QAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,UAAU,IAAI,yBAAyB,CAAC,CAAA;QAC1D,CAAC;QACD,OAAO,OAAO,CAAA;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,MAAM,EAAE,CAAA;IACrB,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,gBAAgB;QAC5B,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAM;QACR,CAAC;QACD,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC5B,MAAM,IAAI,CAAC,kBAAkB,CAAA;YAC7B,OAAM;QACR,CAAC;QACD,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAA;QACzC,MAAM,IAAI,CAAC,kBAAkB,CAAA;IAC/B,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,QAAQ;QACpB,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClC,OAAM;QACR,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,GAAG,IAAI,iBAAW,EAAE,CAAA;YAEhC,+EAA+E;YAC/E,IAAI,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAA;YACpC,kBAAkB;YAClB,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAA;YACvC,wBAAwB;YACxB,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;YAC9B,2DAA2D;YAC3D,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;YACnC,IAAI,SAAS,KAAK,CAAC,CAAC,EAAE,CAAC;gBACrB,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,CAAC,CAAA;YACrC,CAAC;YAED,gFAAgF;YAChF,IAAI,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAA;YAC1C,uBAAuB;YACvB,WAAW,GAAG,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;YAC5C,+DAA+D;YAC/D,WAAW,GAAG,WAAW,CAAC,OAAO,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAA;YACjE,uCAAuC;YACvC,MAAM,gBAAgB,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;YACjD,IAAI,gBAAgB,KAAK,CAAC,CAAC,EAAE,CAAC;gBAC5B,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAA;YAC1D,CAAC;YAED,MAAM,iBAAiB,GAAG;gBACxB,QAAQ,EAAE,kBAA2B;gBACrC,IAAI;gBACJ,aAAa,EAAE,IAAI,CAAC,SAAS;gBAC7B,iBAAiB,EAAE,IAAI,CAAC,aAAa;gBACrC,IAAI,EAAE,uBAAuB,WAAW,EAAE;aAC3C,CAAA;YAED,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAA;YAC7C,IAAI,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAA;QAClD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,sDAAsD;YACtD,MAAM,aAAa,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAA;YAE/E,uDAAuD;YACvD,IAAI,YAAY,GAAG,iCAAiC,CAAA;YACpD,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;gBAC1B,YAAY,IAAI,KAAK,aAAa,CAAC,OAAO,EAAE,CAAA;YAC9C,CAAC;YAED,wDAAwD;YACxD,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;gBAChE,MAAM,UAAU,GAAI,KAAiC,CAAC,UAAU,CAAA;gBAChE,YAAY,IAAI,UAAU,UAAU,GAAG,CAAA;YACzC,CAAC;YAED,MAAM,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,4BAA4B,EAAE,aAAa,CAAC,CAAA;QACtG,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,MAAM;QAClB,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;gBAC3B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAA;YACtB,CAAC;YACD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACjB,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAA;gBAC1B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;YACrB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,8DAA8D;YAC9D,sCAAsC;QACxC,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,YAAY,CAAI,OAAmB,EAAE,SAAiB;QAC5D,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,MAAM,CAAC,IAAI,uBAAe,CAAC,yBAAyB,SAAS,IAAI,EAAE,uBAAe,CAAC,aAAa,CAAC,CAAC,CAAA;YACpG,CAAC,EAAE,SAAS,CAAC,CAAA;YAEb,OAAO;iBACJ,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;gBACf,YAAY,CAAC,KAAK,CAAC,CAAA;gBACnB,OAAO,CAAC,MAAM,CAAC,CAAA;YACjB,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACf,YAAY,CAAC,KAAK,CAAC,CAAA;gBACnB,MAAM,CAAC,KAAK,CAAC,CAAA;YACf,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACJ,CAAC;IAED;;;;;;;OAOG;IACK,KAAK,CAAC,aAAa,CACzB,KAAa,EACb,aAAwB,EAAE,EAC1B,YAAoB,KAAK;QAEzB,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAA;QAE7B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAA;QAC7B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,uBAAe,CAAC,6CAA6C,EAAE,uBAAe,CAAC,uBAAuB,CAAC,CAAA;QACnH,CAAC;QAED,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC/B,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAAC,KAAK,EAAE;oBAC3D,iBAAiB,EAAE,UAAU;oBAC7B,YAAY,EAAE,SAAS,EAAE,iCAAiC;iBAC3D,CAAC,CAAA;gBACF,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,QAAQ,EAAE,CAAA;gBAC9C,MAAM,cAAc,CAAC,KAAK,EAAE,CAAA;gBAC5B,OAAO,MAAa,CAAA;YACtB,CAAC,CAAC,EAAE,CAAA;YAEJ,OAAO,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,SAAS,CAAC,CAAA;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,uBAAe,EAAE,CAAC;gBACrC,MAAM,KAAK,CAAA,CAAC,6CAA6C;YAC3D,CAAC;YACD,MAAM,IAAI,uBAAe,CACvB,4BAA6B,KAAe,CAAC,OAAO,EAAE,EACtD,uBAAe,CAAC,sBAAsB,EACtC,KAAc,CACf,CAAA;QACH,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,kBAAkB,CAAC,aAAuB,EAAE,UAAwB,EAAE;QAC1E,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAA;QACX,CAAC;QAED,MAAM,EAAE,SAAS,GAAG,KAAK,EAAE,GAAG,OAAO,CAAA;QAErC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC5D,MAAM,KAAK,GAAG;;;;;;;2CAOuB,YAAY;;OAEhD,CAAA;YAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,aAAa,EAAE,SAAS,CAAC,CAAA;YACzE,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAoB,CAAC,CAAC,CAAA;YAE9E,sDAAsD;YACtD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,sBAAsB,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAEhF,uEAAuE;YACvE,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAyB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAA;QAC3G,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,uBAAe,EAAE,CAAC;gBACrC,MAAM,KAAK,CAAA;YACb,CAAC;YACD,MAAM,IAAI,uBAAe,CACvB,4BAA6B,KAAe,CAAC,OAAO,EAAE,EACtD,uBAAe,CAAC,sBAAsB,EACtC,KAAc,CACf,CAAA;QACH,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,wBAAwB,CAAC,aAAuB,EAAE,UAAwB,EAAE;QAChF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAA;QACX,CAAC;QAED,MAAM,EAAE,SAAS,GAAG,KAAK,EAAE,GAAG,OAAO,CAAA;QAErC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC5D,MAAM,KAAK,GAAG;;;;;;;;;;;;;;6CAcyB,YAAY;;;OAGlD,CAAA;YAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,aAAa,EAAE,SAAS,CAAC,CAAA;YACzE,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAoB,CAAC,CAAC,CAAA;YAE9E,sDAAsD;YACtD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,sBAAsB,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAEhF,uEAAuE;YACvE,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAyB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAA;QAC3G,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,uBAAe,EAAE,CAAC;gBACrC,MAAM,KAAK,CAAA;YACb,CAAC;YACD,MAAM,IAAI,uBAAe,CACvB,4BAA6B,KAAe,CAAC,OAAO,EAAE,EACtD,uBAAe,CAAC,sBAAsB,EACtC,KAAc,CACf,CAAA;QACH,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,iBAAiB,CAAC,YAAsB,EAAE,UAAwB,EAAE;QACxE,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,EAAE,CAAA;QACX,CAAC;QAED,MAAM,EAAE,SAAS,GAAG,KAAK,EAAE,GAAG,OAAO,CAAA;QAErC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC3D,MAAM,KAAK,GAAG;;;;;;;iCAOa,YAAY;OACtC,CAAA;YAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,YAAY,EAAE,SAAS,CAAC,CAAA;YACxE,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAoB,CAAC,CAAC,CAAA;YAE9E,sDAAsD;YACtD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAEtE,uEAAuE;YACvE,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAyB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAA;QAC1G,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,uBAAe,EAAE,CAAC;gBACrC,MAAM,KAAK,CAAA;YACb,CAAC;YACD,MAAM,IAAI,uBAAe,CACvB,4BAA6B,KAAe,CAAC,OAAO,EAAE,EACtD,uBAAe,CAAC,sBAAsB,EACtC,KAAc,CACf,CAAA;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,SAAS,CAAC,GAAkB;QAClC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,eAAe,CAAyB,CAAA;YACtE,OAAO;gBACL,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;gBACzC,sBAAsB,EAAE,GAAG,CAAC,sBAAsB;gBAClD,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,YAAY,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC;gBACxC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;aAC5C,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,uBAAe,CACvB,iCAAiC,EACjC,uBAAe,CAAC,qBAAqB,EACrC,KAAc,CACf,CAAA;QACH,CAAC;IACH,CAAC;CACF;AA1YD,8CA0YC"}

package/dist/types.d.ts

@@ -0,0 +1,59 @@
+/**
+ * TypeScript type definitions for Rebill Model API
+ */
+/**
+ * Error codes for Rebill Model API exceptions
+ */
+export declare enum RebillErrorCode {
+    DATABRICKS_CONNECTION_FAILED = "DATABRICKS_CONNECTION_FAILED",
+    SESSION_NOT_INITIALIZED = "SESSION_NOT_INITIALIZED",
+    QUERY_EXECUTION_FAILED = "QUERY_EXECUTION_FAILED",
+    QUERY_TIMEOUT = "QUERY_TIMEOUT",
+    RESPONSE_PARSE_FAILED = "RESPONSE_PARSE_FAILED"
+}
+/**
+ * A single rebill line item with detailed information
+ */
+export interface RebillLineItem {
+    /** The rebill type (e.g., "INCORRECT_WEIGHT", "LIFT_GATE_AT_DELIVERY") */
+    type: string;
+    /** The amount for this line item */
+    amount: number;
+    /** Description of the line item from the initial rate and/or invoice */
+    line_item_description: string;
+    /** Root cause(s) of the difference between initial rate and invoice */
+    root_cause: string;
+    /** Documentation or information that supports the root cause */
+    supporting_documentation: string;
+}
+/**
+ * A rebill prediction record from the model
+ */
+export interface RebillPrediction {
+    /** Unique identifier for this inference run */
+    inference_id: string;
+    /** EDI invoice inbound ID (primary key for querying) */
+    edi_invoice_inbound_id: string;
+    /** Timestamp when prediction was processed */
+    processed_at: Date;
+    /** Array of rebill line items with detailed information */
+    rebill_line_items: RebillLineItem[];
+    /** Customer-facing summary of the rebill */
+    customer_summary: string;
+}
+/**
+ * Query options for fetching records
+ */
+export interface QueryOptions {
+    /** Timeout in milliseconds (default: 30000) */
+    timeoutMs?: number;
+}
+/**
+ * Exception thrown by the API
+ */
+export declare class RebillException extends Error {
+    readonly code: RebillErrorCode;
+    readonly originalError?: Error;
+    constructor(message: string, code: RebillErrorCode, originalError?: Error);
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH;;GAEG;AACH,oBAAY,eAAe;IAEzB,4BAA4B,iCAAiC;IAG7D,uBAAuB,4BAA4B;IAGnD,sBAAsB,2BAA2B;IACjD,aAAa,kBAAkB;IAG/B,qBAAqB,0BAA0B;CAChD;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,0EAA0E;IAC1E,IAAI,EAAE,MAAM,CAAA;IACZ,oCAAoC;IACpC,MAAM,EAAE,MAAM,CAAA;IACd,wEAAwE;IACxE,qBAAqB,EAAE,MAAM,CAAA;IAC7B,uEAAuE;IACvE,UAAU,EAAE,MAAM,CAAA;IAClB,gEAAgE;IAChE,wBAAwB,EAAE,MAAM,CAAA;CACjC;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,+CAA+C;IAC/C,YAAY,EAAE,MAAM,CAAA;IACpB,wDAAwD;IACxD,sBAAsB,EAAE,MAAM,CAAA;IAC9B,8CAA8C;IAC9C,YAAY,EAAE,IAAI,CAAA;IAClB,2DAA2D;IAC3D,iBAAiB,EAAE,cAAc,EAAE,CAAA;IACnC,4CAA4C;IAC5C,gBAAgB,EAAE,MAAM,CAAA;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,+CAA+C;IAC/C,SAAS,CAAC,EAAE,MAAM,CAAA;CACnB;AAED;;GAEG;AACH,qBAAa,eAAgB,SAAQ,KAAK;IACxC,SAAgB,IAAI,EAAE,eAAe,CAAA;IACrC,SAAgB,aAAa,CAAC,EAAE,KAAK,CAAA;gBAEzB,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,aAAa,CAAC,EAAE,KAAK;CAW1E"}

package/dist/types.js

@@ -0,0 +1,38 @@
+"use strict";
+/**
+ * TypeScript type definitions for Rebill Model API
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RebillException = exports.RebillErrorCode = void 0;
+/**
+ * Error codes for Rebill Model API exceptions
+ */
+var RebillErrorCode;
+(function (RebillErrorCode) {
+    // Connection Errors
+    RebillErrorCode["DATABRICKS_CONNECTION_FAILED"] = "DATABRICKS_CONNECTION_FAILED";
+    // Session Errors
+    RebillErrorCode["SESSION_NOT_INITIALIZED"] = "SESSION_NOT_INITIALIZED";
+    // Query Errors
+    RebillErrorCode["QUERY_EXECUTION_FAILED"] = "QUERY_EXECUTION_FAILED";
+    RebillErrorCode["QUERY_TIMEOUT"] = "QUERY_TIMEOUT";
+    // Response Errors
+    RebillErrorCode["RESPONSE_PARSE_FAILED"] = "RESPONSE_PARSE_FAILED";
+})(RebillErrorCode || (exports.RebillErrorCode = RebillErrorCode = {}));
+/**
+ * Exception thrown by the API
+ */
+class RebillException extends Error {
+    constructor(message, code, originalError) {
+        super(message);
+        this.name = 'RebillException';
+        this.code = code;
+        this.originalError = originalError;
+        // Maintains proper stack trace for where error was thrown (V8 engines)
+        if (Error.captureStackTrace) {
+            Error.captureStackTrace(this, RebillException);
+        }
+    }
+}
+exports.RebillException = RebillException;
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAEH;;GAEG;AACH,IAAY,eAaX;AAbD,WAAY,eAAe;IACzB,oBAAoB;IACpB,gFAA6D,CAAA;IAE7D,iBAAiB;IACjB,sEAAmD,CAAA;IAEnD,eAAe;IACf,oEAAiD,CAAA;IACjD,kDAA+B,CAAA;IAE/B,kBAAkB;IAClB,kEAA+C,CAAA;AACjD,CAAC,EAbW,eAAe,+BAAf,eAAe,QAa1B;AA0CD;;GAEG;AACH,MAAa,eAAgB,SAAQ,KAAK;IAIxC,YAAY,OAAe,EAAE,IAAqB,EAAE,aAAqB;QACvE,KAAK,CAAC,OAAO,CAAC,CAAA;QACd,IAAI,CAAC,IAAI,GAAG,iBAAiB,CAAA;QAC7B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAA;QAElC,uEAAuE;QACvE,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;YAC5B,KAAK,CAAC,iBAAiB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAA;QAChD,CAAC;IACH,CAAC;CACF;AAfD,0CAeC"}

package/example.ts

@@ -0,0 +1,91 @@
+/**
+ * Example: Using the Rebill Model API
+ *
+ * Demonstrates all available query methods.
+ *
+ * Usage:
+ *   1. Copy .env.example to .env and fill in your credentials
+ *   2. Run: pnpm tsx example.ts
+ */
+
+import 'dotenv/config'
+import { RebillErrorCode, RebillException, RebillModelClient } from './src'
+
+async function main() {
+  // Initialize client with configuration from environment
+  const client = new RebillModelClient({
+    clientId: process.env.DATABRICKS_CLIENT_ID!,
+    clientSecret: process.env.DATABRICKS_CLIENT_SECRET!,
+    warehouseId: process.env.DATABRICKS_WAREHOUSE_ID!,
+    workspaceUrl: process.env.DATABRICKS_WORKSPACE_URL!,
+  })
+
+  try {
+    // Example EDI IDs to query (replace with real IDs)
+    const ediIds = ['12345', '67890']
+
+    console.log('--- Get Latest Predictions by EDI IDs ---')
+    const latest = await client.getLatestByEdiInboundIds(ediIds)
+    console.log(`Found ${latest.length} predictions`)
+
+    for (const prediction of latest) {
+      console.log(`\n  EDI ID: ${prediction.edi_invoice_inbound_id}`)
+      console.log(`  Inference ID: ${prediction.inference_id}`)
+      console.log(`  Processed At: ${prediction.processed_at.toISOString()}`)
+      console.log(`  Line Items: ${prediction.rebill_line_items.length}`)
+      console.log(`  Summary: ${prediction.customer_summary}`)
+
+      // Calculate total rebill amount
+      const total = prediction.rebill_line_items.reduce((sum, item) => sum + item.amount, 0)
+      console.log(`  Total Amount: $${total.toFixed(2)}`)
+    }
+
+    console.log('\n--- Get All Predictions by EDI IDs (including history) ---')
+    const all = await client.getByEdiInboundIds(ediIds)
+    console.log(`Found ${all.length} total predictions`)
+
+    console.log('\n--- Get Predictions by Inference IDs ---')
+    if (latest.length > 0) {
+      const inferenceIds = latest.map((p) => p.inference_id)
+      const byInference = await client.getByInferenceIds(inferenceIds)
+      console.log(`Found ${byInference.length} predictions by inference ID`)
+    }
+
+    console.log('\n--- Query with Custom Timeout ---')
+    const withTimeout = await client.getLatestByEdiInboundIds(ediIds, {
+      timeoutMs: 60000, // 60 seconds
+    })
+    console.log(`Found ${withTimeout.length} predictions (60s timeout)`)
+
+    console.log('\n✅ All examples completed successfully')
+  } catch (error) {
+    if (error instanceof RebillException) {
+      console.error('\n❌ RebillException:', error.code)
+      console.error('   Message:', error.message)
+
+      // Handle specific error codes
+      switch (error.code) {
+        case RebillErrorCode.DATABRICKS_CONNECTION_FAILED:
+          console.error('   → Check your credentials and network connection')
+          break
+        case RebillErrorCode.QUERY_TIMEOUT:
+          console.error('   → Try increasing the timeout or reducing the query size')
+          break
+        case RebillErrorCode.QUERY_EXECUTION_FAILED:
+          console.error('   → Check your query parameters')
+          break
+      }
+
+      if (error.originalError) {
+        console.error('   Original:', error.originalError.message)
+      }
+    } else {
+      console.error('Unexpected error:', error)
+    }
+  } finally {
+    // Always clean up connection
+    await client.close()
+  }
+}
+
+main().catch(console.error)