@mothership/rebill-model-api 0.2.1

package/dist/__tests__/queries.unit.test.js

@@ -0,0 +1,183 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const rebill_model_client_1 = require("../rebill-model.client");
+(0, vitest_1.describe)('RebillModelClient', () => {
+    let client;
+    let mockExecuteQuery;
+    (0, vitest_1.beforeEach)(() => {
+        client = new rebill_model_client_1.RebillModelClient({
+            clientId: 'test-client-id',
+            clientSecret: 'test-secret',
+            warehouseId: 'test-warehouse',
+            workspaceUrl: 'https://test.databricks.com',
+        });
+        // Mock the private _executeQuery method
+        mockExecuteQuery = vitest_1.vi.fn();
+        client._executeQuery = mockExecuteQuery;
+    });
+    (0, vitest_1.describe)('getLatestByEdiInboundIds', () => {
+        (0, vitest_1.it)('returns empty array for empty input', async () => {
+            const result = await client.getLatestByEdiInboundIds([]);
+            (0, vitest_1.expect)(result).toEqual([]);
+            (0, vitest_1.expect)(mockExecuteQuery).not.toHaveBeenCalled();
+        });
+        (0, vitest_1.it)('returns latest predictions per EDI ID', async () => {
+            const mockRows = [
+                {
+                    edi_invoice_inbound_id: 'edi-1',
+                    inference_id: 'inf-1',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test 1"}',
+                    processed_at: '2024-01-02T00:00:00Z',
+                },
+                {
+                    edi_invoice_inbound_id: 'edi-2',
+                    inference_id: 'inf-2',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test 2"}',
+                    processed_at: '2024-01-03T00:00:00Z',
+                },
+            ];
+            mockExecuteQuery.mockResolvedValue(mockRows);
+            const result = await client.getLatestByEdiInboundIds(['edi-1', 'edi-2']);
+            (0, vitest_1.expect)(result).toHaveLength(2);
+            (0, vitest_1.expect)(result[0]).toHaveProperty('edi_invoice_inbound_id', 'edi-1');
+            (0, vitest_1.expect)(result[1]).toHaveProperty('edi_invoice_inbound_id', 'edi-2');
+            (0, vitest_1.expect)(mockExecuteQuery).toHaveBeenCalledWith(vitest_1.expect.stringContaining('ROW_NUMBER() OVER'), ['edi-1', 'edi-2'], 30000);
+        });
+        (0, vitest_1.it)('returns flattened predictions with rebill_line_items at top level', async () => {
+            const mockRows = [
+                {
+                    edi_invoice_inbound_id: 'edi-1',
+                    inference_id: 'inf-1',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test"}',
+                    processed_at: '2024-01-01T00:00:00Z',
+                },
+            ];
+            mockExecuteQuery.mockResolvedValue(mockRows);
+            const result = await client.getLatestByEdiInboundIds(['edi-1']);
+            (0, vitest_1.expect)(result[0]).toHaveProperty('rebill_line_items');
+            (0, vitest_1.expect)(result[0]).toHaveProperty('customer_summary');
+            (0, vitest_1.expect)(result[0]).not.toHaveProperty('prediction');
+            (0, vitest_1.expect)(result[0]).not.toHaveProperty('prediction_json');
+            (0, vitest_1.expect)(result[0].customer_summary).toBe('Test');
+        });
+        (0, vitest_1.it)('filters out IDs not found', async () => {
+            const mockRows = [
+                {
+                    edi_invoice_inbound_id: 'edi-1',
+                    inference_id: 'inf-1',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test"}',
+                    processed_at: '2024-01-01T00:00:00Z',
+                },
+            ];
+            mockExecuteQuery.mockResolvedValue(mockRows);
+            const result = await client.getLatestByEdiInboundIds(['edi-1', 'edi-2']);
+            (0, vitest_1.expect)(result).toHaveLength(1);
+            (0, vitest_1.expect)(result[0]).toHaveProperty('edi_invoice_inbound_id', 'edi-1');
+        });
+        (0, vitest_1.it)('respects custom timeoutMs', async () => {
+            mockExecuteQuery.mockResolvedValue([]);
+            await client.getLatestByEdiInboundIds(['edi-1'], { timeoutMs: 60000 });
+            (0, vitest_1.expect)(mockExecuteQuery).toHaveBeenCalledWith(vitest_1.expect.any(String), ['edi-1'], 60000);
+        });
+    });
+    (0, vitest_1.describe)('getByInferenceIds', () => {
+        (0, vitest_1.it)('returns empty array for empty input', async () => {
+            const result = await client.getByInferenceIds([]);
+            (0, vitest_1.expect)(result).toEqual([]);
+            (0, vitest_1.expect)(mockExecuteQuery).not.toHaveBeenCalled();
+        });
+        (0, vitest_1.it)('returns predictions by inference IDs', async () => {
+            const mockRows = [
+                {
+                    edi_invoice_inbound_id: 'edi-1',
+                    inference_id: 'inf-1',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test 1"}',
+                    processed_at: '2024-01-01T00:00:00Z',
+                },
+                {
+                    edi_invoice_inbound_id: 'edi-2',
+                    inference_id: 'inf-2',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test 2"}',
+                    processed_at: '2024-01-02T00:00:00Z',
+                },
+            ];
+            mockExecuteQuery.mockResolvedValue(mockRows);
+            const result = await client.getByInferenceIds(['inf-1', 'inf-2']);
+            (0, vitest_1.expect)(result).toHaveLength(2);
+            (0, vitest_1.expect)(result[0]).toHaveProperty('inference_id', 'inf-1');
+            (0, vitest_1.expect)(result[1]).toHaveProperty('inference_id', 'inf-2');
+            (0, vitest_1.expect)(mockExecuteQuery).toHaveBeenCalledWith(vitest_1.expect.stringContaining('WHERE inference_id IN'), ['inf-1', 'inf-2'], 30000);
+        });
+        (0, vitest_1.it)('filters out IDs not found', async () => {
+            const mockRows = [
+                {
+                    edi_invoice_inbound_id: 'edi-1',
+                    inference_id: 'inf-1',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test"}',
+                    processed_at: '2024-01-01T00:00:00Z',
+                },
+            ];
+            mockExecuteQuery.mockResolvedValue(mockRows);
+            const result = await client.getByInferenceIds(['inf-1', 'inf-2']);
+            (0, vitest_1.expect)(result).toHaveLength(1);
+            (0, vitest_1.expect)(result[0]).toHaveProperty('inference_id', 'inf-1');
+        });
+        (0, vitest_1.it)('preserves input order', async () => {
+            const mockRows = [
+                {
+                    edi_invoice_inbound_id: 'edi-2',
+                    inference_id: 'inf-2',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test 2"}',
+                    processed_at: '2024-01-02T00:00:00Z',
+                },
+                {
+                    edi_invoice_inbound_id: 'edi-1',
+                    inference_id: 'inf-1',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test 1"}',
+                    processed_at: '2024-01-01T00:00:00Z',
+                },
+            ];
+            mockExecuteQuery.mockResolvedValue(mockRows);
+            const result = await client.getByInferenceIds(['inf-1', 'inf-2']);
+            (0, vitest_1.expect)(result).toHaveLength(2);
+            (0, vitest_1.expect)(result[0]).toHaveProperty('inference_id', 'inf-1');
+            (0, vitest_1.expect)(result[1]).toHaveProperty('inference_id', 'inf-2');
+        });
+        (0, vitest_1.it)('respects custom timeoutMs', async () => {
+            mockExecuteQuery.mockResolvedValue([]);
+            await client.getByInferenceIds(['inf-1'], { timeoutMs: 60000 });
+            (0, vitest_1.expect)(mockExecuteQuery).toHaveBeenCalledWith(vitest_1.expect.any(String), ['inf-1'], 60000);
+        });
+    });
+    (0, vitest_1.describe)('getByEdiInboundIds', () => {
+        (0, vitest_1.it)('returns empty array for empty input', async () => {
+            const result = await client.getByEdiInboundIds([]);
+            (0, vitest_1.expect)(result).toEqual([]);
+            (0, vitest_1.expect)(mockExecuteQuery).not.toHaveBeenCalled();
+        });
+        (0, vitest_1.it)('returns predictions by EDI IDs', async () => {
+            const mockRows = [
+                {
+                    edi_invoice_inbound_id: 'edi-1',
+                    inference_id: 'inf-1',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test 1"}',
+                    processed_at: '2024-01-01T00:00:00Z',
+                },
+                {
+                    edi_invoice_inbound_id: 'edi-2',
+                    inference_id: 'inf-2',
+                    prediction_json: '{"rebill_line_items":[],"customer_summary":"Test 2"}',
+                    processed_at: '2024-01-02T00:00:00Z',
+                },
+            ];
+            mockExecuteQuery.mockResolvedValue(mockRows);
+            const result = await client.getByEdiInboundIds(['edi-1', 'edi-2']);
+            (0, vitest_1.expect)(result).toHaveLength(2);
+            (0, vitest_1.expect)(result[0]).toHaveProperty('edi_invoice_inbound_id', 'edi-1');
+            (0, vitest_1.expect)(result[1]).toHaveProperty('edi_invoice_inbound_id', 'edi-2');
+            (0, vitest_1.expect)(mockExecuteQuery).toHaveBeenCalledWith(vitest_1.expect.stringContaining('WHERE edi_invoice_inbound_id IN'), ['edi-1', 'edi-2'], 30000);
+        });
+    });
+});
+//# sourceMappingURL=queries.unit.test.js.map

