@fgv/ts-extras-transformers 5.1.0-47 → 5.1.0-49

package/lib/test/unit/transformers.test.js

@@ -1,269 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-jest.mock('@huggingface/transformers');
-require("@fgv/ts-utils-jest");
-const upstream = __importStar(require("@huggingface/transformers"));
-const index_1 = require("../../index");
-// ─── loadPipeline ──────────────────────────────────────────────────────────────
-describe('loadPipeline', () => {
-    beforeEach(() => {
-        jest.resetAllMocks();
-    });
-    test('returns Success wrapping the upstream pipeline on success', async () => {
-        const mockPipeline = jest.fn();
-        jest.mocked(upstream.pipeline).mockResolvedValueOnce(mockPipeline);
-        const result = await (0, index_1.loadPipeline)('text-classification', 'Xenova/distilbert-base-uncased-finetuned-sst-2-english');
-        expect(result).toSucceedWith(mockPipeline);
-    });
-    test('passes task and model to the upstream pipeline factory', async () => {
-        const mockPipeline = jest.fn();
-        jest.mocked(upstream.pipeline).mockResolvedValueOnce(mockPipeline);
-        await (0, index_1.loadPipeline)('text-classification', 'some-model-id');
-        expect(upstream.pipeline).toHaveBeenCalledWith('text-classification', 'some-model-id', undefined);
-    });
-    test('passes options through to the upstream pipeline factory', async () => {
-        const mockPipeline = jest.fn();
-        jest.mocked(upstream.pipeline).mockResolvedValueOnce(mockPipeline);
-        const opts = { device: 'cpu' };
-        await (0, index_1.loadPipeline)('text-classification', 'some-model-id', opts);
-        expect(upstream.pipeline).toHaveBeenCalledWith('text-classification', 'some-model-id', opts);
-    });
-    test('returns Success when model is omitted', async () => {
-        const mockPipeline = jest.fn();
-        jest.mocked(upstream.pipeline).mockResolvedValueOnce(mockPipeline);
-        const result = await (0, index_1.loadPipeline)('feature-extraction');
-        expect(result).toSucceed();
-        expect(upstream.pipeline).toHaveBeenCalledWith('feature-extraction', undefined, undefined);
-    });
-    test('returns Failure capturing upstream error message on network failure', async () => {
-        jest
-            .mocked(upstream.pipeline)
-            .mockRejectedValueOnce(new Error('Could not locate the model: model-not-found'));
-        const result = await (0, index_1.loadPipeline)('text-classification', 'model-not-found');
-        expect(result).toFailWith(/could not locate the model/i);
-    });
-    test('returns Failure capturing upstream error on ONNX init failure', async () => {
-        jest.mocked(upstream.pipeline).mockRejectedValueOnce(new Error('Failed to initialize ONNX runtime'));
-        const result = await (0, index_1.loadPipeline)('text-classification', 'some-model');
-        expect(result).toFailWith(/failed to initialize onnx runtime/i);
-    });
-});
-// ─── classify ─────────────────────────────────────────────────────────────────
-describe('classify', () => {
-    let mockClassifier;
-    beforeEach(() => {
-        jest.resetAllMocks();
-        mockClassifier = jest.fn();
-    });
-    test('returns Success wrapping TextClassificationOutput on success', async () => {
-        const mockOutput = [{ label: 'POSITIVE', score: 0.9998 }];
-        mockClassifier.mockResolvedValueOnce(mockOutput);
-        const result = await (0, index_1.classify)(mockClassifier, 'I love transformers!');
-        expect(result).toSucceedWith(mockOutput);
-    });
-    test('passes text to the upstream classifier', async () => {
-        const mockOutput = [{ label: 'NEGATIVE', score: 0.9997 }];
-        mockClassifier.mockResolvedValueOnce(mockOutput);
-        await (0, index_1.classify)(mockClassifier, 'I hate bugs');
-        expect(mockClassifier).toHaveBeenCalledWith('I hate bugs', undefined);
-    });
-    test('passes options through to the upstream classifier', async () => {
-        const mockOutput = [
-            { label: 'POSITIVE', score: 0.9998 },
-            { label: 'NEGATIVE', score: 0.0002 }
-        ];
-        mockClassifier.mockResolvedValueOnce(mockOutput);
-        const opts = { top_k: null };
-        await (0, index_1.classify)(mockClassifier, 'Hello world', opts);
-        expect(mockClassifier).toHaveBeenCalledWith('Hello world', opts);
-    });
-    test('normalises flat array result correctly', async () => {
-        const mockOutput = [
-            { label: 'SAFE', score: 0.95 },
-            { label: 'UNSAFE', score: 0.05 }
-        ];
-        mockClassifier.mockResolvedValueOnce(mockOutput);
-        expect(await (0, index_1.classify)(mockClassifier, 'hello')).toSucceedWith(mockOutput);
-    });
-    test('normalises nested array result by flattening one level', async () => {
-        // When upstream returns array-of-arrays (e.g. batch input path leaked through),
-        // we flatten one level so consumers always receive a flat TextClassificationOutput.
-        const inner = [{ label: 'POSITIVE', score: 0.9 }];
-        mockClassifier.mockResolvedValueOnce([inner]);
-        expect(await (0, index_1.classify)(mockClassifier, 'hello')).toSucceedAndSatisfy((output) => {
-            expect(output).toEqual([{ label: 'POSITIVE', score: 0.9 }]);
-        });
-    });
-    test('returns Failure capturing upstream error on inference failure', async () => {
-        mockClassifier.mockRejectedValueOnce(new Error('Inference session error: out of memory'));
-        const result = await (0, index_1.classify)(mockClassifier, 'some text');
-        expect(result).toFailWith(/inference session error/i);
-    });
-    test('returns Failure capturing upstream tokenisation error', async () => {
-        mockClassifier.mockRejectedValueOnce(new Error('Tokenisation failed: unexpected token'));
-        const result = await (0, index_1.classify)(mockClassifier, 'bad input');
-        expect(result).toFailWith(/tokenisation failed/i);
-    });
-});
-// ─── classifyAll ──────────────────────────────────────────────────────────────
-describe('classifyAll', () => {
-    let mockClassifier;
-    beforeEach(() => {
-        jest.resetAllMocks();
-        mockClassifier = jest.fn();
-    });
-    test('forces top_k: null on the underlying classify call', async () => {
-        const mockOutput = [
-            { label: 'POSITIVE', score: 0.9998 },
-            { label: 'NEGATIVE', score: 0.0002 }
-        ];
-        mockClassifier.mockResolvedValueOnce(mockOutput);
-        await (0, index_1.classifyAll)(mockClassifier, 'hello');
-        expect(mockClassifier).toHaveBeenCalledWith('hello', { top_k: null });
-    });
-    test('overrides caller-supplied top_k with null', async () => {
-        const mockOutput = [
-            { label: 'POSITIVE', score: 0.9998 },
-            { label: 'NEGATIVE', score: 0.0002 }
-        ];
-        mockClassifier.mockResolvedValueOnce(mockOutput);
-        await (0, index_1.classifyAll)(mockClassifier, 'hello', { top_k: 1 });
-        expect(mockClassifier).toHaveBeenCalledWith('hello', { top_k: null });
-    });
-    test('returns all labels on success', async () => {
-        const mockOutput = [
-            { label: 'POSITIVE', score: 0.9998 },
-            { label: 'NEGATIVE', score: 0.0002 }
-        ];
-        mockClassifier.mockResolvedValueOnce(mockOutput);
-        const result = await (0, index_1.classifyAll)(mockClassifier, 'I love transformers!');
-        expect(result).toSucceedWith(mockOutput);
-    });
-    test('propagates upstream inference failure as Failure', async () => {
-        mockClassifier.mockRejectedValueOnce(new Error('Inference session error: out of memory'));
-        const result = await (0, index_1.classifyAll)(mockClassifier, 'some text');
-        expect(result).toFailWith(/inference session error/i);
-    });
-});
-// ─── embed ────────────────────────────────────────────────────────────────────
-describe('embed', () => {
-    let mockExtractor;
-    beforeEach(() => {
-        jest.resetAllMocks();
-        mockExtractor = jest.fn();
-    });
-    test('returns Success wrapping the upstream Tensor on success', async () => {
-        const mockTensor = {
-            type: 'float32',
-            data: new Float32Array([0.1, 0.2, 0.3]),
-            dims: [1, 3]
-        };
-        mockExtractor.mockResolvedValueOnce(mockTensor);
-        const result = await (0, index_1.embed)(mockExtractor, 'This is a test.');
-        expect(result).toSucceedWith(mockTensor);
-    });
-    test('passes text to the upstream extractor', async () => {
-        const mockTensor = { type: 'float32', data: new Float32Array([0.1]), dims: [1, 1] };
-        mockExtractor.mockResolvedValueOnce(mockTensor);
-        await (0, index_1.embed)(mockExtractor, 'hello world');
-        expect(mockExtractor).toHaveBeenCalledWith('hello world', undefined);
-    });
-    test('passes string array to the upstream extractor', async () => {
-        const mockTensor = {
-            type: 'float32',
-            data: new Float32Array([0.1, 0.2]),
-            dims: [2, 1]
-        };
-        mockExtractor.mockResolvedValueOnce(mockTensor);
-        await (0, index_1.embed)(mockExtractor, ['text one', 'text two']);
-        expect(mockExtractor).toHaveBeenCalledWith(['text one', 'text two'], undefined);
-    });
-    test('passes options through to the upstream extractor', async () => {
-        const mockTensor = { type: 'float32', data: new Float32Array([0.5]), dims: [1, 1] };
-        mockExtractor.mockResolvedValueOnce(mockTensor);
-        const opts = { pooling: 'mean', normalize: true };
-        await (0, index_1.embed)(mockExtractor, 'some text', opts);
-        expect(mockExtractor).toHaveBeenCalledWith('some text', opts);
-    });
-    test('returns Failure capturing upstream inference error', async () => {
-        mockExtractor.mockRejectedValueOnce(new Error('Inference session error: out of memory'));
-        const result = await (0, index_1.embed)(mockExtractor, 'some text');
-        expect(result).toFailWith(/inference session error/i);
-    });
-    test('returns Failure capturing upstream tokenisation error', async () => {
-        mockExtractor.mockRejectedValueOnce(new Error('Tokenisation failed: input too long'));
-        const result = await (0, index_1.embed)(mockExtractor, 'bad input');
-        expect(result).toFailWith(/tokenisation failed/i);
-    });
-});
-// ─── summarize ──────────────────────────────────────────────────────────────────
-describe('summarize', () => {
-    let mockSummarizer;
-    beforeEach(() => {
-        jest.resetAllMocks();
-        mockSummarizer = jest.fn();
-    });
-    test('returns Success wrapping SummarizationOutput on success', async () => {
-        const mockOutput = [{ summary_text: 'A short summary.' }];
-        mockSummarizer.mockResolvedValueOnce(mockOutput);
-        const result = await (0, index_1.summarize)(mockSummarizer, 'A long article that needs summarizing.');
-        expect(result).toSucceedWith(mockOutput);
-    });
-    test('passes text to the upstream summarizer', async () => {
-        const mockOutput = [{ summary_text: 'Summary.' }];
-        mockSummarizer.mockResolvedValueOnce(mockOutput);
-        await (0, index_1.summarize)(mockSummarizer, 'some long document');
-        expect(mockSummarizer).toHaveBeenCalledWith('some long document', undefined);
-    });
-    test('passes options through to the upstream summarizer', async () => {
-        const mockOutput = [{ summary_text: 'Bounded summary.' }];
-        mockSummarizer.mockResolvedValueOnce(mockOutput);
-        const opts = { min_length: 10, max_length: 50 };
-        await (0, index_1.summarize)(mockSummarizer, 'a document', opts);
-        expect(mockSummarizer).toHaveBeenCalledWith('a document', opts);
-    });
-    test('returns Failure capturing upstream inference error', async () => {
-        mockSummarizer.mockRejectedValueOnce(new Error('Inference session error: out of memory'));
-        const result = await (0, index_1.summarize)(mockSummarizer, 'some text');
-        expect(result).toFailWith(/inference session error/i);
-    });
-    test('returns Failure capturing upstream tokenisation error', async () => {
-        mockSummarizer.mockRejectedValueOnce(new Error('Tokenisation failed: input too long'));
-        const result = await (0, index_1.summarize)(mockSummarizer, 'bad input');
-        expect(result).toFailWith(/tokenisation failed/i);
-    });
-});
-//# sourceMappingURL=transformers.test.js.map

