kjarni-wasm 0.2.7 → 0.2.9

package/README.md

@@ -1,7 +1,11 @@
 # kjarni-wasm
 
-Local AI inference in the browser. Embeddings, classification, reranking, semantic
-search and chat, running in a Web Worker so the page never freezes.
+**Run transformer models inside your web page. Same output as PyTorch, no server.**
+
+Kjarni is a transformer inference library, and this is its WebAssembly build:
+embeddings, classification, reranking, semantic search and chat in the browser,
+running in a Web Worker so the page never freezes. The same engine ships for
+Python, C#, Go, C++ and Rust.
 
 ```bash
 npm i kjarni-wasm
@@ -11,7 +15,7 @@ Or without a bundler, straight from a CDN:
 
 ```html
 <script type="module">
-  import { Kjarni } from "https://cdn.jsdelivr.net/npm/kjarni-wasm@0.1.4/dist/index.js";
+  import { Kjarni } from "https://cdn.jsdelivr.net/npm/kjarni-wasm@0.2/dist/index.js";
 </script>
 ```
 
@@ -32,7 +36,7 @@ server will do.
 ```html
 <pre id="out">starting…</pre>
 <script type="module">
-  import init, { WasmModel } from "https://cdn.jsdelivr.net/npm/kjarni-wasm@0.1.4/pkg/kjarni_wasm.js";
+  import init, { WasmModel } from "https://cdn.jsdelivr.net/npm/kjarni-wasm@0.2/pkg/kjarni_wasm.js";
   await init();
 
   const url = "https://huggingface.co/olafuraron/all-MiniLM-L6-v2-q8/resolve/main/all-MiniLM-L6-v2-q8.kjq";

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "kjarni-wasm",
-  "version": "0.2.7",
-  "description": "Local AI inference in the browser: embeddings, classification, reranking, semantic search and chat, running in a Web Worker.",
+  "version": "0.2.9",
+  "description": "Run transformer models inside your web page: embeddings, classification, reranking, semantic search and chat in a Web Worker. Same output as PyTorch, no server.",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

package/pkg/kjarni_wasm.d.ts