package/dist/__tests__/queries.unit.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"queries.unit.test.js","sourceRoot":"","sources":["../../src/__tests__/queries.unit.test.ts"],"names":[],"mappings":";;AAAA,mCAA6D;AAC7D,gEAA0D;AAG1D,IAAA,iBAAQ,EAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,MAAyB,CAAA;IAC7B,IAAI,gBAA0C,CAAA;IAE9C,IAAA,mBAAU,EAAC,GAAG,EAAE;QACd,MAAM,GAAG,IAAI,uCAAiB,CAAC;YAC7B,QAAQ,EAAE,gBAAgB;YAC1B,YAAY,EAAE,aAAa;YAC3B,WAAW,EAAE,gBAAgB;YAC7B,YAAY,EAAE,6BAA6B;SAC5C,CAAC,CAAA;QAEF,wCAAwC;QACxC,gBAAgB,GAAG,WAAE,CAAC,EAAE,EAAE,CACzB;QAAC,MAAc,CAAC,aAAa,GAAG,gBAAgB,CAAA;IACnD,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,IAAA,WAAE,EAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,EAAE,CAAC,CAAA;YAExD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;YAC1B,IAAA,eAAM,EAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QACjD,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,QAAQ,GAAG;gBACf;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,sDAAsD;oBACvE,YAAY,EAAE,sBAAsB;iBACrC;gBACD;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,sDAAsD;oBACvE,YAAY,EAAE,sBAAsB;iBACrC;aACF,CAAA;YAED,gBAAgB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAA;YAE5C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;YAExE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAA;YACnE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAA;YACnE,IAAA,eAAM,EAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,eAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAC5C,CAAC,OAAO,EAAE,OAAO,CAAC,EAClB,KAAK,CACN,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,MAAM,QAAQ,GAAG;gBACf;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,oDAAoD;oBACrE,YAAY,EAAE,sBAAsB;iBACrC;aACF,CAAA;YAED,gBAAgB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAA;YAE5C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAA;YAE/D,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAA;YACrD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAA;YACpD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,CAAC,CAAA;YAClD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAA;YACvD,IAAA,eAAM,EAAE,MAAM,CAAC,CAAC,CAAsB,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,QAAQ,GAAG;gBACf;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,oDAAoD;oBACrE,YAAY,EAAE,sBAAsB;iBACrC;aACF,CAAA;YAED,gBAAgB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAA;YAE5C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;YAExE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;YAEtC,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAA;YAEtE,IAAA,eAAM,EAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,eAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,CAAA;QACrF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,IAAA,WAAE,EAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;YAEjD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;YAC1B,IAAA,eAAM,EAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QACjD,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,QAAQ,GAAG;gBACf;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,sDAAsD;oBACvE,YAAY,EAAE,sBAAsB;iBACrC;gBACD;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,sDAAsD;oBACvE,YAAY,EAAE,sBAAsB;iBACrC;aACF,CAAA;YAED,gBAAgB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAA;YAE5C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;YAEjE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAA;YACzD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAA;YACzD,IAAA,eAAM,EAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,eAAM,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,EAChD,CAAC,OAAO,EAAE,OAAO,CAAC,EAClB,KAAK,CACN,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,QAAQ,GAAG;gBACf;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,oDAAoD;oBACrE,YAAY,EAAE,sBAAsB;iBACrC;aACF,CAAA;YAED,gBAAgB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAA;YAE5C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;YAEjE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,QAAQ,GAAG;gBACf;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,sDAAsD;oBACvE,YAAY,EAAE,sBAAsB;iBACrC;gBACD;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,sDAAsD;oBACvE,YAAY,EAAE,sBAAsB;iBACrC;aACF,CAAA;YAED,gBAAgB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAA;YAE5C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;YAEjE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAA;YACzD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;YAEtC,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAA;YAE/D,IAAA,eAAM,EAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,eAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,CAAA;QACrF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,IAAA,WAAE,EAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAA;YAElD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;YAC1B,IAAA,eAAM,EAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QACjD,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,QAAQ,GAAG;gBACf;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,sDAAsD;oBACvE,YAAY,EAAE,sBAAsB;iBACrC;gBACD;oBACE,sBAAsB,EAAE,OAAO;oBAC/B,YAAY,EAAE,OAAO;oBACrB,eAAe,EAAE,sDAAsD;oBACvE,YAAY,EAAE,sBAAsB;iBACrC;aACF,CAAA;YAED,gBAAgB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAA;YAE5C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;YAElE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAA;YACnE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAA;YACnE,IAAA,eAAM,EAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,eAAM,CAAC,gBAAgB,CAAC,iCAAiC,CAAC,EAC1D,CAAC,OAAO,EAAE,OAAO,CAAC,EAClB,KAAK,CACN,CAAA;QACH,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/__tests__/schema.unit.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=schema.unit.test.d.ts.map

