kjarni-wasm 0.2.5 → 0.2.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kjarni-wasm",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "description": "Local AI inference in the browser: embeddings, classification, reranking, semantic search and chat, running in a Web Worker.",
   "type": "module",
   "main": "./dist/index.js",

package/pkg/kjarni_wasm.d.ts

@@ -77,15 +77,18 @@ export class WasmChat {
 
 /**
  * Sequence classification: sentiment, emotion, toxicity.
- *
- * New to the browser. Classification was always in the engine but never had a
- * binding here, because this crate previously carried its own encoder and nobody
- * wrote a second classifier head for it.
  */
 export class WasmClassifier {
     private constructor();
     free(): void;
     [Symbol.dispose](): void;
+    /**
+     * Classify a batch, returning the single best label for each input.
+     *
+     * One result per input, in input order, so the caller can line them up with
+     * whatever it passed in.
+     */
+    classify_batch(texts: string[]): any;
     /**
      * Classify one text, returning every label with its score, highest first.
      *
@@ -94,13 +97,6 @@ export class WasmClassifier {
      * what makes the behaviour testable on the host.
      */
     classify(text: string): any;
-    /**
-     * Classify a batch, returning the single best label for each input.
-     *
-     * One result per input, in input order, so the caller can line them up with
-     * whatever it passed in.
-     */
-    classify_batch(texts: string[]): any;
     /**
      * The labels this model predicts, in index order.
      */
@@ -171,8 +167,7 @@ export class WasmReranker {
     /**
      * Load a cross-encoder from a `.kjq` container.
      *
-     * The classifier head used to be pulled out of the weights by hand here; the
-     * engine's `CrossEncoder` loads it as part of the model.
+     * The engine's `CrossEncoder` loads the classifier head as part of the model.
      */
     static load(data: Uint8Array): WasmReranker;
     /**
@@ -255,8 +250,8 @@ export interface InitOutput {
     readonly wasmsearch_search_keywords: (a: number, b: number, c: number, d: number, e: number) => void;
     readonly wasmsearch_search_semantic: (a: number, b: number, c: number, d: number, e: number) => void;
     readonly wasmsearch_update_file: (a: number, b: number, c: number, d: number, e: number, f: number) => void;
-    readonly __wasm_bindgen_func_elem_5726: (a: number, b: number, c: number, d: number) => void;
-    readonly __wasm_bindgen_func_elem_5744: (a: number, b: number, c: number, d: number) => void;
+    readonly __wasm_bindgen_func_elem_5706: (a: number, b: number, c: number, d: number) => void;
+    readonly __wasm_bindgen_func_elem_5724: (a: number, b: number, c: number, d: number) => void;
     readonly __wbindgen_export: (a: number, b: number) => number;
     readonly __wbindgen_export2: (a: number, b: number, c: number, d: number) => number;
     readonly __wbindgen_export3: (a: number) => void;

package/pkg/kjarni_wasm.js

@@ -229,10 +229,6 @@ if (Symbol.dispose) WasmChat.prototype[Symbol.dispose] = WasmChat.prototype.free
 
 /**
  * Sequence classification: sentiment, emotion, toxicity.
- *
- * New to the browser. Classification was always in the engine but never had a
- * binding here, because this crate previously carried its own encoder and nobody
- * wrote a second classifier head for it.
  */
 export class WasmClassifier {
     static __wrap(ptr) {
@@ -252,20 +248,19 @@ export class WasmClassifier {
         wasm.__wbg_wasmclassifier_free(ptr, 0);
     }
     /**
-     * Classify one text, returning every label with its score, highest first.
+     * Classify a batch, returning the single best label for each input.
      *
-     * Serializing to a `JsValue` panics outside wasm32, so the work lives in
-     * [`WasmClassifier::classify_core`] and this is only the conversion. That is
-     * what makes the behaviour testable on the host.
-     * @param {string} text
+     * One result per input, in input order, so the caller can line them up with
+     * whatever it passed in.
+     * @param {string[]} texts
      * @returns {any}
      */
-    classify(text) {
+    classify_batch(texts) {
         try {
             const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
-            const ptr0 = passStringToWasm0(text, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+            const ptr0 = passArrayJsValueToWasm0(texts, wasm.__wbindgen_export);
             const len0 = WASM_VECTOR_LEN;
-            wasm.wasmclassifier_classify(retptr, this.__wbg_ptr, ptr0, len0);
+            wasm.wasmclassifier_classify_batch(retptr, this.__wbg_ptr, ptr0, len0);
             var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
             var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
             var r2 = getDataViewMemory0().getInt32(retptr + 4 * 2, true);
@@ -278,19 +273,20 @@ export class WasmClassifier {
         }
     }
     /**
-     * Classify a batch, returning the single best label for each input.
+     * Classify one text, returning every label with its score, highest first.
      *
-     * One result per input, in input order, so the caller can line them up with
-     * whatever it passed in.
-     * @param {string[]} texts
+     * Serializing to a `JsValue` panics outside wasm32, so the work lives in
+     * [`WasmClassifier::classify_core`] and this is only the conversion. That is
+     * what makes the behaviour testable on the host.
+     * @param {string} text
      * @returns {any}
      */
-    classify_batch(texts) {
+    classify(text) {
         try {
             const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
-            const ptr0 = passArrayJsValueToWasm0(texts, wasm.__wbindgen_export);
+            const ptr0 = passStringToWasm0(text, wasm.__wbindgen_export, wasm.__wbindgen_export2);
             const len0 = WASM_VECTOR_LEN;
-            wasm.wasmclassifier_classify_batch(retptr, this.__wbg_ptr, ptr0, len0);
+            wasm.wasmclassifier_classify(retptr, this.__wbg_ptr, ptr0, len0);
             var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
             var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
             var r2 = getDataViewMemory0().getInt32(retptr + 4 * 2, true);
@@ -673,8 +669,7 @@ export class WasmReranker {
     /**
      * Load a cross-encoder from a `.kjq` container.
      *
-     * The classifier head used to be pulled out of the weights by hand here; the
-     * engine's `CrossEncoder` loads it as part of the model.
+     * The engine's `CrossEncoder` loads the classifier head as part of the model.
      * @param {Uint8Array} data
      * @returns {WasmReranker}
      */
@@ -1114,7 +1109,7 @@ function __wbg_get_imports() {
                     const a = state0.a;
                     state0.a = 0;
                     try {
-                        return __wasm_bindgen_func_elem_5744(a, state0.b, arg0, arg1);
+                        return __wasm_bindgen_func_elem_5724(a, state0.b, arg0, arg1);
                     } finally {
                         state0.a = a;
                     }
@@ -1228,8 +1223,8 @@ function __wbg_get_imports() {
             return addHeapObject(ret);
         },
         __wbindgen_generic_0000000000000001: function(arg0, arg1) {
-            // Cast intrinsic for `Closure(Closure { owned: true, function: Function { arguments: [Externref], shim_idx: 2101, ret: Result(Unit), inner_ret: Some(Result(Unit)) }, mutable: true }) -> Externref`.
-            const ret = makeMutClosure(arg0, arg1, __wasm_bindgen_func_elem_5726);
+            // Cast intrinsic for `Closure(Closure { owned: true, function: Function { arguments: [Externref], shim_idx: 2079, ret: Result(Unit), inner_ret: Some(Result(Unit)) }, mutable: true }) -> Externref`.
+            const ret = makeMutClosure(arg0, arg1, __wasm_bindgen_func_elem_5706);
             return addHeapObject(ret);
         },
         __wbindgen_generic_0000000000000002: function(arg0) {
@@ -1266,14 +1261,14 @@ function __wbg_get_imports() {
     };
 }
 
-function __wasm_bindgen_func_elem_5744(arg0, arg1, arg2, arg3) {
-    wasm.__wasm_bindgen_func_elem_5744(arg0, arg1, addHeapObject(arg2), addHeapObject(arg3));
+function __wasm_bindgen_func_elem_5724(arg0, arg1, arg2, arg3) {
+    wasm.__wasm_bindgen_func_elem_5724(arg0, arg1, addHeapObject(arg2), addHeapObject(arg3));
 }
 
-function __wasm_bindgen_func_elem_5726(arg0, arg1, arg2) {
+function __wasm_bindgen_func_elem_5706(arg0, arg1, arg2) {
     try {
         const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
-        wasm.__wasm_bindgen_func_elem_5726(retptr, arg0, arg1, addHeapObject(arg2));
+        wasm.__wasm_bindgen_func_elem_5706(retptr, arg0, arg1, addHeapObject(arg2));
         var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
         var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
         if (r1) {

package/pkg/kjarni_wasm_bg.wasm.d.ts

@@ -47,8 +47,8 @@ export const wasmsearch_search: (a: number, b: number, c: number, d: number, e:
 export const wasmsearch_search_keywords: (a: number, b: number, c: number, d: number, e: number) => void;
 export const wasmsearch_search_semantic: (a: number, b: number, c: number, d: number, e: number) => void;
 export const wasmsearch_update_file: (a: number, b: number, c: number, d: number, e: number, f: number) => void;
-export const __wasm_bindgen_func_elem_5726: (a: number, b: number, c: number, d: number) => void;
-export const __wasm_bindgen_func_elem_5744: (a: number, b: number, c: number, d: number) => void;
+export const __wasm_bindgen_func_elem_5706: (a: number, b: number, c: number, d: number) => void;
+export const __wasm_bindgen_func_elem_5724: (a: number, b: number, c: number, d: number) => void;
 export const __wbindgen_export: (a: number, b: number) => number;
 export const __wbindgen_export2: (a: number, b: number, c: number, d: number) => number;
 export const __wbindgen_export3: (a: number) => void;

package/pkg/package.json

@@ -5,7 +5,7 @@
     "Ólafur Aron Jóhannsson <olafurjohannss@gmail.com>"
   ],
   "description": "Local AI inference in the browser: embeddings, classification, reranking, semantic search and chat, compiled to WebAssembly.",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "license": "MIT",
   "repository": {
     "type": "git",