@@ -60,7 +60,7 @@ export class WasmChat {
      * `model_id` selects the chat template, e.g. `qwen2.5-0.5b-instruct`. Without
      * one the model still generates, but its turn markers will not be applied.
      */
-    static load(data: Uint8Array, model_id?: string | null): WasmChat;
+    static load(data: Uint8Array | number[], model_id?: string | null): WasmChat;
     /**
      * Sends a message in an ongoing conversation, streaming the reply.
      *
@@ -82,6 +82,13 @@ 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.
      *
@@ -90,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.
      */
@@ -104,7 +104,7 @@ export class WasmClassifier {
     /**
      * Load a classifier from a `.kjq` container.
      */
-    static load(data: Uint8Array): WasmClassifier;
+    static load(data: Uint8Array | number[]): WasmClassifier;
     /**
      * How many labels the model predicts.
      */
@@ -116,18 +116,18 @@ export class WasmEncoder {
     free(): void;
     [Symbol.dispose](): void;
     encode_file(text: string, source_path: string): any;
-    static new(model_data: Uint8Array): WasmEncoder;
+    static new(model_data: Uint8Array | number[]): WasmEncoder;
 }
 
 export class WasmIndexBuilder {
     private constructor();
     free(): void;
     [Symbol.dispose](): void;
-    add_chunk(text: string, embedding: Float32Array, source: string, chunk_index: number): void;
+    add_chunk(text: string, embedding: Float32Array | number[], source: string, chunk_index: number): void;
     add_file(text: string, source_path: string): number;
     doc_count(): number;
     finish(): Uint8Array;
-    static new(model_data: Uint8Array): WasmIndexBuilder;
+    static new(model_data: Uint8Array | number[]): WasmIndexBuilder;
 }
 
 export class WasmModel {
@@ -145,7 +145,7 @@ export class WasmModel {
     /**
      * Load from a `.kjq` container: config, tokenizer and int8 tensors in one file.
      */
-    static from_quantized(data: Uint8Array): WasmModel;
+    static from_quantized(data: Uint8Array | number[]): WasmModel;
     /**
      * Fetch a model straight from Hugging Face. Browser-only: it uses `fetch`.
      */
@@ -153,7 +153,7 @@ export class WasmModel {
     /**
      * Load from raw safetensors plus its config and tokenizer.
      */
-    constructor(weights_data: Uint8Array, config_json: string, tokenizer_json: string);
+    constructor(weights_data: Uint8Array | number[], config_json: string, tokenizer_json: string);
 }
 
 export enum WasmModelType {
@@ -169,7 +169,7 @@ export class WasmReranker {
      *
      * The engine's `CrossEncoder` loads the classifier head as part of the model.
      */
-    static load(data: Uint8Array): WasmReranker;
+    static load(data: Uint8Array | number[]): WasmReranker;
     /**
      * Score every document against the query and return the best `limit`,
      * highest first. Scores are raw logits, so negative values are normal.
@@ -185,9 +185,9 @@ export class WasmSearch {
     private constructor();
     free(): void;
     [Symbol.dispose](): void;
-    add_chunk(text: string, embedding: Float32Array, source: string, chunk_index: number): void;
+    add_chunk(text: string, embedding: Float32Array | number[], source: string, chunk_index: number): void;
     doc_count(): number;
-    static load(model_data: Uint8Array, index_data: Uint8Array): WasmSearch;
+    static load(model_data: Uint8Array | number[], index_data: Uint8Array | number[]): WasmSearch;
     remove_file(source_path: string): number;
     save_index(): Uint8Array;
     search(query: string, limit: number): any;
@@ -217,11 +217,11 @@ export interface InitOutput {
     readonly wasmchat_context_size: (a: number) => number;
     readonly wasmchat_enablePrefixCache: (a: number, b: number, c: number) => void;
     readonly wasmchat_generate: (a: number, b: number, c: number, d: number, e: number, f: number) => void;
-    readonly wasmchat_generate_stream: (a: number, b: number, c: number, d: number, e: number, f: number, g: number) => void;
+    readonly wasmchat_generate_stream: (a: number, b: number, c: number, d: number, e: number, f: number, g: any) => void;
     readonly wasmchat_hasPrefixCache: (a: number) => number;
     readonly wasmchat_history_len: (a: number) => number;
     readonly wasmchat_load: (a: number, b: number, c: number, d: number, e: number) => void;
-    readonly wasmchat_send_stream: (a: number, b: number, c: number, d: number, e: number, f: number, g: number) => void;
+    readonly wasmchat_send_stream: (a: number, b: number, c: number, d: number, e: number, f: number, g: any) => void;
     readonly wasmclassifier_classify: (a: number, b: number, c: number, d: number) => void;
     readonly wasmclassifier_classify_batch: (a: number, b: number, c: number, d: number) => void;
     readonly wasmclassifier_labels: (a: number, b: number) => void;
@@ -236,7 +236,7 @@ export interface InitOutput {
     readonly wasmindexbuilder_new: (a: number, b: number, c: number) => void;
     readonly wasmmodel_encode: (a: number, b: number, c: number, d: number, e: number) => void;
     readonly wasmmodel_from_quantized: (a: number, b: number, c: number) => void;
-    readonly wasmmodel_from_type: (a: number) => number;
+    readonly wasmmodel_from_type: (a: number) => any;
     readonly wasmmodel_new: (a: number, b: number, c: number, d: number, e: number, f: number, g: number) => void;
     readonly wasmreranker_load: (a: number, b: number, c: number) => void;
     readonly wasmreranker_rerank: (a: number, b: number, c: number, d: number, e: number, f: number, g: number) => void;
@@ -250,14 +250,18 @@ 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_5727: (a: number, b: number, c: number, d: number) => void;
-    readonly __wasm_bindgen_func_elem_5745: (a: number, b: number, c: number, d: number) => void;
+    readonly __wasm_bindgen_func_elem_5740: (a: number, b: number, c: number, d: any) => void;
+    readonly __wasm_bindgen_func_elem_5769: (a: number, b: number, c: any, d: any) => 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;
-    readonly __wbindgen_export4: (a: number, b: number, c: number) => void;
-    readonly __wbindgen_export5: (a: number, b: number) => void;
+    readonly __wbindgen_export3: (a: number, b: number, c: number) => void;
+    readonly __wbindgen_export4: (a: number) => void;
+    readonly __wbindgen_export5: () => number;
+    readonly __wbindgen_externrefs: WebAssembly.Table;
+    readonly __wbindgen_export6: (a: number, b: number) => void;
     readonly __wbindgen_add_to_stack_pointer: (a: number) => number;
+    readonly __wbindgen_export7: (a: number) => void;
+    readonly __wbindgen_export8: (a: number, b: number) => void;
     readonly __wbindgen_start: () => void;
 }