package/lib/test/unit/transformers.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"transformers.test.js","sourceRoot":"","sources":["../../../src/test/unit/transformers.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,IAAI,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;AAEvC,8BAA4B;AAC5B,oEAAsD;AACtD,uCAaqB;AAErB,kFAAkF;AAElF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAgD,CAAC;QAC7E,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;QAEnE,MAAM,MAAM,GAAG,MAAM,IAAA,oBAAY,EAC/B,qBAAqB,EACrB,wDAAwD,CACzD,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAgD,CAAC;QAC7E,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;QAEnE,MAAM,IAAA,oBAAY,EAAC,qBAAqB,EAAE,eAAe,CAAC,CAAC;QAC3D,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,qBAAqB,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC;IACpG,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAgD,CAAC;QAC7E,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;QAEnE,MAAM,IAAI,GAAG,EAAE,MAAM,EAAE,KAAK,EAA6C,CAAC;QAC1E,MAAM,IAAA,oBAAY,EAAC,qBAAqB,EAAE,eAAe,EAAE,IAAI,CAAC,CAAC;QACjE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,qBAAqB,EAAE,eAAe,EAAE,IAAI,CAAC,CAAC;IAC/F,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAA+C,CAAC;QAC5E,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;QAEnE,MAAM,MAAM,GAAG,MAAM,IAAA,oBAAY,EAAC,oBAAoB,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,CAAC;QAC3B,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACrF,IAAI;aACD,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;aACzB,qBAAqB,CAAC,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC,CAAC;QAEnF,MAAM,MAAM,GAAG,MAAM,IAAA,oBAAY,EAAC,qBAAqB,EAAE,iBAAiB,CAAC,CAAC;QAC5E,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC/E,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC,CAAC;QAErG,MAAM,MAAM,GAAG,MAAM,IAAA,oBAAY,EAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,oCAAoC,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,iFAAiF;AAEjF,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,IAAI,cAA+D,CAAC;IAEpE,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,cAAc,GAAG,IAAI,CAAC,EAAE,EAAgE,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,UAAU,GAA6B,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QACpF,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,MAAM,GAAG,MAAM,IAAA,gBAAQ,EAAC,cAAc,EAAE,sBAAsB,CAAC,CAAC;QACtE,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,UAAU,GAA6B,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QACpF,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,IAAA,gBAAQ,EAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QAC9C,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,UAAU,GAA6B;YAC3C,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE;YACpC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE;SACrC,CAAC;QACF,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,IAAI,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAC7B,MAAM,IAAA,gBAAQ,EAAC,cAAc,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;QACpD,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,UAAU,GAA6B;YAC3C,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE;YAC9B,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE;SACjC,CAAC;QACF,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,IAAA,gBAAQ,EAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACxE,gFAAgF;QAChF,oFAAoF;QACpF,MAAM,KAAK,GAA6B,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5E,cAAc,CAAC,qBAAqB,CAAC,CAAC,KAAK,CAAwC,CAAC,CAAC;QAErF,MAAM,CAAC,MAAM,IAAA,gBAAQ,EAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,MAAM,EAAE,EAAE;YAC7E,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC/E,cAAc,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC,CAAC;QAE1F,MAAM,MAAM,GAAG,MAAM,IAAA,gBAAQ,EAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,0BAA0B,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACvE,cAAc,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC,CAAC;QAEzF,MAAM,MAAM,GAAG,MAAM,IAAA,gBAAQ,EAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,iFAAiF;AAEjF,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,cAA+D,CAAC;IAEpE,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,cAAc,GAAG,IAAI,CAAC,EAAE,EAAgE,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,UAAU,GAA6B;YAC3C,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE;YACpC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE;SACrC,CAAC;QACF,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,IAAA,mBAAW,EAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAC3C,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,UAAU,GAA6B;YAC3C,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE;YACpC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE;SACrC,CAAC;QACF,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,IAAA,mBAAW,EAAC,cAAc,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACzD,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,UAAU,GAA6B;YAC3C,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE;YACpC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE;SACrC,CAAC;QACF,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,MAAM,GAAG,MAAM,IAAA,mBAAW,EAAC,cAAc,EAAE,sBAAsB,CAAC,CAAC;QACzE,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAClE,cAAc,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC,CAAC;QAE1F,MAAM,MAAM,GAAG,MAAM,IAAA,mBAAW,EAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,0BAA0B,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,iFAAiF;AAEjF,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;IACrB,IAAI,aAA6D,CAAC;IAElE,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,aAAa,GAAG,IAAI,CAAC,EAAE,EAA+D,CAAC;IACzF,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,UAAU,GAAG;YACjB,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,IAAI,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACvC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;SACQ,CAAC;QACvB,aAAa,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEhD,MAAM,MAAM,GAAG,MAAM,IAAA,aAAK,EAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,UAAU,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAuB,CAAC;QACzG,aAAa,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEhD,MAAM,IAAA,aAAK,EAAC,aAAa,EAAE,aAAa,CAAC,CAAC;QAC1C,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,UAAU,GAAG;YACjB,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,IAAI,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAClC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;SACQ,CAAC;QACvB,aAAa,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEhD,MAAM,IAAA,aAAK,EAAC,aAAa,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,CAAC,UAAU,EAAE,UAAU,CAAC,EAAE,SAAS,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,UAAU,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAuB,CAAC;QACzG,aAAa,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEhD,MAAM,IAAI,GAAG,EAAE,OAAO,EAAE,MAAe,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QAC3D,MAAM,IAAA,aAAK,EAAC,aAAa,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACpE,aAAa,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC,CAAC;QAEzF,MAAM,MAAM,GAAG,MAAM,IAAA,aAAK,EAAC,aAAa,EAAE,WAAW,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,0BAA0B,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACvE,aAAa,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC,CAAC;QAEtF,MAAM,MAAM,GAAG,MAAM,IAAA,aAAK,EAAC,aAAa,EAAE,WAAW,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,mFAAmF;AAEnF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,cAA0D,CAAC;IAE/D,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,cAAc,GAAG,IAAI,CAAC,EAAE,EAA2D,CAAC;IACtF,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,UAAU,GAAwB,CAAC,EAAE,YAAY,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAC/E,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAS,EAAC,cAAc,EAAE,wCAAwC,CAAC,CAAC;QACzF,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,UAAU,GAAwB,CAAC,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,CAAC;QACvE,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,IAAA,iBAAS,EAAC,cAAc,EAAE,oBAAoB,CAAC,CAAC;QACtD,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,EAAE,SAAS,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,UAAU,GAAwB,CAAC,EAAE,YAAY,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAC/E,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QAChD,MAAM,IAAA,iBAAS,EAAC,cAAc,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;QACpD,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACpE,cAAc,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC,CAAC;QAE1F,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAS,EAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,0BAA0B,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACvE,cAAc,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC,CAAC;QAEvF,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAS,EAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["jest.mock('@huggingface/transformers');\n\nimport '@fgv/ts-utils-jest';\nimport * as upstream from '@huggingface/transformers';\nimport {\n  loadPipeline,\n  classify,\n  classifyAll,\n  embed,\n  summarize,\n  type TextClassificationPipeline,\n  type TextClassificationOutput,\n  type FeatureExtractionPipeline,\n  type SummarizationPipeline,\n  type SummarizationOutput,\n  type Tensor,\n  type AllTasks\n} from '../../index';\n\n// ─── loadPipeline ──────────────────────────────────────────────────────────────\n\ndescribe('loadPipeline', () => {\n  beforeEach(() => {\n    jest.resetAllMocks();\n  });\n\n  test('returns Success wrapping the upstream pipeline on success', async () => {\n    const mockPipeline = jest.fn() as unknown as AllTasks['text-classification'];\n    jest.mocked(upstream.pipeline).mockResolvedValueOnce(mockPipeline);\n\n    const result = await loadPipeline(\n      'text-classification',\n      'Xenova/distilbert-base-uncased-finetuned-sst-2-english'\n    );\n    expect(result).toSucceedWith(mockPipeline);\n  });\n\n  test('passes task and model to the upstream pipeline factory', async () => {\n    const mockPipeline = jest.fn() as unknown as AllTasks['text-classification'];\n    jest.mocked(upstream.pipeline).mockResolvedValueOnce(mockPipeline);\n\n    await loadPipeline('text-classification', 'some-model-id');\n    expect(upstream.pipeline).toHaveBeenCalledWith('text-classification', 'some-model-id', undefined);\n  });\n\n  test('passes options through to the upstream pipeline factory', async () => {\n    const mockPipeline = jest.fn() as unknown as AllTasks['text-classification'];\n    jest.mocked(upstream.pipeline).mockResolvedValueOnce(mockPipeline);\n\n    const opts = { device: 'cpu' } as Parameters<typeof upstream.pipeline>[2];\n    await loadPipeline('text-classification', 'some-model-id', opts);\n    expect(upstream.pipeline).toHaveBeenCalledWith('text-classification', 'some-model-id', opts);\n  });\n\n  test('returns Success when model is omitted', async () => {\n    const mockPipeline = jest.fn() as unknown as AllTasks['feature-extraction'];\n    jest.mocked(upstream.pipeline).mockResolvedValueOnce(mockPipeline);\n\n    const result = await loadPipeline('feature-extraction');\n    expect(result).toSucceed();\n    expect(upstream.pipeline).toHaveBeenCalledWith('feature-extraction', undefined, undefined);\n  });\n\n  test('returns Failure capturing upstream error message on network failure', async () => {\n    jest\n      .mocked(upstream.pipeline)\n      .mockRejectedValueOnce(new Error('Could not locate the model: model-not-found'));\n\n    const result = await loadPipeline('text-classification', 'model-not-found');\n    expect(result).toFailWith(/could not locate the model/i);\n  });\n\n  test('returns Failure capturing upstream error on ONNX init failure', async () => {\n    jest.mocked(upstream.pipeline).mockRejectedValueOnce(new Error('Failed to initialize ONNX runtime'));\n\n    const result = await loadPipeline('text-classification', 'some-model');\n    expect(result).toFailWith(/failed to initialize onnx runtime/i);\n  });\n});\n\n// ─── classify ─────────────────────────────────────────────────────────────────\n\ndescribe('classify', () => {\n  let mockClassifier: jest.MockedFunction<TextClassificationPipeline>;\n\n  beforeEach(() => {\n    jest.resetAllMocks();\n    mockClassifier = jest.fn() as unknown as jest.MockedFunction<TextClassificationPipeline>;\n  });\n\n  test('returns Success wrapping TextClassificationOutput on success', async () => {\n    const mockOutput: TextClassificationOutput = [{ label: 'POSITIVE', score: 0.9998 }];\n    mockClassifier.mockResolvedValueOnce(mockOutput);\n\n    const result = await classify(mockClassifier, 'I love transformers!');\n    expect(result).toSucceedWith(mockOutput);\n  });\n\n  test('passes text to the upstream classifier', async () => {\n    const mockOutput: TextClassificationOutput = [{ label: 'NEGATIVE', score: 0.9997 }];\n    mockClassifier.mockResolvedValueOnce(mockOutput);\n\n    await classify(mockClassifier, 'I hate bugs');\n    expect(mockClassifier).toHaveBeenCalledWith('I hate bugs', undefined);\n  });\n\n  test('passes options through to the upstream classifier', async () => {\n    const mockOutput: TextClassificationOutput = [\n      { label: 'POSITIVE', score: 0.9998 },\n      { label: 'NEGATIVE', score: 0.0002 }\n    ];\n    mockClassifier.mockResolvedValueOnce(mockOutput);\n\n    const opts = { top_k: null };\n    await classify(mockClassifier, 'Hello world', opts);\n    expect(mockClassifier).toHaveBeenCalledWith('Hello world', opts);\n  });\n\n  test('normalises flat array result correctly', async () => {\n    const mockOutput: TextClassificationOutput = [\n      { label: 'SAFE', score: 0.95 },\n      { label: 'UNSAFE', score: 0.05 }\n    ];\n    mockClassifier.mockResolvedValueOnce(mockOutput);\n\n    expect(await classify(mockClassifier, 'hello')).toSucceedWith(mockOutput);\n  });\n\n  test('normalises nested array result by flattening one level', async () => {\n    // When upstream returns array-of-arrays (e.g. batch input path leaked through),\n    // we flatten one level so consumers always receive a flat TextClassificationOutput.\n    const inner: TextClassificationOutput = [{ label: 'POSITIVE', score: 0.9 }];\n    mockClassifier.mockResolvedValueOnce([inner] as unknown as TextClassificationOutput);\n\n    expect(await classify(mockClassifier, 'hello')).toSucceedAndSatisfy((output) => {\n      expect(output).toEqual([{ label: 'POSITIVE', score: 0.9 }]);\n    });\n  });\n\n  test('returns Failure capturing upstream error on inference failure', async () => {\n    mockClassifier.mockRejectedValueOnce(new Error('Inference session error: out of memory'));\n\n    const result = await classify(mockClassifier, 'some text');\n    expect(result).toFailWith(/inference session error/i);\n  });\n\n  test('returns Failure capturing upstream tokenisation error', async () => {\n    mockClassifier.mockRejectedValueOnce(new Error('Tokenisation failed: unexpected token'));\n\n    const result = await classify(mockClassifier, 'bad input');\n    expect(result).toFailWith(/tokenisation failed/i);\n  });\n});\n\n// ─── classifyAll ──────────────────────────────────────────────────────────────\n\ndescribe('classifyAll', () => {\n  let mockClassifier: jest.MockedFunction<TextClassificationPipeline>;\n\n  beforeEach(() => {\n    jest.resetAllMocks();\n    mockClassifier = jest.fn() as unknown as jest.MockedFunction<TextClassificationPipeline>;\n  });\n\n  test('forces top_k: null on the underlying classify call', async () => {\n    const mockOutput: TextClassificationOutput = [\n      { label: 'POSITIVE', score: 0.9998 },\n      { label: 'NEGATIVE', score: 0.0002 }\n    ];\n    mockClassifier.mockResolvedValueOnce(mockOutput);\n\n    await classifyAll(mockClassifier, 'hello');\n    expect(mockClassifier).toHaveBeenCalledWith('hello', { top_k: null });\n  });\n\n  test('overrides caller-supplied top_k with null', async () => {\n    const mockOutput: TextClassificationOutput = [\n      { label: 'POSITIVE', score: 0.9998 },\n      { label: 'NEGATIVE', score: 0.0002 }\n    ];\n    mockClassifier.mockResolvedValueOnce(mockOutput);\n\n    await classifyAll(mockClassifier, 'hello', { top_k: 1 });\n    expect(mockClassifier).toHaveBeenCalledWith('hello', { top_k: null });\n  });\n\n  test('returns all labels on success', async () => {\n    const mockOutput: TextClassificationOutput = [\n      { label: 'POSITIVE', score: 0.9998 },\n      { label: 'NEGATIVE', score: 0.0002 }\n    ];\n    mockClassifier.mockResolvedValueOnce(mockOutput);\n\n    const result = await classifyAll(mockClassifier, 'I love transformers!');\n    expect(result).toSucceedWith(mockOutput);\n  });\n\n  test('propagates upstream inference failure as Failure', async () => {\n    mockClassifier.mockRejectedValueOnce(new Error('Inference session error: out of memory'));\n\n    const result = await classifyAll(mockClassifier, 'some text');\n    expect(result).toFailWith(/inference session error/i);\n  });\n});\n\n// ─── embed ────────────────────────────────────────────────────────────────────\n\ndescribe('embed', () => {\n  let mockExtractor: jest.MockedFunction<FeatureExtractionPipeline>;\n\n  beforeEach(() => {\n    jest.resetAllMocks();\n    mockExtractor = jest.fn() as unknown as jest.MockedFunction<FeatureExtractionPipeline>;\n  });\n\n  test('returns Success wrapping the upstream Tensor on success', async () => {\n    const mockTensor = {\n      type: 'float32',\n      data: new Float32Array([0.1, 0.2, 0.3]),\n      dims: [1, 3]\n    } as unknown as Tensor;\n    mockExtractor.mockResolvedValueOnce(mockTensor);\n\n    const result = await embed(mockExtractor, 'This is a test.');\n    expect(result).toSucceedWith(mockTensor);\n  });\n\n  test('passes text to the upstream extractor', async () => {\n    const mockTensor = { type: 'float32', data: new Float32Array([0.1]), dims: [1, 1] } as unknown as Tensor;\n    mockExtractor.mockResolvedValueOnce(mockTensor);\n\n    await embed(mockExtractor, 'hello world');\n    expect(mockExtractor).toHaveBeenCalledWith('hello world', undefined);\n  });\n\n  test('passes string array to the upstream extractor', async () => {\n    const mockTensor = {\n      type: 'float32',\n      data: new Float32Array([0.1, 0.2]),\n      dims: [2, 1]\n    } as unknown as Tensor;\n    mockExtractor.mockResolvedValueOnce(mockTensor);\n\n    await embed(mockExtractor, ['text one', 'text two']);\n    expect(mockExtractor).toHaveBeenCalledWith(['text one', 'text two'], undefined);\n  });\n\n  test('passes options through to the upstream extractor', async () => {\n    const mockTensor = { type: 'float32', data: new Float32Array([0.5]), dims: [1, 1] } as unknown as Tensor;\n    mockExtractor.mockResolvedValueOnce(mockTensor);\n\n    const opts = { pooling: 'mean' as const, normalize: true };\n    await embed(mockExtractor, 'some text', opts);\n    expect(mockExtractor).toHaveBeenCalledWith('some text', opts);\n  });\n\n  test('returns Failure capturing upstream inference error', async () => {\n    mockExtractor.mockRejectedValueOnce(new Error('Inference session error: out of memory'));\n\n    const result = await embed(mockExtractor, 'some text');\n    expect(result).toFailWith(/inference session error/i);\n  });\n\n  test('returns Failure capturing upstream tokenisation error', async () => {\n    mockExtractor.mockRejectedValueOnce(new Error('Tokenisation failed: input too long'));\n\n    const result = await embed(mockExtractor, 'bad input');\n    expect(result).toFailWith(/tokenisation failed/i);\n  });\n});\n\n// ─── summarize ──────────────────────────────────────────────────────────────────\n\ndescribe('summarize', () => {\n  let mockSummarizer: jest.MockedFunction<SummarizationPipeline>;\n\n  beforeEach(() => {\n    jest.resetAllMocks();\n    mockSummarizer = jest.fn() as unknown as jest.MockedFunction<SummarizationPipeline>;\n  });\n\n  test('returns Success wrapping SummarizationOutput on success', async () => {\n    const mockOutput: SummarizationOutput = [{ summary_text: 'A short summary.' }];\n    mockSummarizer.mockResolvedValueOnce(mockOutput);\n\n    const result = await summarize(mockSummarizer, 'A long article that needs summarizing.');\n    expect(result).toSucceedWith(mockOutput);\n  });\n\n  test('passes text to the upstream summarizer', async () => {\n    const mockOutput: SummarizationOutput = [{ summary_text: 'Summary.' }];\n    mockSummarizer.mockResolvedValueOnce(mockOutput);\n\n    await summarize(mockSummarizer, 'some long document');\n    expect(mockSummarizer).toHaveBeenCalledWith('some long document', undefined);\n  });\n\n  test('passes options through to the upstream summarizer', async () => {\n    const mockOutput: SummarizationOutput = [{ summary_text: 'Bounded summary.' }];\n    mockSummarizer.mockResolvedValueOnce(mockOutput);\n\n    const opts = { min_length: 10, max_length: 50 };\n    await summarize(mockSummarizer, 'a document', opts);\n    expect(mockSummarizer).toHaveBeenCalledWith('a document', opts);\n  });\n\n  test('returns Failure capturing upstream inference error', async () => {\n    mockSummarizer.mockRejectedValueOnce(new Error('Inference session error: out of memory'));\n\n    const result = await summarize(mockSummarizer, 'some text');\n    expect(result).toFailWith(/inference session error/i);\n  });\n\n  test('returns Failure capturing upstream tokenisation error', async () => {\n    mockSummarizer.mockRejectedValueOnce(new Error('Tokenisation failed: input too long'));\n\n    const result = await summarize(mockSummarizer, 'bad input');\n    expect(result).toFailWith(/tokenisation failed/i);\n  });\n});\n"]}

package/rush-logs/ts-extras-transformers.build.cache.log

@@ -1,3 +0,0 @@
-Caching build output folders: dist, lib, temp, .rush/temp/operation/build
-Successfully set cache entry.
-Cache key: 07e0671e768f0e37ed8eb3497bda5f1e14278f10

package/rush-logs/ts-extras-transformers.build.log

@@ -1,9 +0,0 @@
-Invoking: heft build --clean 
- ---- build started ---- 
-[build:typescript] The TypeScript compiler version 5.9.3 is newer than the latest version that was tested with Heft (5.8); it may not work correctly.
-[build:typescript] Using TypeScript version 5.9.3
-[build:lint] Using ESLint version 9.39.5
-[build:api-extractor] Using API Extractor version 7.58.9
-[build:api-extractor] Analysis will use the bundled TypeScript version 5.9.3
- ---- build finished (7.772s) ---- 
--------------------- Finished (7.779s) --------------------

package/src/index.ts

@@ -1,212 +0,0 @@
-/**
- * `@fgv/ts-extras-transformers` — Result-integration boundary over `@huggingface/transformers`
- * (Node-side).
- *
- * A thin facade that wraps `@huggingface/transformers` calls in `Result<T>` from `@fgv/ts-utils`,
- * mirroring the discipline established by `@fgv/ts-extras-webauthn`: one-line `captureAsyncResult`
- * wrappers around upstream primitives with **no opinionated orchestration** above the boundary.
- *
- * **In scope:** `loadPipeline`, `classify`, `classifyAll`, `embed`, `summarize` — the task types
- * exercised by real consumers so far. The general `generate` primitive is explicitly deferred
- * until a concrete consumer use case surfaces (summarization is its own task-specific primitive).
- *
- * **Explicitly NOT in scope:**
- * - Pipeline cache / lifecycle management
- * - Model registry or download management
- * - GPU/CPU device selection policy
- * - Quantization selection
- * - Embedding-store integration
- * - Classifier label allowlists
- * - Request batching
- * - Pipeline dispose semantics
- *
- * For any of the above, use `@huggingface/transformers` directly (with `captureAsyncResult` for
- * your own Result wrapping).
- *
- * @packageDocumentation
- */
-
-import {
-  pipeline as _pipeline,
-  type TextClassificationPipeline,
-  type TextClassificationOutput,
-  type FeatureExtractionPipeline,
-  type SummarizationPipeline,
-  type SummarizationOutput,
-  type Tensor,
-  type AllTasks,
-  type PipelineType
-} from '@huggingface/transformers';
-import { captureAsyncResult, type Result } from '@fgv/ts-utils';
-
-export type {
-  TextClassificationPipeline,
-  TextClassificationOutput,
-  FeatureExtractionPipeline,
-  SummarizationPipeline,
-  SummarizationOutput,
-  Tensor,
-  AllTasks,
-  PipelineType
-};
-
-export type { PretrainedModelOptions } from '@huggingface/transformers';
-
-/**
- * Result-integration wrapper around `@huggingface/transformers`'s `pipeline` factory.
- * Loads a model and returns a ready-to-use pipeline object.
- *
- * The returned pipeline is the upstream `AllTasks[T]` instance — consumers retain full access
- * to the upstream API. Lifecycle management (caching, disposal, GPU/CPU selection) is the
- * consumer's responsibility.
- *
- * Returns `Promise<Result<AllTasks[T]>>`; upstream errors (network, model-not-found, ONNX
- * initialization failures) are captured as `Failure` with the original message.
- *
- * @param task - The pipeline task type (e.g. `'text-classification'`, `'feature-extraction'`).
- * @param model - The model identifier (HuggingFace Hub ID or local path). If omitted, the
- *   upstream default for the task is used.
- * @param options - Optional `PretrainedModelOptions` (device, dtype, cache_dir, etc.). Passed
- *   through verbatim to the upstream `pipeline()` call.
- *
- * @see https://huggingface.co/docs/transformers.js
- * @public
- */
-export async function loadPipeline<T extends PipelineType>(
-  task: T,
-  model?: string,
-  options?: Parameters<typeof _pipeline>[2]
-): Promise<Result<AllTasks[T]>> {
-  return captureAsyncResult(() => _pipeline(task, model, options));
-}
-
-/**
- * Normalises the upstream pipeline's output to a flat `TextClassificationOutput`.
- *
- * The upstream `TextClassificationPipeline` has an overloaded call signature: a single-string
- * input returns `TextClassificationOutput` (flat array), while a string-array input returns
- * `TextClassificationOutput[]` (array-of-arrays). Since `classify` always passes a single
- * string, the flat-array path is the live path. The nested-array branch is defensive and
- * ensures consumers always receive a flat array even if the upstream type union leaks through.
- */
-function flattenIfNeeded(
-  result: TextClassificationOutput | TextClassificationOutput[]
-): TextClassificationOutput {
-  if (Array.isArray(result) && result.length > 0 && !Array.isArray(result[0])) {
-    return result as TextClassificationOutput;
-  }
-  // Defensive: if somehow an array-of-arrays came back, flatten one level.
-  return (result as unknown as TextClassificationOutput[]).flat();
-}
-
-/**
- * Result-integration wrapper that invokes a `TextClassificationPipeline` on a single text input.
- * Returns the classification results as a flat `TextClassificationOutput` (array of
- * `{ label: string; score: number }` entries).
- *
- * Callers should retrieve the pipeline via `loadPipeline('text-classification', modelId)` (or the
- * `'sentiment-analysis'` alias). This helper always passes a single string to the upstream
- * pipeline and normalises the result to `TextClassificationOutput` so consumers don't need to
- * handle the `string | string[]` overload union.
- *
- * Returns `Promise<Result<TextClassificationOutput>>`; upstream errors (inference failures,
- * tokenisation errors) are captured as `Failure` with the original message.
- *
- * @param classifier - A `TextClassificationPipeline` obtained from `loadPipeline`.
- * @param text - The text to classify.
- * @param options - Optional upstream classification options (e.g. `{ top_k: null }` to return
- *   all labels). Passed through verbatim to the pipeline call.
- *
- * @see https://huggingface.co/docs/transformers.js
- * @public
- */
-export async function classify(
-  classifier: TextClassificationPipeline,
-  text: string,
-  options?: Parameters<TextClassificationPipeline>[1]
-): Promise<Result<TextClassificationOutput>> {
-  return captureAsyncResult(async () => {
-    const result = await classifier(text, options);
-    return flattenIfNeeded(result);
-  });
-}
-
-/**
- * Convenience wrapper over `classify` that forces `top_k: null` so the full per-label vector
- * is returned for every call. Callers no longer need to remember to pass `{ top_k: null }`.
- *
- * Any caller-supplied options are honoured except `top_k`, which is always overridden to `null`.
- *
- * Returns `Promise<Result<TextClassificationOutput>>`; upstream errors are captured as `Failure`
- * with the original message.
- *
- * @param classifier - A `TextClassificationPipeline` obtained from `loadPipeline`.
- * @param text - The text to classify.
- * @param options - Optional upstream classification options. `top_k` is always set to `null`
- *   regardless of any value supplied here.
- *
- * @see https://huggingface.co/docs/transformers.js
- * @public
- */
-export async function classifyAll(
-  classifier: TextClassificationPipeline,
-  text: string,
-  options?: Parameters<TextClassificationPipeline>[1]
-): Promise<Result<TextClassificationOutput>> {
-  return classify(classifier, text, { ...options, top_k: null });
-}
-
-/**
- * Result-integration wrapper that invokes a `FeatureExtractionPipeline` on a text input.
- * Returns the upstream `Tensor` Result-wrapped — no pooling, normalisation, or reshaping is
- * applied. Callers receive the raw output and are responsible for any downstream processing.
- *
- * Callers should retrieve the extractor via `loadPipeline('feature-extraction', modelId)`.
- *
- * Returns `Promise<Result<Tensor>>`; upstream errors (inference failures, tokenisation errors)
- * are captured as `Failure` with the original message.
- *
- * @param extractor - A `FeatureExtractionPipeline` obtained from `loadPipeline`.
- * @param text - The text (or texts) to embed.
- * @param options - Optional upstream feature-extraction options (e.g. `pooling`, `normalize`).
- *   Passed through verbatim to the pipeline call.
- *
- * @see https://huggingface.co/docs/transformers.js
- * @public
- */
-export async function embed(
-  extractor: FeatureExtractionPipeline,
-  text: string | string[],
-  options?: Parameters<FeatureExtractionPipeline['_call']>[1]
-): Promise<Result<Tensor>> {
-  return captureAsyncResult(() => extractor(text, options));
-}
-
-/**
- * Result-integration wrapper that invokes a `SummarizationPipeline` on a single text input.
- * Returns the upstream `SummarizationOutput` Result-wrapped (an array of `{ summary_text }`
- * entries; one for a single-string input).
- *
- * Callers should retrieve the summarizer via `loadPipeline('summarization', modelId)` — e.g.
- * `Xenova/distilbart-cnn-6-6` for a small, local-friendly model. This facade applies no opinionated
- * orchestration (no length policy, no local-vs-cloud routing) — `options` are passed through
- * verbatim and the caller owns any escalation decision.
- *
- * Returns `Promise<Result<SummarizationOutput>>`; upstream errors (inference failures, tokenisation
- * errors) are captured as `Failure` with the original message.
- *
- * @param summarizer - A `SummarizationPipeline` obtained from `loadPipeline`.
- * @param text - The text to summarize.
- * @param options - Optional upstream summarization options (e.g. `min_length`, `max_length`,
- *   `max_new_tokens`). Passed through verbatim to the pipeline call.
- *
- * @see https://huggingface.co/docs/transformers.js
- * @public
- */
-export async function summarize(
-  summarizer: SummarizationPipeline,
-  text: string,
-  options?: Parameters<SummarizationPipeline>[1]
-): Promise<Result<SummarizationOutput>> {
-  return captureAsyncResult(() => summarizer(text, options));
-}