package/dist/__tests__/schema.unit.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.unit.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/schema.unit.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/schema.unit.test.js

@@ -0,0 +1,184 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+(0, vitest_1.describe)('RebillPrediction Schema', () => {
+    (0, vitest_1.describe)('RebillLineItem', () => {
+        (0, vitest_1.it)('has all required fields', () => {
+            const lineItem = {
+                amount: 150.5,
+                line_item_description: 'Weight difference charge',
+                root_cause: 'Shipper provided incorrect weight on BOL',
+                supporting_documentation: 'Actual scale ticket vs BOL weight',
+                type: 'INCORRECT_WEIGHT',
+            };
+            (0, vitest_1.expect)(lineItem.type).toBe('INCORRECT_WEIGHT');
+            (0, vitest_1.expect)(lineItem.amount).toBe(150.5);
+            (0, vitest_1.expect)(lineItem.line_item_description).toBe('Weight difference charge');
+            (0, vitest_1.expect)(lineItem.root_cause).toBe('Shipper provided incorrect weight on BOL');
+            (0, vitest_1.expect)(lineItem.supporting_documentation).toBe('Actual scale ticket vs BOL weight');
+        });
+    });
+    (0, vitest_1.describe)('RebillPrediction', () => {
+        (0, vitest_1.it)('has all required fields including metadata and prediction data', () => {
+            const prediction = {
+                customer_summary: 'Weight discrepancy and liftgate service required at delivery location.',
+                edi_invoice_inbound_id: '12345',
+                inference_id: 'abc-123',
+                processed_at: new Date('2024-01-01'),
+                rebill_line_items: [
+                    {
+                        amount: 150.5,
+                        line_item_description: 'Weight difference charge',
+                        root_cause: 'Shipper provided incorrect weight on BOL',
+                        supporting_documentation: 'Actual scale ticket vs BOL weight',
+                        type: 'INCORRECT_WEIGHT',
+                    },
+                    {
+                        amount: 75.0,
+                        line_item_description: 'Liftgate service at delivery',
+                        root_cause: 'Consignee location has no loading dock',
+                        supporting_documentation: 'Delivery appointment notes and photos',
+                        type: 'LIFT_GATE_AT_DELIVERY',
+                    },
+                ],
+            };
+            // Metadata fields
+            (0, vitest_1.expect)(prediction.inference_id).toBe('abc-123');
+            (0, vitest_1.expect)(prediction.edi_invoice_inbound_id).toBe('12345');
+            (0, vitest_1.expect)(prediction.processed_at).toBeInstanceOf(Date);
+            // Prediction data fields (flattened)
+            (0, vitest_1.expect)(prediction.rebill_line_items).toHaveLength(2);
+            (0, vitest_1.expect)(prediction.customer_summary).toBeTruthy();
+            (0, vitest_1.expect)(typeof prediction.customer_summary).toBe('string');
+        });
+        (0, vitest_1.it)('can have empty rebill_line_items array', () => {
+            const prediction = {
+                customer_summary: 'No rebills identified.',
+                edi_invoice_inbound_id: '12345',
+                inference_id: 'abc-123',
+                processed_at: new Date('2024-01-01'),
+                rebill_line_items: [],
+            };
+            (0, vitest_1.expect)(prediction.rebill_line_items).toHaveLength(0);
+            (0, vitest_1.expect)(prediction.customer_summary).toBe('No rebills identified.');
+        });
+        (0, vitest_1.it)('can calculate total amount from line items', () => {
+            const prediction = {
+                customer_summary: 'Two rebill items identified.',
+                edi_invoice_inbound_id: '12345',
+                inference_id: 'abc-123',
+                processed_at: new Date('2024-01-01'),
+                rebill_line_items: [
+                    {
+                        amount: 150.5,
+                        line_item_description: 'Weight difference',
+                        root_cause: 'Weight discrepancy',
+                        supporting_documentation: 'Scale ticket',
+                        type: 'INCORRECT_WEIGHT',
+                    },
+                    {
+                        amount: 75.0,
+                        line_item_description: 'Liftgate service',
+                        root_cause: 'No loading dock',
+                        supporting_documentation: 'Photos',
+                        type: 'LIFT_GATE_AT_DELIVERY',
+                    },
+                ],
+            };
+            const totalAmount = prediction.rebill_line_items.reduce((sum, item) => sum + item.amount, 0);
+            (0, vitest_1.expect)(totalAmount).toBe(225.5);
+        });
+        (0, vitest_1.it)('can extract unique rebill types from line items', () => {
+            const prediction = {
+                customer_summary: 'Multiple line items.',
+                edi_invoice_inbound_id: '12345',
+                inference_id: 'abc-123',
+                processed_at: new Date('2024-01-01'),
+                rebill_line_items: [
+                    {
+                        amount: 100.0,
+                        line_item_description: 'Weight 1',
+                        root_cause: 'Cause 1',
+                        supporting_documentation: 'Doc 1',
+                        type: 'INCORRECT_WEIGHT',
+                    },
+                    {
+                        amount: 50.0,
+                        line_item_description: 'Weight 2',
+                        root_cause: 'Cause 2',
+                        supporting_documentation: 'Doc 2',
+                        type: 'INCORRECT_WEIGHT',
+                    },
+                    {
+                        amount: 75.0,
+                        line_item_description: 'Liftgate',
+                        root_cause: 'Cause 3',
+                        supporting_documentation: 'Doc 3',
+                        type: 'LIFT_GATE_AT_DELIVERY',
+                    },
+                ],
+            };
+            const uniqueTypes = [...new Set(prediction.rebill_line_items.map((item) => item.type))];
+            (0, vitest_1.expect)(uniqueTypes).toHaveLength(2);
+            (0, vitest_1.expect)(uniqueTypes).toContain('INCORRECT_WEIGHT');
+            (0, vitest_1.expect)(uniqueTypes).toContain('LIFT_GATE_AT_DELIVERY');
+        });
+    });
+    (0, vitest_1.describe)('JSON Serialization', () => {
+        (0, vitest_1.it)('can parse prediction_json from database and merge with metadata', () => {
+            // This simulates what the SDK does internally
+            const predictionJson = JSON.stringify({
+                customer_summary: 'Weight discrepancy identified.',
+                rebill_line_items: [
+                    {
+                        amount: 150.5,
+                        line_item_description: 'Weight difference charge',
+                        root_cause: 'Shipper provided incorrect weight on BOL',
+                        supporting_documentation: 'Actual scale ticket vs BOL weight',
+                        type: 'INCORRECT_WEIGHT',
+                    },
+                ],
+            });
+            const parsed = JSON.parse(predictionJson);
+            const prediction = {
+                customer_summary: parsed.customer_summary,
+                edi_invoice_inbound_id: '12345',
+                inference_id: 'abc-123',
+                processed_at: new Date('2024-01-01'),
+                rebill_line_items: parsed.rebill_line_items,
+            };
+            (0, vitest_1.expect)(prediction.rebill_line_items).toHaveLength(1);
+            (0, vitest_1.expect)(prediction.rebill_line_items[0].type).toBe('INCORRECT_WEIGHT');
+            (0, vitest_1.expect)(prediction.rebill_line_items[0].amount).toBe(150.5);
+            (0, vitest_1.expect)(prediction.customer_summary).toBe('Weight discrepancy identified.');
+        });
+        (0, vitest_1.it)('matches the structure stored in Databricks', () => {
+            // This is the actual structure from the Python model (InferenceResponseItem)
+            const databricksJson = JSON.stringify({
+                customer_summary: 'Detention charges due to extended wait time at pickup.',
+                rebill_line_items: [
+                    {
+                        amount: 200.0,
+                        line_item_description: 'Driver waiting time exceeded',
+                        root_cause: 'Warehouse was not ready for pickup',
+                        supporting_documentation: 'Driver logs and appointment confirmation',
+                        type: 'DETENTION',
+                    },
+                ],
+            });
+            const parsed = JSON.parse(databricksJson);
+            // Verify all prediction data fields are present
+            (0, vitest_1.expect)(parsed).toHaveProperty('rebill_line_items');
+            (0, vitest_1.expect)(parsed).toHaveProperty('customer_summary');
+            (0, vitest_1.expect)(Array.isArray(parsed.rebill_line_items)).toBe(true);
+            (0, vitest_1.expect)(typeof parsed.customer_summary).toBe('string');
+            const lineItem = parsed.rebill_line_items[0];
+            (0, vitest_1.expect)(lineItem).toHaveProperty('type');
+            (0, vitest_1.expect)(lineItem).toHaveProperty('amount');
+            (0, vitest_1.expect)(lineItem).toHaveProperty('line_item_description');
+            (0, vitest_1.expect)(lineItem).toHaveProperty('root_cause');
+            (0, vitest_1.expect)(lineItem).toHaveProperty('supporting_documentation');
+        });
+    });
+});
+//# sourceMappingURL=schema.unit.test.js.map

