@vivantel/virage-reranker-cross-encoder 0.1.3

package/dist/index.d.ts

@@ -0,0 +1,5 @@
+export { CrossEncoderReranker } from "./reranker.js";
+export type { CrossEncoderRerankerOptions } from "./reranker.js";
+import type { Reranker } from "@vivantel/virage-core";
+export declare function createReranker(config: Record<string, unknown>): Reranker;
+//# 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,OAAO,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AACrD,YAAY,EAAE,2BAA2B,EAAE,MAAM,eAAe,CAAC;AAEjE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAMtD,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,QAAQ,CAExE"}

package/dist/index.js

@@ -0,0 +1,6 @@
+export { CrossEncoderReranker } from "./reranker.js";
+import { CrossEncoderReranker, } from "./reranker.js";
+export function createReranker(config) {
+    return new CrossEncoderReranker(config);
+}
+//# 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,OAAO,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAIrD,OAAO,EACL,oBAAoB,GAErB,MAAM,eAAe,CAAC;AAEvB,MAAM,UAAU,cAAc,CAAC,MAA+B;IAC5D,OAAO,IAAI,oBAAoB,CAAC,MAAqC,CAAC,CAAC;AACzE,CAAC"}

package/dist/reranker.d.ts

@@ -0,0 +1,17 @@
+import type { Reranker, VectorSearchResult } from "@vivantel/virage-core";
+export interface CrossEncoderRerankerOptions {
+    /** HuggingFace model ID. Defaults to "Xenova/ms-marco-MiniLM-L-6-v2". */
+    model?: string;
+    /** Number of results to return after re-ranking. Defaults to 5. */
+    topK?: number;
+}
+export declare class CrossEncoderReranker implements Reranker {
+    readonly name = "cross-encoder";
+    private readonly modelId;
+    private readonly defaultTopK;
+    private _pipeline;
+    constructor(options?: CrossEncoderRerankerOptions);
+    private getPipeline;
+    rerank(query: string, candidates: VectorSearchResult[], topK?: number): Promise<VectorSearchResult[]>;
+}
+//# sourceMappingURL=reranker.d.ts.map

package/dist/reranker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.d.ts","sourceRoot":"","sources":["../src/reranker.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE1E,MAAM,WAAW,2BAA2B;IAC1C,yEAAyE;IACzE,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,mEAAmE;IACnE,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,qBAAa,oBAAqB,YAAW,QAAQ;IACnD,QAAQ,CAAC,IAAI,mBAAmB;IAEhC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IAErC,OAAO,CAAC,SAAS,CAAa;gBAElB,OAAO,GAAE,2BAAgC;YAMvC,WAAW;IASnB,MAAM,CACV,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,kBAAkB,EAAE,EAChC,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,kBAAkB,EAAE,CAAC;CAsBjC"}

package/dist/reranker.js

