@vivantel/virage-reranker-llm 0.1.3

package/dist/index.d.ts

@@ -0,0 +1,5 @@
+export { LlmReranker } from "./reranker.js";
+export type { LlmRerankerOptions } 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,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,YAAY,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAExD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAGtD,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,QAAQ,CAExE"}

package/dist/index.js

@@ -0,0 +1,6 @@
+export { LlmReranker } from "./reranker.js";
+import { LlmReranker } from "./reranker.js";
+export function createReranker(config) {
+    return new LlmReranker(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,WAAW,EAAE,MAAM,eAAe,CAAC;AAI5C,OAAO,EAAE,WAAW,EAA2B,MAAM,eAAe,CAAC;AAErE,MAAM,UAAU,cAAc,CAAC,MAA+B;IAC5D,OAAO,IAAI,WAAW,CAAC,MAA4B,CAAC,CAAC;AACvD,CAAC"}

package/dist/reranker.d.ts

@@ -0,0 +1,18 @@
+import type { Reranker, VectorSearchResult } from "@vivantel/virage-core";
+export interface LlmRerankerOptions {
+    /** Anthropic model ID. Defaults to "claude-haiku-4-5". */
+    model?: string;
+    /** API key. Defaults to ANTHROPIC_API_KEY env var. */
+    apiKey?: string;
+    /** Number of results to return after re-ranking. Defaults to 5. */
+    topK?: number;
+}
+export declare class LlmReranker implements Reranker {
+    readonly name = "llm";
+    private readonly modelId;
+    private readonly defaultTopK;
+    private readonly client;
+    constructor(options?: LlmRerankerOptions);
+    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;AAG1E,MAAM,WAAW,kBAAkB;IACjC,0DAA0D;IAC1D,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,sDAAsD;IACtD,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,mEAAmE;IACnE,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAcD,qBAAa,WAAY,YAAW,QAAQ;IAC1C,QAAQ,CAAC,IAAI,SAAS;IAEtB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAY;gBAEvB,OAAO,GAAE,kBAAuB;IAMtC,MAAM,CACV,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,kBAAkB,EAAE,EAChC,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,kBAAkB,EAAE,CAAC;CA6CjC"}

package/dist/reranker.js

@@ -0,0 +1,62 @@
+import Anthropic from "@anthropic-ai/sdk";
+const RANK_PROMPT = (query, docs) => `You are a relevance ranking assistant. Given the query and a list of document excerpts numbered 0 to ${docs.length - 1}, return a JSON array of indices in order from most relevant to least relevant.
+
+Query: ${query}
+
+Documents:
+${docs.map((d, i) => `[${i}] ${d.slice(0, 300)}`).join("\n\n")}
+
+Respond with ONLY a JSON array of indices, e.g. [2, 0, 1]. No explanation.`;
+export class LlmReranker {
+    name = "llm";
+    modelId;
+    defaultTopK;
+    client;
+    constructor(options = {}) {
+        this.modelId = options.model ?? "claude-haiku-4-5";
+        this.defaultTopK = options.topK ?? 5;
+        this.client = new Anthropic({ apiKey: options.apiKey });
+    }
+    async rerank(query, candidates, topK) {
+        if (candidates.length === 0)
+            return [];
+        const k = topK ?? this.defaultTopK;
+        let indices;
+        try {
+            const message = await this.client.messages.create({
+                model: this.modelId,
+                max_tokens: 256,
+                messages: [
+                    {
+                        role: "user",
+                        content: RANK_PROMPT(query, candidates.map((c) => c.content)),
+                    },
+                ],
+            });
+            const text = message.content[0]?.type === "text" ? message.content[0].text : "[]";
+            const match = text.match(/\[[\d,\s]+\]/);
+            indices = match ? JSON.parse(match[0]) : [];
+        }
+        catch {
+            // Fall back to original order on error
+            return candidates.slice(0, k);
+        }
+        // Build reranked list: LLM-ordered first, then any remaining not mentioned
+        const seen = new Set();
+        const reranked = [];
+        for (const idx of indices) {
+            if (idx >= 0 && idx < candidates.length && !seen.has(idx)) {
+                const score = 1 - reranked.length / candidates.length;
+                reranked.push({ ...candidates[idx], similarity: score });
+                seen.add(idx);
+            }
+        }
+        for (let i = 0; i < candidates.length; i++) {
+            if (!seen.has(i)) {
+                reranked.push(candidates[i]);
+            }
+        }
+        return reranked.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":"AACA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAW1C,MAAM,WAAW,GAAG,CAClB,KAAa,EACb,IAAc,EACd,EAAE,CAAC,wGAAwG,IAAI,CAAC,MAAM,GAAG,CAAC;;SAEnH,KAAK;;;EAGZ,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;;2EAEa,CAAC;AAE5E,MAAM,OAAO,WAAW;IACb,IAAI,GAAG,KAAK,CAAC;IAEL,OAAO,CAAS;IAChB,WAAW,CAAS;IACpB,MAAM,CAAY;IAEnC,YAAY,UAA8B,EAAE;QAC1C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,KAAK,IAAI,kBAAkB,CAAC;QACnD,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1D,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,IAAI,OAAiB,CAAC;QACtB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAChD,KAAK,EAAE,IAAI,CAAC,OAAO;gBACnB,UAAU,EAAE,GAAG;gBACf,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,WAAW,CAClB,KAAK,EACL,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CACjC;qBACF;iBACF;aACF,CAAC,CAAC;YACH,MAAM,IAAI,GACR,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;YACvE,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACzC,OAAO,GAAG,KAAK,CAAC,CAAC,CAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAc,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5D,CAAC;QAAC,MAAM,CAAC;YACP,uCAAuC;YACvC,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAChC,CAAC;QAED,2EAA2E;QAC3E,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/B,MAAM,QAAQ,GAAyB,EAAE,CAAC;QAC1C,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;YAC1B,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,GAAG,UAAU,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC1D,MAAM,KAAK,GAAG,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;gBACtD,QAAQ,CAAC,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;gBACzD,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChB,CAAC;QACH,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBACjB,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC;QACD,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9B,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,127 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { LlmReranker } from "./reranker.js";
+import { createReranker } from "./index.js";
+const mockCreate = vi.fn();
+vi.mock("@anthropic-ai/sdk", () => ({
+    default: vi.fn().mockImplementation(function () {
+        return { messages: { create: mockCreate } };
+    }),
+}));
+function makeCandidate(id, content, similarity = 0.5) {
+    return { id, content, similarity, metadata: {} };
+}
+function mockLlmResponse(indices) {
+    mockCreate.mockResolvedValue({
+        content: [{ type: "text", text: JSON.stringify(indices) }],
+    });
+}
+describe("LlmReranker", () => {
+    beforeEach(() => {
+        mockCreate.mockReset();
+    });
+    it("returns empty array immediately when no candidates given", async () => {
+        const reranker = new LlmReranker();
+        const result = await reranker.rerank("query", []);
+        expect(result).toEqual([]);
+        expect(mockCreate).not.toHaveBeenCalled();
+    });
+    it("reorders candidates according to LLM-returned indices", async () => {
+        const reranker = new LlmReranker();
+        const candidates = [
+            makeCandidate("a", "least relevant"),
+            makeCandidate("b", "most relevant"),
+            makeCandidate("c", "middle"),
+        ];
+        mockLlmResponse([1, 2, 0]);
+        const result = await reranker.rerank("query", candidates);
+        expect(result.map((r) => r.id)).toEqual(["b", "c", "a"]);
+    });
+    it("assigns descending similarity scores to reranked results", async () => {
+        const reranker = new LlmReranker();
+        const candidates = [
+            makeCandidate("a", "a"),
+            makeCandidate("b", "b"),
+            makeCandidate("c", "c"),
+        ];
+        mockLlmResponse([2, 0, 1]);
+        const result = await reranker.rerank("query", candidates);
+        expect(result[0].similarity).toBeGreaterThan(result[1].similarity);
+        expect(result[1].similarity).toBeGreaterThan(result[2].similarity);
+    });
+    it("respects topK from constructor", async () => {
+        const reranker = new LlmReranker({ topK: 2 });
+        const candidates = [
+            makeCandidate("a", "a"),
+            makeCandidate("b", "b"),
+            makeCandidate("c", "c"),
+        ];
+        mockLlmResponse([0, 1, 2]);
+        const result = await reranker.rerank("query", candidates);
+        expect(result).toHaveLength(2);
+    });
+    it("respects topK override per call", async () => {
+        const reranker = new LlmReranker({ topK: 10 });
+        const candidates = [
+            makeCandidate("a", "a"),
+            makeCandidate("b", "b"),
+            makeCandidate("c", "c"),
+        ];
+        mockLlmResponse([0, 1, 2]);
+        const result = await reranker.rerank("query", candidates, 1);
+        expect(result).toHaveLength(1);
+    });
+    it("falls back to original order when API throws", async () => {
+        const reranker = new LlmReranker();
+        const candidates = [makeCandidate("a", "a"), makeCandidate("b", "b")];
+        mockCreate.mockRejectedValue(new Error("API error"));
+        const result = await reranker.rerank("query", candidates);
+        expect(result.map((r) => r.id)).toEqual(["a", "b"]);
+    });
+    it("falls back to original order when LLM returns no JSON array", async () => {
+        const reranker = new LlmReranker();
+        const candidates = [makeCandidate("a", "a"), makeCandidate("b", "b")];
+        mockCreate.mockResolvedValue({
+            content: [{ type: "text", text: "I cannot rank these." }],
+        });
+        const result = await reranker.rerank("query", candidates);
+        expect(result.map((r) => r.id)).toEqual(["a", "b"]);
+    });
+    it("ignores out-of-range indices from LLM", async () => {
+        const reranker = new LlmReranker();
+        const candidates = [makeCandidate("a", "a"), makeCandidate("b", "b")];
+        mockLlmResponse([99, 0, 1]);
+        const result = await reranker.rerank("query", candidates);
+        expect(result.map((r) => r.id)).toEqual(["a", "b"]);
+    });
+    it("ignores duplicate indices from LLM", async () => {
+        const reranker = new LlmReranker();
+        const candidates = [makeCandidate("a", "a"), makeCandidate("b", "b")];
+        mockLlmResponse([0, 0, 1]);
+        const result = await reranker.rerank("query", candidates);
+        expect(result.map((r) => r.id)).toEqual(["a", "b"]);
+        expect(result).toHaveLength(2);
+    });
+    it("appends candidates not mentioned by LLM at the end", async () => {
+        const reranker = new LlmReranker({ topK: 10 });
+        const candidates = [
+            makeCandidate("a", "a"),
+            makeCandidate("b", "b"),
+            makeCandidate("c", "c"),
+        ];
+        mockLlmResponse([2]); // only mentions index 2
+        const result = await reranker.rerank("query", candidates);
+        expect(result[0].id).toBe("c");
+        expect(result.map((r) => r.id)).toContain("a");
+        expect(result.map((r) => r.id)).toContain("b");
+    });
+    it("exposes name 'llm'", () => {
+        expect(new LlmReranker().name).toBe("llm");
+    });
+});
+describe("createReranker factory", () => {
+    it("returns an LlmReranker instance", () => {
+        const reranker = createReranker({});
+        expect(reranker).toBeInstanceOf(LlmReranker);
+    });
+});
+//# 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,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5C,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAE3B,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC;QAClC,OAAO,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,CAAC;IAC9C,CAAC,CAAC;CACH,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,SAAS,eAAe,CAAC,OAAiB;IACxC,UAAU,CAAC,iBAAiB,CAAC;QAC3B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;KAC3D,CAAC,CAAC;AACL,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,QAAQ,GAAG,IAAI,WAAW,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,QAAQ,GAAG,IAAI,WAAW,EAAE,CAAC;QACnC,MAAM,UAAU,GAAG;YACjB,aAAa,CAAC,GAAG,EAAE,gBAAgB,CAAC;YACpC,aAAa,CAAC,GAAG,EAAE,eAAe,CAAC;YACnC,aAAa,CAAC,GAAG,EAAE,QAAQ,CAAC;SAC7B,CAAC;QACF,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE3B,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,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,QAAQ,GAAG,IAAI,WAAW,EAAE,CAAC;QACnC,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,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE3B,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9C,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,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE3B,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,WAAW,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC/C,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,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE3B,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,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,QAAQ,GAAG,IAAI,WAAW,EAAE,CAAC;QACnC,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACtE,UAAU,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;QAErD,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,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,QAAQ,GAAG,IAAI,WAAW,EAAE,CAAC;QACnC,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACtE,UAAU,CAAC,iBAAiB,CAAC;YAC3B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;SAC1D,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,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,QAAQ,GAAG,IAAI,WAAW,EAAE,CAAC;QACnC,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACtE,eAAe,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE5B,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,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,EAAE,CAAC;QACnC,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACtE,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE3B,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,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC/C,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,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,wBAAwB;QAE9C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,CAAC,IAAI,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,QAAQ,GAAG,cAAc,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/package.json

@@ -0,0 +1,57 @@
+{
+  "name": "@vivantel/virage-reranker-llm",
+  "version": "0.1.3",
+  "description": "LLM-based re-ranker for Virage using the Anthropic API",
+  "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",
+    "llm",
+    "anthropic",
+    "claude"
+  ],
+  "author": "Vivantel",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/vivantel/virage",
+    "directory": "packages/virage-reranker-llm"
+  },
+  "dependencies": {
+    "@anthropic-ai/sdk": "^0.55.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"
+  }
+}