package/dist/__tests__/schema.unit.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.unit.test.js","sourceRoot":"","sources":["../../src/__tests__/schema.unit.test.ts"],"names":[],"mappings":";;AAAA,mCAA6C;AAG7C,IAAA,iBAAQ,EAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAA,iBAAQ,EAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,IAAA,WAAE,EAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,QAAQ,GAAmB;gBAC/B,MAAM,EAAE,KAAK;gBACb,qBAAqB,EAAE,0BAA0B;gBACjD,UAAU,EAAE,0CAA0C;gBACtD,wBAAwB,EAAE,mCAAmC;gBAC7D,IAAI,EAAE,kBAAkB;aACzB,CAAA;YAED,IAAA,eAAM,EAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;YAC9C,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACnC,IAAA,eAAM,EAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAA;YACvE,IAAA,eAAM,EAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAA;YAC5E,IAAA,eAAM,EAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAA;QACrF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,IAAA,WAAE,EAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,UAAU,GAAqB;gBACnC,gBAAgB,EAAE,wEAAwE;gBAC1F,sBAAsB,EAAE,OAAO;gBAC/B,YAAY,EAAE,SAAS;gBACvB,YAAY,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;gBACpC,iBAAiB,EAAE;oBACjB;wBACE,MAAM,EAAE,KAAK;wBACb,qBAAqB,EAAE,0BAA0B;wBACjD,UAAU,EAAE,0CAA0C;wBACtD,wBAAwB,EAAE,mCAAmC;wBAC7D,IAAI,EAAE,kBAAkB;qBACzB;oBACD;wBACE,MAAM,EAAE,IAAI;wBACZ,qBAAqB,EAAE,8BAA8B;wBACrD,UAAU,EAAE,wCAAwC;wBACpD,wBAAwB,EAAE,uCAAuC;wBACjE,IAAI,EAAE,uBAAuB;qBAC9B;iBACF;aACF,CAAA;YAED,kBAAkB;YAClB,IAAA,eAAM,EAAC,UAAU,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YAC/C,IAAA,eAAM,EAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YACvD,IAAA,eAAM,EAAC,UAAU,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,CAAA;YAEpD,qCAAqC;YACrC,IAAA,eAAM,EAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACpD,IAAA,eAAM,EAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,UAAU,EAAE,CAAA;YAChD,IAAA,eAAM,EAAC,OAAO,UAAU,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,UAAU,GAAqB;gBACnC,gBAAgB,EAAE,wBAAwB;gBAC1C,sBAAsB,EAAE,OAAO;gBAC/B,YAAY,EAAE,SAAS;gBACvB,YAAY,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;gBACpC,iBAAiB,EAAE,EAAE;aACtB,CAAA;YAED,IAAA,eAAM,EAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACpD,IAAA,eAAM,EAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAA;QACpE,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,UAAU,GAAqB;gBACnC,gBAAgB,EAAE,8BAA8B;gBAChD,sBAAsB,EAAE,OAAO;gBAC/B,YAAY,EAAE,SAAS;gBACvB,YAAY,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;gBACpC,iBAAiB,EAAE;oBACjB;wBACE,MAAM,EAAE,KAAK;wBACb,qBAAqB,EAAE,mBAAmB;wBAC1C,UAAU,EAAE,oBAAoB;wBAChC,wBAAwB,EAAE,cAAc;wBACxC,IAAI,EAAE,kBAAkB;qBACzB;oBACD;wBACE,MAAM,EAAE,IAAI;wBACZ,qBAAqB,EAAE,kBAAkB;wBACzC,UAAU,EAAE,iBAAiB;wBAC7B,wBAAwB,EAAE,QAAQ;wBAClC,IAAI,EAAE,uBAAuB;qBAC9B;iBACF;aACF,CAAA;YAED,MAAM,WAAW,GAAG,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;YAE5F,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,UAAU,GAAqB;gBACnC,gBAAgB,EAAE,sBAAsB;gBACxC,sBAAsB,EAAE,OAAO;gBAC/B,YAAY,EAAE,SAAS;gBACvB,YAAY,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;gBACpC,iBAAiB,EAAE;oBACjB;wBACE,MAAM,EAAE,KAAK;wBACb,qBAAqB,EAAE,UAAU;wBACjC,UAAU,EAAE,SAAS;wBACrB,wBAAwB,EAAE,OAAO;wBACjC,IAAI,EAAE,kBAAkB;qBACzB;oBACD;wBACE,MAAM,EAAE,IAAI;wBACZ,qBAAqB,EAAE,UAAU;wBACjC,UAAU,EAAE,SAAS;wBACrB,wBAAwB,EAAE,OAAO;wBACjC,IAAI,EAAE,kBAAkB;qBACzB;oBACD;wBACE,MAAM,EAAE,IAAI;wBACZ,qBAAqB,EAAE,UAAU;wBACjC,UAAU,EAAE,SAAS;wBACrB,wBAAwB,EAAE,OAAO;wBACjC,IAAI,EAAE,uBAAuB;qBAC9B;iBACF;aACF,CAAA;YAED,MAAM,WAAW,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAEvF,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACnC,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAA;YACjD,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAA;QACxD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,IAAA,WAAE,EAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,8CAA8C;YAC9C,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC;gBACpC,gBAAgB,EAAE,gCAAgC;gBAClD,iBAAiB,EAAE;oBACjB;wBACE,MAAM,EAAE,KAAK;wBACb,qBAAqB,EAAE,0BAA0B;wBACjD,UAAU,EAAE,0CAA0C;wBACtD,wBAAwB,EAAE,mCAAmC;wBAC7D,IAAI,EAAE,kBAAkB;qBACzB;iBACF;aACF,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;YAEzC,MAAM,UAAU,GAAqB;gBACnC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;gBACzC,sBAAsB,EAAE,OAAO;gBAC/B,YAAY,EAAE,SAAS;gBACvB,YAAY,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;gBACpC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;aAC5C,CAAA;YAED,IAAA,eAAM,EAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACpD,IAAA,eAAM,EAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;YACrE,IAAA,eAAM,EAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC1D,IAAA,eAAM,EAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAA;QAC5E,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,6EAA6E;YAC7E,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC;gBACpC,gBAAgB,EAAE,wDAAwD;gBAC1E,iBAAiB,EAAE;oBACjB;wBACE,MAAM,EAAE,KAAK;wBACb,qBAAqB,EAAE,8BAA8B;wBACrD,UAAU,EAAE,oCAAoC;wBAChD,wBAAwB,EAAE,0CAA0C;wBACpE,IAAI,EAAE,WAAW;qBAClB;iBACF;aACF,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;YAEzC,gDAAgD;YAChD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAA;YAClD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAA;YACjD,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC1D,IAAA,eAAM,EAAC,OAAO,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;YAErD,MAAM,QAAQ,GAAG,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAA;YAC5C,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAA;YACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAA;YACzC,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAA;YACxD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAA;YAC7C,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,0BAA0B,CAAC,CAAA;QAC7D,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/__tests__/types.unit.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.unit.test.d.ts.map