@@ -0,0 +1,42 @@
+export class CrossEncoderReranker {
+    name = "cross-encoder";
+    modelId;
+    defaultTopK;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _pipeline = null;
+    constructor(options = {}) {
+        this.modelId = options.model ?? "Xenova/ms-marco-MiniLM-L-6-v2";
+        this.defaultTopK = options.topK ?? 5;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    async getPipeline() {
+        if (!this._pipeline) {
+            const { pipeline } = await import("@huggingface/transformers");
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            this._pipeline = await pipeline("text-ranking", this.modelId);
+        }
+        return this._pipeline;
+    }
+    async rerank(query, candidates, topK) {
+        if (candidates.length === 0)
+            return [];
+        const k = topK ?? this.defaultTopK;
+        const pipe = await this.getPipeline();
+        const inputs = candidates.map((c) => ({
+            text: query,
+            text_pair: c.content,
+        }));
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const outputs = await pipe(inputs, { top_k: null });
+        return candidates
+            .map((c, i) => ({
+            ...c,
+            similarity: Array.isArray(outputs)
+                ? (outputs[i]?.score ?? c.similarity)
+                : c.similarity,
+        }))
+            .sort((a, b) => b.similarity - a.similarity)
+            .slice(0, k);
+    }
+}
+//# sourceMappingURL=reranker.js.map

package/dist/reranker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.js","sourceRoot":"","sources":["../src/reranker.ts"],"names":[],"mappings":"AASA,MAAM,OAAO,oBAAoB;IACtB,IAAI,GAAG,eAAe,CAAC;IAEf,OAAO,CAAS;IAChB,WAAW,CAAS;IACrC,8DAA8D;IACtD,SAAS,GAAQ,IAAI,CAAC;IAE9B,YAAY,UAAuC,EAAE;QACnD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,KAAK,IAAI,+BAA+B,CAAC;QAChE,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC;IACvC,CAAC;IAED,8DAA8D;IACtD,KAAK,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;YAC/D,8DAA8D;YAC9D,IAAI,CAAC,SAAS,GAAG,MAAM,QAAQ,CAAC,cAAqB,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACvE,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,MAAM,CACV,KAAa,EACb,UAAgC,EAChC,IAAa;QAEb,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACvC,MAAM,CAAC,GAAG,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC;QAEnC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACpC,IAAI,EAAE,KAAK;YACX,SAAS,EAAE,CAAC,CAAC,OAAO;SACrB,CAAC,CAAC,CAAC;QACJ,8DAA8D;QAC9D,MAAM,OAAO,GAAU,MAAM,IAAI,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAE3D,OAAO,UAAU;aACd,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACd,GAAG,CAAC;YACJ,UAAU,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;gBAChC,CAAC,CAAC,CAAE,OAAO,CAAC,CAAC,CAAC,EAAE,KAA4B,IAAI,CAAC,CAAC,UAAU,CAAC;gBAC7D,CAAC,CAAC,CAAC,CAAC,UAAU;SACjB,CAAC,CAAC;aACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC;aAC3C,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACjB,CAAC;CACF"}

package/dist/reranker.test.d.ts

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

package/dist/reranker.test.d.ts.map

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

package/dist/reranker.test.js

@@ -0,0 +1,109 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { CrossEncoderReranker } from "./reranker.js";
+import { createReranker } from "./index.js";
+const mockPipelineFn = vi.fn();
+const mockPipelineFactory = vi.fn().mockResolvedValue(mockPipelineFn);
+vi.mock("@huggingface/transformers", () => ({
+    pipeline: mockPipelineFactory,
+}));
+function makeCandidate(id, content, similarity = 0.5) {
+    return { id, content, similarity, metadata: {} };
+}
+describe("CrossEncoderReranker", () => {
+    beforeEach(() => {
+        mockPipelineFn.mockReset();
+        mockPipelineFactory.mockClear();
+    });
+    it("returns empty array immediately when no candidates given", async () => {
+        const reranker = new CrossEncoderReranker();
+        const result = await reranker.rerank("some query", []);
+        expect(result).toEqual([]);
+        expect(mockPipelineFn).not.toHaveBeenCalled();
+    });
+    it("reorders candidates by pipeline score descending", async () => {
+        const reranker = new CrossEncoderReranker();
+        const candidates = [
+            makeCandidate("a", "low relevance", 0.8),
+            makeCandidate("b", "high relevance", 0.3),
+            makeCandidate("c", "medium relevance", 0.5),
+        ];
+        mockPipelineFn.mockResolvedValue([
+            { score: 0.1 },
+            { score: 0.9 },
+            { score: 0.5 },
+        ]);
+        const result = await reranker.rerank("query", candidates);
+        expect(result.map((r) => r.id)).toEqual(["b", "c", "a"]);
+    });
+    it("replaces similarity with pipeline score", async () => {
+        const reranker = new CrossEncoderReranker();
+        const candidates = [makeCandidate("x", "some content", 0.5)];
+        mockPipelineFn.mockResolvedValue([{ score: 0.77 }]);
+        const result = await reranker.rerank("query", candidates);
+        expect(result[0].similarity).toBe(0.77);
+    });
+    it("respects topK from constructor", async () => {
+        const reranker = new CrossEncoderReranker({ topK: 2 });
+        const candidates = [
+            makeCandidate("a", "a"),
+            makeCandidate("b", "b"),
+            makeCandidate("c", "c"),
+        ];
+        mockPipelineFn.mockResolvedValue([
+            { score: 0.9 },
+            { score: 0.8 },
+            { score: 0.7 },
+        ]);
+        const result = await reranker.rerank("query", candidates);
+        expect(result).toHaveLength(2);
+    });
+    it("respects topK override per call", async () => {
+        const reranker = new CrossEncoderReranker({ topK: 10 });
+        const candidates = [
+            makeCandidate("a", "a"),
+            makeCandidate("b", "b"),
+            makeCandidate("c", "c"),
+        ];
+        mockPipelineFn.mockResolvedValue([
+            { score: 0.9 },
+            { score: 0.8 },
+            { score: 0.7 },
+        ]);
+        const result = await reranker.rerank("query", candidates, 1);
+        expect(result).toHaveLength(1);
+    });
+    it("lazily initialises and caches the pipeline across calls", async () => {
+        const reranker = new CrossEncoderReranker();
+        mockPipelineFn.mockResolvedValue([{ score: 0.5 }]);
+        await reranker.rerank("q1", [makeCandidate("a", "text")]);
+        await reranker.rerank("q2", [makeCandidate("b", "text")]);
+        expect(mockPipelineFactory).toHaveBeenCalledTimes(1);
+    });
+    it("forwards custom model to the pipeline factory", async () => {
+        const reranker = new CrossEncoderReranker({ model: "custom/model-v1" });
+        mockPipelineFn.mockResolvedValue([{ score: 0.5 }]);
+        await reranker.rerank("q", [makeCandidate("a", "text")]);
+        expect(mockPipelineFactory).toHaveBeenCalledWith("text-ranking", "custom/model-v1");
+    });
+    it("falls back to original similarity when pipeline output is not an array", async () => {
+        const reranker = new CrossEncoderReranker();
+        const candidates = [makeCandidate("a", "content", 0.42)];
+        mockPipelineFn.mockResolvedValue("unexpected non-array");
+        const result = await reranker.rerank("q", candidates);
+        expect(result[0].similarity).toBe(0.42);
+    });
+    it("exposes name 'cross-encoder'", () => {
+        expect(new CrossEncoderReranker().name).toBe("cross-encoder");
+    });
+});
+describe("createReranker factory", () => {
+    it("returns a CrossEncoderReranker", () => {
+        const reranker = createReranker({});
+        expect(reranker).toBeInstanceOf(CrossEncoderReranker);
+    });
+    it("forwards config options", () => {
+        const reranker = createReranker({ topK: 3 });
+        expect(reranker.name).toBe("cross-encoder");
+    });
+});
+//# sourceMappingURL=reranker.test.js.map

package/dist/reranker.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.test.js","sourceRoot":"","sources":["../src/reranker.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAE9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5C,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAC/B,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;AAEtE,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,QAAQ,EAAE,mBAAmB;CAC9B,CAAC,CAAC,CAAC;AAEJ,SAAS,aAAa,CACpB,EAAU,EACV,OAAe,EACf,UAAU,GAAG,GAAG;IAEhB,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;AACnD,CAAC;AAED,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,UAAU,CAAC,GAAG,EAAE;QACd,cAAc,CAAC,SAAS,EAAE,CAAC;QAC3B,mBAAmB,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,QAAQ,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC5C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,QAAQ,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC5C,MAAM,UAAU,GAAG;YACjB,aAAa,CAAC,GAAG,EAAE,eAAe,EAAE,GAAG,CAAC;YACxC,aAAa,CAAC,GAAG,EAAE,gBAAgB,EAAE,GAAG,CAAC;YACzC,aAAa,CAAC,GAAG,EAAE,kBAAkB,EAAE,GAAG,CAAC;SAC5C,CAAC;QACF,cAAc,CAAC,iBAAiB,CAAC;YAC/B,EAAE,KAAK,EAAE,GAAG,EAAE;YACd,EAAE,KAAK,EAAE,GAAG,EAAE;YACd,EAAE,KAAK,EAAE,GAAG,EAAE;SACf,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,QAAQ,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC5C,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,cAAc,EAAE,GAAG,CAAC,CAAC,CAAC;QAC7D,cAAc,CAAC,iBAAiB,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAEpD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,QAAQ,GAAG,IAAI,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QACvD,MAAM,UAAU,GAAG;YACjB,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC;YACvB,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC;YACvB,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC;SACxB,CAAC;QACF,cAAc,CAAC,iBAAiB,CAAC;YAC/B,EAAE,KAAK,EAAE,GAAG,EAAE;YACd,EAAE,KAAK,EAAE,GAAG,EAAE;YACd,EAAE,KAAK,EAAE,GAAG,EAAE;SACf,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,QAAQ,GAAG,IAAI,oBAAoB,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QACxD,MAAM,UAAU,GAAG;YACjB,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC;YACvB,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC;YACvB,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC;SACxB,CAAC;QACF,cAAc,CAAC,iBAAiB,CAAC;YAC/B,EAAE,KAAK,EAAE,GAAG,EAAE;YACd,EAAE,KAAK,EAAE,GAAG,EAAE;YACd,EAAE,KAAK,EAAE,GAAG,EAAE;SACf,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,QAAQ,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC5C,cAAc,CAAC,iBAAiB,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEnD,MAAM,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,QAAQ,GAAG,IAAI,oBAAoB,CAAC,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC,CAAC;QACxE,cAAc,CAAC,iBAAiB,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEnD,MAAM,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C,cAAc,EACd,iBAAiB,CAClB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC5C,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;QACzD,cAAc,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,CAAC;QAEzD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,CAAC,IAAI,oBAAoB,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,QAAQ,GAAG,cAAc,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,QAAQ,GAAG,cAAc,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAyB,CAAC;QACrE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/package.json

@@ -0,0 +1,57 @@
+{
+  "name": "@vivantel/virage-reranker-cross-encoder",
+  "version": "0.1.3",
+  "description": "Local cross-encoder re-ranker for Virage (ONNX, no API key required)",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "sideEffects": false,
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc",
+    "type-check": "tsc --noEmit",
+    "test": "vitest run",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "rag",
+    "reranker",
+    "cross-encoder",
+    "onnx",
+    "local"
+  ],
+  "author": "Vivantel",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/vivantel/virage",
+    "directory": "packages/virage-reranker-cross-encoder"
+  },
+  "dependencies": {
+    "@huggingface/transformers": "^3.0.0"
+  },
+  "peerDependencies": {
+    "@vivantel/virage-core": "*"
+  },
+  "devDependencies": {
+    "@vivantel/virage-core": "0.2.29",
+    "@types/node": "^25.9.3",
+    "typescript": "^6.0.3",
+    "vitest": "^4.1.9"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}