@vivantel/virage-reranker-cross-encoder 0.1.15 → 0.1.18
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/reranker.d.ts.map +1 -1
- package/dist/reranker.js +12 -13
- package/dist/reranker.js.map +1 -1
- package/dist/reranker.test.js +21 -21
- package/dist/reranker.test.js.map +1 -1
- package/package.json +2 -2
package/dist/reranker.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
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;
|
|
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;AAKD,qBAAa,oBAAqB,YAAW,QAAQ;IACnD,QAAQ,CAAC,IAAI,mBAAmB;IAEhC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,SAAS,CAA2B;gBAEhC,OAAO,GAAE,2BAAgC;YAKvC,WAAW;IAUnB,MAAM,CACV,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,kBAAkB,EAAE,EAChC,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,kBAAkB,EAAE,CAAC;CAuBjC"}
|
package/dist/reranker.js
CHANGED
|
@@ -2,18 +2,17 @@ export class CrossEncoderReranker {
|
|
|
2
2
|
name = "cross-encoder";
|
|
3
3
|
modelId;
|
|
4
4
|
defaultTopK;
|
|
5
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
6
5
|
_pipeline = null;
|
|
7
6
|
constructor(options = {}) {
|
|
8
7
|
this.modelId = options.model ?? "Xenova/ms-marco-MiniLM-L-6-v2";
|
|
9
8
|
this.defaultTopK = options.topK ?? 5;
|
|
10
9
|
}
|
|
11
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
12
10
|
async getPipeline() {
|
|
13
11
|
if (!this._pipeline) {
|
|
14
12
|
const { pipeline } = await import("@huggingface/transformers");
|
|
15
|
-
|
|
16
|
-
|
|
13
|
+
this._pipeline = await pipeline("text-classification", this.modelId, {
|
|
14
|
+
dtype: "fp32",
|
|
15
|
+
});
|
|
17
16
|
}
|
|
18
17
|
return this._pipeline;
|
|
19
18
|
}
|
|
@@ -26,16 +25,16 @@ export class CrossEncoderReranker {
|
|
|
26
25
|
text: query,
|
|
27
26
|
text_pair: c.content,
|
|
28
27
|
}));
|
|
29
|
-
|
|
30
|
-
const outputs = await pipe(inputs);
|
|
28
|
+
const outputs = await pipe._call(inputs);
|
|
31
29
|
return candidates
|
|
32
|
-
.map((c, i) =>
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
: c.similarity,
|
|
38
|
-
|
|
30
|
+
.map((c, i) => {
|
|
31
|
+
const result = Array.isArray(outputs) ? outputs[i] : undefined;
|
|
32
|
+
const score = Array.isArray(result) ? result[0]?.score : result?.score;
|
|
33
|
+
return {
|
|
34
|
+
...c,
|
|
35
|
+
similarity: score ?? c.similarity,
|
|
36
|
+
};
|
|
37
|
+
})
|
|
39
38
|
.sort((a, b) => b.similarity - a.similarity)
|
|
40
39
|
.slice(0, k);
|
|
41
40
|
}
|
package/dist/reranker.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"reranker.js","sourceRoot":"","sources":["../src/reranker.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"reranker.js","sourceRoot":"","sources":["../src/reranker.ts"],"names":[],"mappings":"AAYA,MAAM,OAAO,oBAAoB;IACtB,IAAI,GAAG,eAAe,CAAC;IAEf,OAAO,CAAS;IAChB,WAAW,CAAS;IAC7B,SAAS,GAAsB,IAAI,CAAC;IAE5C,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;IAEO,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,IAAI,CAAC,SAAS,GAAG,MAAM,QAAQ,CAAC,qBAAqB,EAAE,IAAI,CAAC,OAAO,EAAE;gBACnE,KAAK,EAAE,MAAM;aACd,CAAC,CAAC;QACL,CAAC;QACD,OAAO,IAAI,CAAC,SAAU,CAAC;IACzB,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,MAAM,OAAO,GAAY,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAElD,OAAO,UAAU;aACd,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACZ,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAC/D,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC;YACvE,OAAO;gBACL,GAAG,CAAC;gBACJ,UAAU,EAAG,KAA4B,IAAI,CAAC,CAAC,UAAU;aAC1D,CAAC;QACJ,CAAC,CAAC;aACD,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.js
CHANGED
|
@@ -1,8 +1,8 @@
|
|
|
1
1
|
import { describe, it, expect, vi, beforeEach } from "vitest";
|
|
2
2
|
import { CrossEncoderReranker } from "./reranker.js";
|
|
3
3
|
import { createReranker } from "./index.js";
|
|
4
|
-
const
|
|
5
|
-
const mockPipelineFactory = vi.fn().mockResolvedValue(
|
|
4
|
+
const mockCall = vi.fn();
|
|
5
|
+
const mockPipelineFactory = vi.fn().mockResolvedValue({ _call: mockCall });
|
|
6
6
|
vi.mock("@huggingface/transformers", () => ({
|
|
7
7
|
pipeline: mockPipelineFactory,
|
|
8
8
|
}));
|
|
@@ -11,14 +11,14 @@ function makeCandidate(id, content, similarity = 0.5) {
|
|
|
11
11
|
}
|
|
12
12
|
describe("CrossEncoderReranker", () => {
|
|
13
13
|
beforeEach(() => {
|
|
14
|
-
|
|
14
|
+
mockCall.mockReset();
|
|
15
15
|
mockPipelineFactory.mockClear();
|
|
16
16
|
});
|
|
17
17
|
it("returns empty array immediately when no candidates given", async () => {
|
|
18
18
|
const reranker = new CrossEncoderReranker();
|
|
19
19
|
const result = await reranker.rerank("some query", []);
|
|
20
20
|
expect(result).toEqual([]);
|
|
21
|
-
expect(
|
|
21
|
+
expect(mockCall).not.toHaveBeenCalled();
|
|
22
22
|
});
|
|
23
23
|
it("reorders candidates by pipeline score descending", async () => {
|
|
24
24
|
const reranker = new CrossEncoderReranker();
|
|
@@ -27,10 +27,10 @@ describe("CrossEncoderReranker", () => {
|
|
|
27
27
|
makeCandidate("b", "high relevance", 0.3),
|
|
28
28
|
makeCandidate("c", "medium relevance", 0.5),
|
|
29
29
|
];
|
|
30
|
-
|
|
31
|
-
{ score: 0.1 },
|
|
32
|
-
{ score: 0.9 },
|
|
33
|
-
{ score: 0.5 },
|
|
30
|
+
mockCall.mockResolvedValue([
|
|
31
|
+
[{ label: "LABEL_0", score: 0.1 }],
|
|
32
|
+
[{ label: "LABEL_0", score: 0.9 }],
|
|
33
|
+
[{ label: "LABEL_0", score: 0.5 }],
|
|
34
34
|
]);
|
|
35
35
|
const result = await reranker.rerank("query", candidates);
|
|
36
36
|
expect(result.map((r) => r.id)).toEqual(["b", "c", "a"]);
|
|
@@ -38,7 +38,7 @@ describe("CrossEncoderReranker", () => {
|
|
|
38
38
|
it("replaces similarity with pipeline score", async () => {
|
|
39
39
|
const reranker = new CrossEncoderReranker();
|
|
40
40
|
const candidates = [makeCandidate("x", "some content", 0.5)];
|
|
41
|
-
|
|
41
|
+
mockCall.mockResolvedValue([[{ label: "LABEL_0", score: 0.77 }]]);
|
|
42
42
|
const result = await reranker.rerank("query", candidates);
|
|
43
43
|
expect(result[0].similarity).toBe(0.77);
|
|
44
44
|
});
|
|
@@ -49,10 +49,10 @@ describe("CrossEncoderReranker", () => {
|
|
|
49
49
|
makeCandidate("b", "b"),
|
|
50
50
|
makeCandidate("c", "c"),
|
|
51
51
|
];
|
|
52
|
-
|
|
53
|
-
{ score: 0.9 },
|
|
54
|
-
{ score: 0.8 },
|
|
55
|
-
{ score: 0.7 },
|
|
52
|
+
mockCall.mockResolvedValue([
|
|
53
|
+
[{ label: "LABEL_0", score: 0.9 }],
|
|
54
|
+
[{ label: "LABEL_0", score: 0.8 }],
|
|
55
|
+
[{ label: "LABEL_0", score: 0.7 }],
|
|
56
56
|
]);
|
|
57
57
|
const result = await reranker.rerank("query", candidates);
|
|
58
58
|
expect(result).toHaveLength(2);
|
|
@@ -64,31 +64,31 @@ describe("CrossEncoderReranker", () => {
|
|
|
64
64
|
makeCandidate("b", "b"),
|
|
65
65
|
makeCandidate("c", "c"),
|
|
66
66
|
];
|
|
67
|
-
|
|
68
|
-
{ score: 0.9 },
|
|
69
|
-
{ score: 0.8 },
|
|
70
|
-
{ score: 0.7 },
|
|
67
|
+
mockCall.mockResolvedValue([
|
|
68
|
+
[{ label: "LABEL_0", score: 0.9 }],
|
|
69
|
+
[{ label: "LABEL_0", score: 0.8 }],
|
|
70
|
+
[{ label: "LABEL_0", score: 0.7 }],
|
|
71
71
|
]);
|
|
72
72
|
const result = await reranker.rerank("query", candidates, 1);
|
|
73
73
|
expect(result).toHaveLength(1);
|
|
74
74
|
});
|
|
75
75
|
it("lazily initialises and caches the pipeline across calls", async () => {
|
|
76
76
|
const reranker = new CrossEncoderReranker();
|
|
77
|
-
|
|
77
|
+
mockCall.mockResolvedValue([[{ label: "LABEL_0", score: 0.5 }]]);
|
|
78
78
|
await reranker.rerank("q1", [makeCandidate("a", "text")]);
|
|
79
79
|
await reranker.rerank("q2", [makeCandidate("b", "text")]);
|
|
80
80
|
expect(mockPipelineFactory).toHaveBeenCalledTimes(1);
|
|
81
81
|
});
|
|
82
82
|
it("forwards custom model to the pipeline factory", async () => {
|
|
83
83
|
const reranker = new CrossEncoderReranker({ model: "custom/model-v1" });
|
|
84
|
-
|
|
84
|
+
mockCall.mockResolvedValue([[{ label: "LABEL_0", score: 0.5 }]]);
|
|
85
85
|
await reranker.rerank("q", [makeCandidate("a", "text")]);
|
|
86
|
-
expect(mockPipelineFactory).toHaveBeenCalledWith("text-
|
|
86
|
+
expect(mockPipelineFactory).toHaveBeenCalledWith("text-classification", "custom/model-v1", { dtype: "fp32" });
|
|
87
87
|
});
|
|
88
88
|
it("falls back to original similarity when pipeline output is not an array", async () => {
|
|
89
89
|
const reranker = new CrossEncoderReranker();
|
|
90
90
|
const candidates = [makeCandidate("a", "content", 0.42)];
|
|
91
|
-
|
|
91
|
+
mockCall.mockResolvedValue("unexpected non-array");
|
|
92
92
|
const result = await reranker.rerank("q", candidates);
|
|
93
93
|
expect(result[0].similarity).toBe(0.42);
|
|
94
94
|
});
|
|
@@ -1 +1 @@
|
|
|
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,
|
|
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,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACzB,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;AAE3E,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,QAAQ,CAAC,SAAS,EAAE,CAAC;QACrB,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,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC1C,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,QAAQ,CAAC,iBAAiB,CAAC;YACzB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;YAClC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;YAClC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;SACnC,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,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAElE,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,QAAQ,CAAC,iBAAiB,CAAC;YACzB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;YAClC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;YAClC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;SACnC,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,QAAQ,CAAC,iBAAiB,CAAC;YACzB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;YAClC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;YAClC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;SACnC,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,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;QAEjE,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,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;QAEjE,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,qBAAqB,EACrB,iBAAiB,EACjB,EAAE,KAAK,EAAE,MAAM,EAAE,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,QAAQ,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,CAAC;QAEnD,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
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@vivantel/virage-reranker-cross-encoder",
|
|
3
|
-
"version": "0.1.
|
|
3
|
+
"version": "0.1.18",
|
|
4
4
|
"description": "Local cross-encoder re-ranker for Virage (ONNX, no API key required)",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|
|
@@ -46,7 +46,7 @@
|
|
|
46
46
|
"@vivantel/virage-core": "*"
|
|
47
47
|
},
|
|
48
48
|
"devDependencies": {
|
|
49
|
-
"@vivantel/virage-core": "0.2.
|
|
49
|
+
"@vivantel/virage-core": "0.2.41",
|
|
50
50
|
"@types/node": "^25.9.3",
|
|
51
51
|
"typescript": "^6.0.3",
|
|
52
52
|
"vitest": "^4.1.9"
|