package/dist/__tests__/types.unit.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.unit.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/types.unit.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/types.unit.test.js

@@ -0,0 +1,88 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const types_1 = require("../types");
+(0, vitest_1.describe)('RebillException', () => {
+    (0, vitest_1.describe)('constructor', () => {
+        (0, vitest_1.it)('sets the error message', () => {
+            const error = new types_1.RebillException('Test error message', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            (0, vitest_1.expect)(error.message).toBe('Test error message');
+        });
+        (0, vitest_1.it)('sets the error name to RebillException', () => {
+            const error = new types_1.RebillException('Test error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            (0, vitest_1.expect)(error.name).toBe('RebillException');
+        });
+        (0, vitest_1.it)('sets the code property', () => {
+            const error = new types_1.RebillException('Test error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            (0, vitest_1.expect)(error.code).toBe(types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+        });
+        (0, vitest_1.it)('sets the originalError when provided', () => {
+            const originalError = new Error('Original error');
+            const error = new types_1.RebillException('Test error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED, originalError);
+            (0, vitest_1.expect)(error.originalError).toBe(originalError);
+        });
+        (0, vitest_1.it)('has undefined originalError when not provided', () => {
+            const error = new types_1.RebillException('Test error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            (0, vitest_1.expect)(error.originalError).toBeUndefined();
+        });
+        (0, vitest_1.it)('has a stack trace', () => {
+            const error = new types_1.RebillException('Test error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            (0, vitest_1.expect)(error.stack).toBeDefined();
+        });
+    });
+    (0, vitest_1.describe)('prototype chain', () => {
+        (0, vitest_1.it)('is an instance of RebillException', () => {
+            const error = new types_1.RebillException('Test error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            (0, vitest_1.expect)(error).toBeInstanceOf(types_1.RebillException);
+        });
+        (0, vitest_1.it)('is an instance of Error', () => {
+            const error = new types_1.RebillException('Test error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            (0, vitest_1.expect)(error).toBeInstanceOf(Error);
+        });
+        (0, vitest_1.it)('can be caught as an Error', () => {
+            (0, vitest_1.expect)(() => {
+                throw new types_1.RebillException('Test error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            }).toThrow(Error);
+        });
+        (0, vitest_1.it)('can be caught as a RebillException', () => {
+            (0, vitest_1.expect)(() => {
+                throw new types_1.RebillException('Test error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            }).toThrow(types_1.RebillException);
+        });
+    });
+    (0, vitest_1.describe)('error codes', () => {
+        (0, vitest_1.it)('handles DATABRICKS_CONNECTION_FAILED code', () => {
+            const error = new types_1.RebillException('Connection failed', types_1.RebillErrorCode.DATABRICKS_CONNECTION_FAILED);
+            (0, vitest_1.expect)(error.code).toBe(types_1.RebillErrorCode.DATABRICKS_CONNECTION_FAILED);
+        });
+        (0, vitest_1.it)('handles SESSION_NOT_INITIALIZED code', () => {
+            const error = new types_1.RebillException('Not connected', types_1.RebillErrorCode.SESSION_NOT_INITIALIZED);
+            (0, vitest_1.expect)(error.code).toBe(types_1.RebillErrorCode.SESSION_NOT_INITIALIZED);
+        });
+        (0, vitest_1.it)('handles QUERY_EXECUTION_FAILED code', () => {
+            const error = new types_1.RebillException('Query failed', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+            (0, vitest_1.expect)(error.code).toBe(types_1.RebillErrorCode.QUERY_EXECUTION_FAILED);
+        });
+        (0, vitest_1.it)('handles QUERY_TIMEOUT code', () => {
+            const error = new types_1.RebillException('Query timed out', types_1.RebillErrorCode.QUERY_TIMEOUT);
+            (0, vitest_1.expect)(error.code).toBe(types_1.RebillErrorCode.QUERY_TIMEOUT);
+        });
+        (0, vitest_1.it)('handles RESPONSE_PARSE_FAILED code', () => {
+            const error = new types_1.RebillException('Parse failed', types_1.RebillErrorCode.RESPONSE_PARSE_FAILED);
+            (0, vitest_1.expect)(error.code).toBe(types_1.RebillErrorCode.RESPONSE_PARSE_FAILED);
+        });
+    });
+    (0, vitest_1.describe)('with original error', () => {
+        (0, vitest_1.it)('preserves original error message', () => {
+            const originalError = new Error('Database connection timeout');
+            const error = new types_1.RebillException('Failed to connect to Databricks', types_1.RebillErrorCode.DATABRICKS_CONNECTION_FAILED, originalError);
+            (0, vitest_1.expect)(error.originalError?.message).toBe('Database connection timeout');
+        });
+        (0, vitest_1.it)('preserves original error stack', () => {
+            const originalError = new Error('Original error');
+            const error = new types_1.RebillException('Wrapper error', types_1.RebillErrorCode.QUERY_EXECUTION_FAILED, originalError);
+            (0, vitest_1.expect)(error.originalError?.stack).toBeDefined();
+        });
+    });
+});
+//# sourceMappingURL=types.unit.test.js.map

package/dist/__tests__/types.unit.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.unit.test.js","sourceRoot":"","sources":["../../src/__tests__/types.unit.test.ts"],"names":[],"mappings":";;AAAA,mCAA6C;AAC7C,oCAA2D;AAE3D,IAAA,iBAAQ,EAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAA,iBAAQ,EAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,IAAA,WAAE,EAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,oBAAoB,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YAE/F,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YAEvF,IAAA,eAAM,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YAEvF,IAAA,eAAM,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,uBAAe,CAAC,sBAAsB,CAAC,CAAA;QACjE,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAA;YACjD,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,sBAAsB,EAAE,aAAa,CAAC,CAAA;YAEtG,IAAA,eAAM,EAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QACjD,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YAEvF,IAAA,eAAM,EAAC,KAAK,CAAC,aAAa,CAAC,CAAC,aAAa,EAAE,CAAA;QAC7C,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,mBAAmB,EAAE,GAAG,EAAE;YAC3B,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YAEvF,IAAA,eAAM,EAAC,KAAK,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;QACnC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,IAAA,WAAE,EAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YAEvF,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,cAAc,CAAC,uBAAe,CAAC,CAAA;QAC/C,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YAEvF,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAA;QACrC,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,IAAA,eAAM,EAAC,GAAG,EAAE;gBACV,MAAM,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YACjF,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,IAAA,eAAM,EAAC,GAAG,EAAE;gBACV,MAAM,IAAI,uBAAe,CAAC,YAAY,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YACjF,CAAC,CAAC,CAAC,OAAO,CAAC,uBAAe,CAAC,CAAA;QAC7B,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,IAAA,WAAE,EAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,mBAAmB,EAAE,uBAAe,CAAC,4BAA4B,CAAC,CAAA;YAEpG,IAAA,eAAM,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,uBAAe,CAAC,4BAA4B,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,eAAe,EAAE,uBAAe,CAAC,uBAAuB,CAAC,CAAA;YAE3F,IAAA,eAAM,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,uBAAe,CAAC,uBAAuB,CAAC,CAAA;QAClE,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,cAAc,EAAE,uBAAe,CAAC,sBAAsB,CAAC,CAAA;YAEzF,IAAA,eAAM,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,uBAAe,CAAC,sBAAsB,CAAC,CAAA;QACjE,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,iBAAiB,EAAE,uBAAe,CAAC,aAAa,CAAC,CAAA;YAEnF,IAAA,eAAM,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,uBAAe,CAAC,aAAa,CAAC,CAAA;QACxD,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,cAAc,EAAE,uBAAe,CAAC,qBAAqB,CAAC,CAAA;YAExF,IAAA,eAAM,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,uBAAe,CAAC,qBAAqB,CAAC,CAAA;QAChE,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,IAAA,WAAE,EAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAA;YAC9D,MAAM,KAAK,GAAG,IAAI,uBAAe,CAC/B,iCAAiC,EACjC,uBAAe,CAAC,4BAA4B,EAC5C,aAAa,CACd,CAAA;YAED,IAAA,eAAM,EAAC,KAAK,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAA;QAC1E,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAA;YACjD,MAAM,KAAK,GAAG,IAAI,uBAAe,CAAC,eAAe,EAAE,uBAAe,CAAC,sBAAsB,EAAE,aAAa,CAAC,CAAA;YAEzG,IAAA,eAAM,EAAC,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;QAClD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/index.d.ts

@@ -0,0 +1,11 @@
+/**
+ * Rebill Model API
+ *
+ * TypeScript API for querying Rebill model predictions from Databricks
+ *
+ * @packageDocumentation
+ */
+export type { RebillModelClientConfig } from './rebill-model.client';
+export { RebillModelClient } from './rebill-model.client';
+export { QueryOptions, RebillErrorCode, RebillException, RebillLineItem, RebillPrediction, } from './types';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,YAAY,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAA;AAEpE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAA;AAGzD,OAAO,EACL,YAAY,EACZ,eAAe,EACf,eAAe,EACf,cAAc,EACd,gBAAgB,GACjB,MAAM,SAAS,CAAA"}

package/dist/index.js

@@ -0,0 +1,18 @@
+"use strict";
+/**
+ * Rebill Model API
+ *
+ * TypeScript API for querying Rebill model predictions from Databricks
+ *
+ * @packageDocumentation
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RebillException = exports.RebillErrorCode = exports.RebillModelClient = void 0;
+// Client
+var rebill_model_client_1 = require("./rebill-model.client");
+Object.defineProperty(exports, "RebillModelClient", { enumerable: true, get: function () { return rebill_model_client_1.RebillModelClient; } });
+// Types
+var types_1 = require("./types");
+Object.defineProperty(exports, "RebillErrorCode", { enumerable: true, get: function () { return types_1.RebillErrorCode; } });
+Object.defineProperty(exports, "RebillException", { enumerable: true, get: function () { return types_1.RebillException; } });
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAGH,SAAS;AACT,6DAAyD;AAAhD,wHAAA,iBAAiB,OAAA;AAE1B,QAAQ;AACR,iCAMgB;AAJd,wGAAA,eAAe,OAAA;AACf,wGAAA,eAAe,OAAA"}