needle-rs 0.1.0

package/needle_wasm.d.ts

@@ -0,0 +1,105 @@
+/* tslint:disable */
+/* eslint-disable */
+
+/**
+ * WASM-exposed inference engine handle.
+ */
+export class NeedleWasm {
+    private constructor();
+    free(): void;
+    [Symbol.dispose](): void;
+    /**
+     * Return the contrastive embedding dimension (0 if no contrastive head loaded).
+     */
+    contrastive_dim(): number;
+    /**
+     * Encode text to a L2-normalized contrastive embedding.
+     *
+     * Returns a `Float32Array` of length `contrastive_dim()`, or `null` if the
+     * model was loaded without a contrastive head.
+     *
+     * Cosine similarity between query and tool embeddings equals the dot product
+     * (both vectors are already L2-normalized).
+     */
+    encode_contrastive(text: string): Float32Array | undefined;
+    /**
+     * Load the engine from raw bytes.
+     *
+     * `weights_bytes`: ArrayBuffer / Uint8Array containing the .safetensors file.
+     * `vocab_text`:    String content of vocab.txt (one piece per line).
+     */
+    static load(weights_bytes: Uint8Array, vocab_text: string): NeedleWasm | undefined;
+    /**
+     * Rank tool descriptions by contrastive similarity to a query.
+     *
+     * `tool_descs_json`: JSON array of description strings, e.g.
+     *   `'["Get current weather", "Search the web", "Send email"]'`
+     *
+     * Returns a JSON string `[{"index":0,"score":0.95},{"index":2,"score":0.71}]`
+     * sorted by descending score, or `"[]"` if the model has no contrastive head.
+     *
+     * Mirrors Python `retrieve_tools(query, tools, top_k)` from `run.py`.
+     */
+    retrieve_tools(query: string, tool_descs_json: string, top_k: number): string;
+    /**
+     * Run inference and return the output JSON string.
+     *
+     * Returns e.g. `[{"name":"get_weather","arguments":{"location":"Paris"}}]`.
+     */
+    run(query: string, tools_json: string): string;
+    /**
+     * Run inference on multiple examples and return a JS Array of output strings.
+     *
+     * Input: JS Array of `{query: string, tools: string}` objects.
+     * Output: JS Array of output strings, one per input.
+     */
+    run_batch(examples: Array<any>): Array<any>;
+    /**
+     * Run inference with a per-token streaming callback.
+     *
+     * `on_token(tokenId: number, piece: string)` fires for each generated token
+     * in decode order. Returns the final post-processed output string.
+     */
+    run_stream(query: string, tools_json: string, on_token: Function): string;
+}
+
+export type InitInput = RequestInfo | URL | Response | BufferSource | WebAssembly.Module;
+
+export interface InitOutput {
+    readonly memory: WebAssembly.Memory;
+    readonly __wbg_needlewasm_free: (a: number, b: number) => void;
+    readonly needlewasm_contrastive_dim: (a: number) => number;
+    readonly needlewasm_encode_contrastive: (a: number, b: number, c: number, d: number) => void;
+    readonly needlewasm_load: (a: number, b: number, c: number, d: number) => number;
+    readonly needlewasm_retrieve_tools: (a: number, b: number, c: number, d: number, e: number, f: number, g: number) => void;
+    readonly needlewasm_run: (a: number, b: number, c: number, d: number, e: number, f: number) => void;
+    readonly needlewasm_run_batch: (a: number, b: number) => number;
+    readonly needlewasm_run_stream: (a: number, b: number, c: number, d: number, e: number, f: number, g: 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;
+    readonly __wbindgen_add_to_stack_pointer: (a: number) => number;
+    readonly __wbindgen_export4: (a: number, b: number, c: number) => void;
+}
+
+export type SyncInitInput = BufferSource | WebAssembly.Module;
+
+/**
+ * Instantiates the given `module`, which can either be bytes or
+ * a precompiled `WebAssembly.Module`.
+ *
+ * @param {{ module: SyncInitInput }} module - Passing `SyncInitInput` directly is deprecated.
+ *
+ * @returns {InitOutput}
+ */
+export function initSync(module: { module: SyncInitInput } | SyncInitInput): InitOutput;
+
+/**
+ * If `module_or_path` is {RequestInfo} or {URL}, makes a request and
+ * for everything else, calls `WebAssembly.instantiate` directly.
+ *
+ * @param {{ module_or_path: InitInput | Promise<InitInput> }} module_or_path - Passing `InitInput` directly is deprecated.
+ *
+ * @returns {Promise<InitOutput>}
+ */
+export default function __wbg_init (module_or_path?: { module_or_path: InitInput | Promise<InitInput> } | InitInput | Promise<InitInput>): Promise<InitOutput>;

package/needle_wasm.js

@@ -0,0 +1,500 @@
+/* @ts-self-types="./needle_wasm.d.ts" */
+
+/**
+ * WASM-exposed inference engine handle.
+ */
+export class NeedleWasm {
+    static __wrap(ptr) {
+        const obj = Object.create(NeedleWasm.prototype);
+        obj.__wbg_ptr = ptr;
+        NeedleWasmFinalization.register(obj, obj.__wbg_ptr, obj);
+        return obj;
+    }
+    __destroy_into_raw() {
+        const ptr = this.__wbg_ptr;
+        this.__wbg_ptr = 0;
+        NeedleWasmFinalization.unregister(this);
+        return ptr;
+    }
+    free() {
+        const ptr = this.__destroy_into_raw();
+        wasm.__wbg_needlewasm_free(ptr, 0);
+    }
+    /**
+     * Return the contrastive embedding dimension (0 if no contrastive head loaded).
+     * @returns {number}
+     */
+    contrastive_dim() {
+        const ret = wasm.needlewasm_contrastive_dim(this.__wbg_ptr);
+        return ret >>> 0;
+    }
+    /**
+     * Encode text to a L2-normalized contrastive embedding.
+     *
+     * Returns a `Float32Array` of length `contrastive_dim()`, or `null` if the
+     * model was loaded without a contrastive head.
+     *
+     * Cosine similarity between query and tool embeddings equals the dot product
+     * (both vectors are already L2-normalized).
+     * @param {string} text
+     * @returns {Float32Array | undefined}
+     */
+    encode_contrastive(text) {
+        try {
+            const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
+            const ptr0 = passStringToWasm0(text, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+            const len0 = WASM_VECTOR_LEN;
+            wasm.needlewasm_encode_contrastive(retptr, this.__wbg_ptr, ptr0, len0);
+            var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
+            var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
+            let v2;
+            if (r0 !== 0) {
+                v2 = getArrayF32FromWasm0(r0, r1).slice();
+                wasm.__wbindgen_export4(r0, r1 * 4, 4);
+            }
+            return v2;
+        } finally {
+            wasm.__wbindgen_add_to_stack_pointer(16);
+        }
+    }
+    /**
+     * Load the engine from raw bytes.
+     *
+     * `weights_bytes`: ArrayBuffer / Uint8Array containing the .safetensors file.
+     * `vocab_text`:    String content of vocab.txt (one piece per line).
+     * @param {Uint8Array} weights_bytes
+     * @param {string} vocab_text
+     * @returns {NeedleWasm | undefined}
+     */
+    static load(weights_bytes, vocab_text) {
+        const ptr0 = passArray8ToWasm0(weights_bytes, wasm.__wbindgen_export);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passStringToWasm0(vocab_text, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+        const len1 = WASM_VECTOR_LEN;
+        const ret = wasm.needlewasm_load(ptr0, len0, ptr1, len1);
+        return ret === 0 ? undefined : NeedleWasm.__wrap(ret);
+    }
+    /**
+     * Rank tool descriptions by contrastive similarity to a query.
+     *
+     * `tool_descs_json`: JSON array of description strings, e.g.
+     *   `'["Get current weather", "Search the web", "Send email"]'`
+     *
+     * Returns a JSON string `[{"index":0,"score":0.95},{"index":2,"score":0.71}]`
+     * sorted by descending score, or `"[]"` if the model has no contrastive head.
+     *
+     * Mirrors Python `retrieve_tools(query, tools, top_k)` from `run.py`.
+     * @param {string} query
+     * @param {string} tool_descs_json
+     * @param {number} top_k
+     * @returns {string}
+     */
+    retrieve_tools(query, tool_descs_json, top_k) {
+        let deferred3_0;
+        let deferred3_1;
+        try {
+            const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
+            const ptr0 = passStringToWasm0(query, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+            const len0 = WASM_VECTOR_LEN;
+            const ptr1 = passStringToWasm0(tool_descs_json, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+            const len1 = WASM_VECTOR_LEN;
+            wasm.needlewasm_retrieve_tools(retptr, this.__wbg_ptr, ptr0, len0, ptr1, len1, top_k);
+            var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
+            var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
+            deferred3_0 = r0;
+            deferred3_1 = r1;
+            return getStringFromWasm0(r0, r1);
+        } finally {
+            wasm.__wbindgen_add_to_stack_pointer(16);
+            wasm.__wbindgen_export4(deferred3_0, deferred3_1, 1);
+        }
+    }
+    /**
+     * Run inference and return the output JSON string.
+     *
+     * Returns e.g. `[{"name":"get_weather","arguments":{"location":"Paris"}}]`.
+     * @param {string} query
+     * @param {string} tools_json
+     * @returns {string}
+     */
+    run(query, tools_json) {
+        let deferred3_0;
+        let deferred3_1;
+        try {
+            const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
+            const ptr0 = passStringToWasm0(query, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+            const len0 = WASM_VECTOR_LEN;
+            const ptr1 = passStringToWasm0(tools_json, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+            const len1 = WASM_VECTOR_LEN;
+            wasm.needlewasm_run(retptr, this.__wbg_ptr, ptr0, len0, ptr1, len1);
+            var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
+            var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
+            deferred3_0 = r0;
+            deferred3_1 = r1;
+            return getStringFromWasm0(r0, r1);
+        } finally {
+            wasm.__wbindgen_add_to_stack_pointer(16);
+            wasm.__wbindgen_export4(deferred3_0, deferred3_1, 1);
+        }
+    }
+    /**
+     * Run inference on multiple examples and return a JS Array of output strings.
+     *
+     * Input: JS Array of `{query: string, tools: string}` objects.
+     * Output: JS Array of output strings, one per input.
+     * @param {Array<any>} examples
+     * @returns {Array<any>}
+     */
+    run_batch(examples) {
+        try {
+            const ret = wasm.needlewasm_run_batch(this.__wbg_ptr, addBorrowedObject(examples));
+            return takeObject(ret);
+        } finally {
+            heap[stack_pointer++] = undefined;
+        }
+    }
+    /**
+     * Run inference with a per-token streaming callback.
+     *
+     * `on_token(tokenId: number, piece: string)` fires for each generated token
+     * in decode order. Returns the final post-processed output string.
+     * @param {string} query
+     * @param {string} tools_json
+     * @param {Function} on_token
+     * @returns {string}
+     */
+    run_stream(query, tools_json, on_token) {
+        let deferred3_0;
+        let deferred3_1;
+        try {
+            const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
+            const ptr0 = passStringToWasm0(query, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+            const len0 = WASM_VECTOR_LEN;
+            const ptr1 = passStringToWasm0(tools_json, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+            const len1 = WASM_VECTOR_LEN;
+            wasm.needlewasm_run_stream(retptr, this.__wbg_ptr, ptr0, len0, ptr1, len1, addBorrowedObject(on_token));
+            var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
+            var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
+            deferred3_0 = r0;
+            deferred3_1 = r1;
+            return getStringFromWasm0(r0, r1);
+        } finally {
+            wasm.__wbindgen_add_to_stack_pointer(16);
+            heap[stack_pointer++] = undefined;
+            wasm.__wbindgen_export4(deferred3_0, deferred3_1, 1);
+        }
+    }
+}
+if (Symbol.dispose) NeedleWasm.prototype[Symbol.dispose] = NeedleWasm.prototype.free;
+function __wbg_get_imports() {
+    const import0 = {
+        __proto__: null,
+        __wbg___wbindgen_string_get_d109740c0d18f4d7: function(arg0, arg1) {
+            const obj = getObject(arg1);
+            const ret = typeof(obj) === 'string' ? obj : undefined;
+            var ptr1 = isLikeNone(ret) ? 0 : passStringToWasm0(ret, wasm.__wbindgen_export, wasm.__wbindgen_export2);
+            var len1 = WASM_VECTOR_LEN;
+            getDataViewMemory0().setInt32(arg0 + 4 * 1, len1, true);
+            getDataViewMemory0().setInt32(arg0 + 4 * 0, ptr1, true);
+        },
+        __wbg___wbindgen_throw_9c31b086c2b26051: function(arg0, arg1) {
+            throw new Error(getStringFromWasm0(arg0, arg1));
+        },
+        __wbg_call_faa0a261f288f846: function() { return handleError(function (arg0, arg1, arg2, arg3) {
+            const ret = getObject(arg0).call(getObject(arg1), getObject(arg2), getObject(arg3));
+            return addHeapObject(ret);
+        }, arguments); },
+        __wbg_error_f085d7e62279b703: function(arg0) {
+            console.error(getObject(arg0));
+        },
+        __wbg_get_98fdf51d029a75eb: function(arg0, arg1) {
+            const ret = getObject(arg0)[arg1 >>> 0];
+            return addHeapObject(ret);
+        },
+        __wbg_get_dcf82ab8aad1a593: function() { return handleError(function (arg0, arg1) {
+            const ret = Reflect.get(getObject(arg0), getObject(arg1));
+            return addHeapObject(ret);
+        }, arguments); },
+        __wbg_length_2591a0f4f659a55c: function(arg0) {
+            const ret = getObject(arg0).length;
+            return ret;
+        },
+        __wbg_new_310879b66b6e95e1: function() {
+            const ret = new Array();
+            return addHeapObject(ret);
+        },
+        __wbg_push_b77c476b01548d0a: function(arg0, arg1) {
+            const ret = getObject(arg0).push(getObject(arg1));
+            return ret;
+        },
+        __wbindgen_cast_0000000000000001: function(arg0) {
+            // Cast intrinsic for `F64 -> Externref`.
+            const ret = arg0;
+            return addHeapObject(ret);
+        },
+        __wbindgen_cast_0000000000000002: function(arg0, arg1) {
+            // Cast intrinsic for `Ref(String) -> Externref`.
+            const ret = getStringFromWasm0(arg0, arg1);
+            return addHeapObject(ret);
+        },
+        __wbindgen_object_drop_ref: function(arg0) {
+            takeObject(arg0);
+        },
+    };
+    return {
+        __proto__: null,
+        "./needle_wasm_bg.js": import0,
+    };
+}
+
+const NeedleWasmFinalization = (typeof FinalizationRegistry === 'undefined')
+    ? { register: () => {}, unregister: () => {} }
+    : new FinalizationRegistry(ptr => wasm.__wbg_needlewasm_free(ptr, 1));
+
+function addHeapObject(obj) {
+    if (heap_next === heap.length) heap.push(heap.length + 1);
+    const idx = heap_next;
+    heap_next = heap[idx];
+
+    heap[idx] = obj;
+    return idx;
+}
+
+function addBorrowedObject(obj) {
+    if (stack_pointer == 1) throw new Error('out of js stack');
+    heap[--stack_pointer] = obj;
+    return stack_pointer;
+}
+
+function dropObject(idx) {
+    if (idx < 1028) return;
+    heap[idx] = heap_next;
+    heap_next = idx;
+}
+
+function getArrayF32FromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return getFloat32ArrayMemory0().subarray(ptr / 4, ptr / 4 + len);
+}
+
+let cachedDataViewMemory0 = null;
+function getDataViewMemory0() {
+    if (cachedDataViewMemory0 === null || cachedDataViewMemory0.buffer.detached === true || (cachedDataViewMemory0.buffer.detached === undefined && cachedDataViewMemory0.buffer !== wasm.memory.buffer)) {
+        cachedDataViewMemory0 = new DataView(wasm.memory.buffer);
+    }
+    return cachedDataViewMemory0;
+}
+
+let cachedFloat32ArrayMemory0 = null;
+function getFloat32ArrayMemory0() {
+    if (cachedFloat32ArrayMemory0 === null || cachedFloat32ArrayMemory0.byteLength === 0) {
+        cachedFloat32ArrayMemory0 = new Float32Array(wasm.memory.buffer);
+    }
+    return cachedFloat32ArrayMemory0;
+}
+
+function getStringFromWasm0(ptr, len) {
+    return decodeText(ptr >>> 0, len);
+}
+
+let cachedUint8ArrayMemory0 = null;
+function getUint8ArrayMemory0() {
+    if (cachedUint8ArrayMemory0 === null || cachedUint8ArrayMemory0.byteLength === 0) {
+        cachedUint8ArrayMemory0 = new Uint8Array(wasm.memory.buffer);
+    }
+    return cachedUint8ArrayMemory0;
+}
+
+function getObject(idx) { return heap[idx]; }
+
+function handleError(f, args) {
+    try {
+        return f.apply(this, args);
+    } catch (e) {
+        wasm.__wbindgen_export3(addHeapObject(e));
+    }
+}
+
+let heap = new Array(1024).fill(undefined);
+heap.push(undefined, null, true, false);
+
+let heap_next = heap.length;
+
+function isLikeNone(x) {
+    return x === undefined || x === null;
+}
+
+function passArray8ToWasm0(arg, malloc) {
+    const ptr = malloc(arg.length * 1, 1) >>> 0;
+    getUint8ArrayMemory0().set(arg, ptr / 1);
+    WASM_VECTOR_LEN = arg.length;
+    return ptr;
+}
+
+function passStringToWasm0(arg, malloc, realloc) {
+    if (realloc === undefined) {
+        const buf = cachedTextEncoder.encode(arg);
+        const ptr = malloc(buf.length, 1) >>> 0;
+        getUint8ArrayMemory0().subarray(ptr, ptr + buf.length).set(buf);
+        WASM_VECTOR_LEN = buf.length;
+        return ptr;
+    }
+
+    let len = arg.length;
+    let ptr = malloc(len, 1) >>> 0;
+
+    const mem = getUint8ArrayMemory0();
+
+    let offset = 0;
+
+    for (; offset < len; offset++) {
+        const code = arg.charCodeAt(offset);
+        if (code > 0x7F) break;
+        mem[ptr + offset] = code;
+    }
+    if (offset !== len) {
+        if (offset !== 0) {
+            arg = arg.slice(offset);
+        }
+        ptr = realloc(ptr, len, len = offset + arg.length * 3, 1) >>> 0;
+        const view = getUint8ArrayMemory0().subarray(ptr + offset, ptr + len);
+        const ret = cachedTextEncoder.encodeInto(arg, view);
+
+        offset += ret.written;
+        ptr = realloc(ptr, len, offset, 1) >>> 0;
+    }
+
+    WASM_VECTOR_LEN = offset;
+    return ptr;
+}
+
+let stack_pointer = 1024;
+
+function takeObject(idx) {
+    const ret = getObject(idx);
+    dropObject(idx);
+    return ret;
+}
+
+let cachedTextDecoder = new TextDecoder('utf-8', { ignoreBOM: true, fatal: true });
+cachedTextDecoder.decode();
+const MAX_SAFARI_DECODE_BYTES = 2146435072;
+let numBytesDecoded = 0;
+function decodeText(ptr, len) {
+    numBytesDecoded += len;
+    if (numBytesDecoded >= MAX_SAFARI_DECODE_BYTES) {
+        cachedTextDecoder = new TextDecoder('utf-8', { ignoreBOM: true, fatal: true });
+        cachedTextDecoder.decode();
+        numBytesDecoded = len;
+    }
+    return cachedTextDecoder.decode(getUint8ArrayMemory0().subarray(ptr, ptr + len));
+}
+
+const cachedTextEncoder = new TextEncoder();
+
+if (!('encodeInto' in cachedTextEncoder)) {
+    cachedTextEncoder.encodeInto = function (arg, view) {
+        const buf = cachedTextEncoder.encode(arg);
+        view.set(buf);
+        return {
+            read: arg.length,
+            written: buf.length
+        };
+    };
+}
+
+let WASM_VECTOR_LEN = 0;
+
+let wasmModule, wasmInstance, wasm;
+function __wbg_finalize_init(instance, module) {
+    wasmInstance = instance;
+    wasm = instance.exports;
+    wasmModule = module;
+    cachedDataViewMemory0 = null;
+    cachedFloat32ArrayMemory0 = null;
+    cachedUint8ArrayMemory0 = null;
+    return wasm;
+}
+
+async function __wbg_load(module, imports) {
+    if (typeof Response === 'function' && module instanceof Response) {
+        if (typeof WebAssembly.instantiateStreaming === 'function') {
+            try {
+                return await WebAssembly.instantiateStreaming(module, imports);
+            } catch (e) {
+                const validResponse = module.ok && expectedResponseType(module.type);
+
+                if (validResponse && module.headers.get('Content-Type') !== 'application/wasm') {
+                    console.warn("`WebAssembly.instantiateStreaming` failed because your server does not serve Wasm with `application/wasm` MIME type. Falling back to `WebAssembly.instantiate` which is slower. Original error:\n", e);
+
+                } else { throw e; }
+            }
+        }
+
+        const bytes = await module.arrayBuffer();
+        return await WebAssembly.instantiate(bytes, imports);
+    } else {
+        const instance = await WebAssembly.instantiate(module, imports);
+
+        if (instance instanceof WebAssembly.Instance) {
+            return { instance, module };
+        } else {
+            return instance;
+        }
+    }
+
+    function expectedResponseType(type) {
+        switch (type) {
+            case 'basic': case 'cors': case 'default': return true;
+        }
+        return false;
+    }
+}
+
+function initSync(module) {
+    if (wasm !== undefined) return wasm;
+
+
+    if (module !== undefined) {
+        if (Object.getPrototypeOf(module) === Object.prototype) {
+            ({module} = module)
+        } else {
+            console.warn('using deprecated parameters for `initSync()`; pass a single object instead')
+        }
+    }
+
+    const imports = __wbg_get_imports();
+    if (!(module instanceof WebAssembly.Module)) {
+        module = new WebAssembly.Module(module);
+    }
+    const instance = new WebAssembly.Instance(module, imports);
+    return __wbg_finalize_init(instance, module);
+}
+
+async function __wbg_init(module_or_path) {
+    if (wasm !== undefined) return wasm;
+
+
+    if (module_or_path !== undefined) {
+        if (Object.getPrototypeOf(module_or_path) === Object.prototype) {
+            ({module_or_path} = module_or_path)
+        } else {
+            console.warn('using deprecated parameters for the initialization function; pass a single object instead')
+        }
+    }
+
+    if (module_or_path === undefined) {
+        module_or_path = new URL('needle_wasm_bg.wasm', import.meta.url);
+    }
+    const imports = __wbg_get_imports();
+
+    if (typeof module_or_path === 'string' || (typeof Request === 'function' && module_or_path instanceof Request) || (typeof URL === 'function' && module_or_path instanceof URL)) {
+        module_or_path = fetch(module_or_path);
+    }
+
+    const { instance, module } = await __wbg_load(await module_or_path, imports);
+
+    return __wbg_finalize_init(instance, module);
+}
+
+export { initSync, __wbg_init as default };

package/needle_wasm_bg.wasm.d.ts

@@ -0,0 +1,16 @@
+/* tslint:disable */
+/* eslint-disable */
+export const memory: WebAssembly.Memory;
+export const __wbg_needlewasm_free: (a: number, b: number) => void;
+export const needlewasm_contrastive_dim: (a: number) => number;
+export const needlewasm_encode_contrastive: (a: number, b: number, c: number, d: number) => void;
+export const needlewasm_load: (a: number, b: number, c: number, d: number) => number;
+export const needlewasm_retrieve_tools: (a: number, b: number, c: number, d: number, e: number, f: number, g: number) => void;
+export const needlewasm_run: (a: number, b: number, c: number, d: number, e: number, f: number) => void;
+export const needlewasm_run_batch: (a: number, b: number) => number;
+export const needlewasm_run_stream: (a: number, b: number, c: number, d: number, e: number, f: number, g: 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;
+export const __wbindgen_add_to_stack_pointer: (a: number) => number;
+export const __wbindgen_export4: (a: number, b: number, c: number) => void;

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "needle-rs",
+  "version": "0.1.0",
+  "description": "258 KB WASM runtime for Needle \u2014 a 26M-parameter tool-calling transformer. Runs in the browser, Cloudflare Workers, and Node.js. No backend required.",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/geekgineer/needle-rs"
+  },
+  "homepage": "https://github.com/geekgineer/needle-rs",
+  "keywords": [
+    "wasm",
+    "webassembly",
+    "ai",
+    "llm",
+    "tool-calling",
+    "function-calling",
+    "transformer",
+    "rust",
+    "no-backend",
+    "browser-ai"
+  ],
+  "files": [
+    "needle_wasm_bg.wasm",
+    "needle_wasm_bg.wasm.d.ts",
+    "needle_wasm.js",
+    "needle_wasm.d.ts"
+  ],
+  "module": "needle_wasm.js",
+  "types": "needle_wasm.d.ts",
+  "exports": {
+    ".": {
+      "import": "./needle_wasm.js",
+      "types": "./needle_wasm.d.ts"
+    }
+  },
+  "sideEffects": false
+}