@vivantel/virage-reranker-cross-encoder 0.1.52 → 0.2.1

package/dist/index.d.ts

@@ -1,5 +1,23 @@
 export { CrossEncoderReranker } from "./reranker.js";
 export type { CrossEncoderRerankerOptions } from "./reranker.js";
+import { z } from "zod";
 import type { Reranker } from "@vivantel/virage-core";
+export declare const optionsSchema: z.ZodObject<{
+    model: z.ZodOptional<z.ZodString>;
+    topK: z.ZodOptional<z.ZodNumber>;
+    cacheDir: z.ZodOptional<z.ZodString>;
+    device: z.ZodDefault<z.ZodOptional<z.ZodEnum<["cpu", "cuda", "webgpu"]>>>;
+}, "strip", z.ZodTypeAny, {
+    device: "cpu" | "cuda" | "webgpu";
+    model?: string | undefined;
+    topK?: number | undefined;
+    cacheDir?: string | undefined;
+}, {
+    model?: string | undefined;
+    topK?: number | undefined;
+    cacheDir?: string | undefined;
+    device?: "cpu" | "cuda" | "webgpu" | undefined;
+}>;
+export type PluginOptions = z.infer<typeof optionsSchema>;
 export declare function createReranker(config: Record<string, unknown>): Reranker;
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +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"}
+{"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,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAMtD,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;EAUxB,CAAC;AACH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAE1D,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,QAAQ,CAExE"}

package/dist/index.js

@@ -1,5 +1,17 @@
 export { CrossEncoderReranker } from "./reranker.js";
+import { z } from "zod";
 import { CrossEncoderReranker, } from "./reranker.js";
+export const optionsSchema = z.object({
+    model: z.string().optional().describe("Cross-encoder model name"),
+    topK: z
+        .number()
+        .int()
+        .positive()
+        .optional()
+        .describe("Max results to return after reranking"),
+    cacheDir: z.string().optional(),
+    device: z.enum(["cpu", "cuda", "webgpu"]).optional().default("cpu"),
+});
 export function createReranker(config) {
     return new CrossEncoderReranker(config);
 }

package/dist/index.js.map

@@ -1 +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"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAGrD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EACL,oBAAoB,GAErB,MAAM,eAAe,CAAC;AAEvB,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;IACjE,IAAI,EAAE,CAAC;SACJ,MAAM,EAAE;SACR,GAAG,EAAE;SACL,QAAQ,EAAE;SACV,QAAQ,EAAE;SACV,QAAQ,CAAC,uCAAuC,CAAC;IACpD,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC;CACpE,CAAC,CAAC;AAGH,MAAM,UAAU,cAAc,CAAC,MAA+B;IAC5D,OAAO,IAAI,oBAAoB,CAAC,MAAqC,CAAC,CAAC;AACzE,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vivantel/virage-reranker-cross-encoder",
-  "version": "0.1.52",
+  "version": "0.2.1",
   "description": "Local cross-encoder re-ranker for Virage (ONNX, no API key required)",
   "type": "module",
   "main": "dist/index.js",
@@ -40,13 +40,14 @@
     "directory": "packages/virage-reranker-cross-encoder"
   },
   "dependencies": {
-    "@huggingface/transformers": "^4.0.0"
+    "@huggingface/transformers": "^4.0.0",
+    "zod": "^3.24.0"
   },
   "peerDependencies": {
     "@vivantel/virage-core": "*"
   },
   "devDependencies": {
-    "@vivantel/virage-core": "0.3.10",
+    "@vivantel/virage-core": "0.4.1",
     "@types/node": "^26.0.1",
     "typescript": "^6.0.3",
     "vitest": "^4.1.9"