@taladb/web 0.1.2 → 0.2.0

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@taladb/web",
-  "version": "0.1.2",
-  "description": "TalaDB WASM bindings for browser (wasm-bindgen + OPFS SharedWorker)",
+  "version": "0.2.0",
+  "description": "TalaDB WASM bindings — document queries and vector search in the browser",
   "main": "pkg/taladb_web.js",
   "types": "pkg/taladb_web.d.ts",
   "exports": {

package/pkg/taladb_web.js

@@ -42,6 +42,25 @@ export class CollectionWasm {
             throw takeFromExternrefTable0(ret[0]);
         }
     }
+    /**
+     * Create a vector index on `field`.
+     *
+     * `dimensions` — expected vector length.
+     * `metric`     — optional string: `"cosine"` (default), `"dot"`, or `"euclidean"`.
+     * @param {string} field
+     * @param {number} dimensions
+     * @param {string | null} [metric]
+     */
+    createVectorIndex(field, dimensions, metric) {
+        const ptr0 = passStringToWasm0(field, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        var ptr1 = isLikeNone(metric) ? 0 : passStringToWasm0(metric, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        var len1 = WASM_VECTOR_LEN;
+        const ret = wasm.collectionwasm_createVectorIndex(this.__wbg_ptr, ptr0, len0, dimensions, ptr1, len1);
+        if (ret[1]) {
+            throw takeFromExternrefTable0(ret[0]);
+        }
+    }
     /**
      * Delete all matching documents. Returns the count deleted.
      * @param {any} filter
@@ -78,6 +97,18 @@ export class CollectionWasm {
             throw takeFromExternrefTable0(ret[0]);
         }
     }
+    /**
+     * Drop a vector index.
+     * @param {string} field
+     */
+    dropVectorIndex(field) {
+        const ptr0 = passStringToWasm0(field, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ret = wasm.collectionwasm_dropVectorIndex(this.__wbg_ptr, ptr0, len0);
+        if (ret[1]) {
+            throw takeFromExternrefTable0(ret[0]);
+        }
+    }
     /**
      * Find documents matching the filter. Returns a JS array of plain objects.
      * @param {any} filter
@@ -90,6 +121,30 @@ export class CollectionWasm {
         }
         return takeFromExternrefTable0(ret[0]);
     }
+    /**
+     * Find the `top_k` nearest documents to `query` on a vector index.
+     *
+     * `filter` — optional pre-filter (same format as `find`). Pass `null` to
+     *            search across all documents that have the vector field.
+     *
+     * Returns a JSON array of `{ document: {...}, score: number }` objects.
+     * @param {string} field
+     * @param {Float32Array} query
+     * @param {number} top_k
+     * @param {any} filter
+     * @returns {any}
+     */
+    findNearest(field, query, top_k, filter) {
+        const ptr0 = passStringToWasm0(field, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passArrayF32ToWasm0(query, wasm.__wbindgen_malloc);
+        const len1 = WASM_VECTOR_LEN;
+        const ret = wasm.collectionwasm_findNearest(this.__wbg_ptr, ptr0, len0, ptr1, len1, top_k, filter);
+        if (ret[2]) {
+            throw takeFromExternrefTable0(ret[1]);
+        }
+        return takeFromExternrefTable0(ret[0]);
+    }
     /**
      * Find a single document. Returns the document or null.
      * @param {any} filter
@@ -314,6 +369,25 @@ export class WorkerDB {
             throw takeFromExternrefTable0(ret[0]);
         }
     }
+    /**
+     * Create a vector index. `metric_str`: `"cosine"` | `"dot"` | `"euclidean"` (default cosine).
+     * @param {string} collection
+     * @param {string} field
+     * @param {number} dimensions
+     * @param {string | null} [metric_str]
+     */
+    createVectorIndex(collection, field, dimensions, metric_str) {
+        const ptr0 = passStringToWasm0(collection, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passStringToWasm0(field, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len1 = WASM_VECTOR_LEN;
+        var ptr2 = isLikeNone(metric_str) ? 0 : passStringToWasm0(metric_str, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        var len2 = WASM_VECTOR_LEN;
+        const ret = wasm.workerdb_createVectorIndex(this.__wbg_ptr, ptr0, len0, ptr1, len1, dimensions, ptr2, len2);
+        if (ret[1]) {
+            throw takeFromExternrefTable0(ret[0]);
+        }
+    }
     /**
      * Delete all matching documents. Returns the count deleted.
      * @param {string} collection
@@ -376,6 +450,21 @@ export class WorkerDB {
             throw takeFromExternrefTable0(ret[0]);
         }
     }
+    /**
+     * Drop a vector index.
+     * @param {string} collection
+     * @param {string} field
+     */
+    dropVectorIndex(collection, field) {
+        const ptr0 = passStringToWasm0(collection, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passStringToWasm0(field, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len1 = WASM_VECTOR_LEN;
+        const ret = wasm.workerdb_dropVectorIndex(this.__wbg_ptr, ptr0, len0, ptr1, len1);
+        if (ret[1]) {
+            throw takeFromExternrefTable0(ret[0]);
+        }
+    }
     /**
      * Find documents. Returns a JSON array of document objects.
      * @param {string} collection
@@ -404,6 +493,41 @@ export class WorkerDB {
             wasm.__wbindgen_free(deferred4_0, deferred4_1, 1);
         }
     }
+    /**
+     * Find nearest neighbours. Returns a JSON string of `[{ document, score }]`.
+     * @param {string} collection
+     * @param {string} field
+     * @param {string} query_json
+     * @param {number} top_k
+     * @param {string} filter_json
+     * @returns {string}
+     */
+    findNearest(collection, field, query_json, top_k, filter_json) {
+        let deferred6_0;
+        let deferred6_1;
+        try {
+            const ptr0 = passStringToWasm0(collection, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+            const len0 = WASM_VECTOR_LEN;
+            const ptr1 = passStringToWasm0(field, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+            const len1 = WASM_VECTOR_LEN;
+            const ptr2 = passStringToWasm0(query_json, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+            const len2 = WASM_VECTOR_LEN;
+            const ptr3 = passStringToWasm0(filter_json, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+            const len3 = WASM_VECTOR_LEN;
+            const ret = wasm.workerdb_findNearest(this.__wbg_ptr, ptr0, len0, ptr1, len1, ptr2, len2, top_k, ptr3, len3);
+            var ptr5 = ret[0];
+            var len5 = ret[1];
+            if (ret[3]) {
+                ptr5 = 0; len5 = 0;
+                throw takeFromExternrefTable0(ret[2]);
+            }
+            deferred6_0 = ptr5;
+            deferred6_1 = len5;
+            return getStringFromWasm0(ptr5, len5);
+        } finally {
+            wasm.__wbindgen_free(deferred6_0, deferred6_1, 1);
+        }
+    }
     /**
      * Find one document. Returns a JSON object or `"null"`.
      * @param {string} collection
@@ -1119,7 +1243,7 @@ function __wbg_get_imports() {
             return ret;
         },
         __wbindgen_cast_0000000000000001: function(arg0, arg1) {
-            // Cast intrinsic for `Closure(Closure { dtor_idx: 204, function: Function { arguments: [Externref], shim_idx: 205, ret: Result(Unit), inner_ret: Some(Result(Unit)) }, mutable: true }) -> Externref`.
+            // Cast intrinsic for `Closure(Closure { dtor_idx: 218, function: Function { arguments: [Externref], shim_idx: 219, ret: Result(Unit), inner_ret: Some(Result(Unit)) }, mutable: true }) -> Externref`.
             const ret = makeMutClosure(arg0, arg1, wasm.wasm_bindgen__closure__destroy__h1411c0b1433bd547, wasm_bindgen__convert__closures_____invoke__h4bd31e7387ab665d);
             return ret;
         },
@@ -1275,6 +1399,14 @@ function getDataViewMemory0() {
     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) {
     ptr = ptr >>> 0;
     return decodeText(ptr, len);
@@ -1336,6 +1468,13 @@ function passArray8ToWasm0(arg, malloc) {
     return ptr;
 }
 
+function passArrayF32ToWasm0(arg, malloc) {
+    const ptr = malloc(arg.length * 4, 4) >>> 0;
+    getFloat32ArrayMemory0().set(arg, ptr / 4);
+    WASM_VECTOR_LEN = arg.length;
+    return ptr;
+}
+
 function passStringToWasm0(arg, malloc, realloc) {
     if (realloc === undefined) {
         const buf = cachedTextEncoder.encode(arg);
@@ -1413,6 +1552,7 @@ function __wbg_finalize_init(instance, module) {
     wasm = instance.exports;
     wasmModule = module;
     cachedDataViewMemory0 = null;
+    cachedFloat32ArrayMemory0 = null;
     cachedUint8ArrayMemory0 = null;
     wasm.__wbindgen_start();
     return wasm;

package/worker/taladb.worker.js

@@ -34,9 +34,12 @@
  * count         { collection, filterJson }
  * createIndex   { collection, field }
  * dropIndex     { collection, field }
- * createFtsIndex { collection, field }
- * dropFtsIndex  { collection, field }
- * close         {}
+ * createFtsIndex    { collection, field }
+ * dropFtsIndex      { collection, field }
+ * createVectorIndex { collection, field, dimensions, metric? }
+ * dropVectorIndex   { collection, field }
+ * findNearest       { collection, field, queryJson, topK, filterJson? }
+ * close             {}
  */
 
 // ---------------------------------------------------------------------------
@@ -150,6 +153,23 @@ async function dispatch(op, args) {
       db.dropFtsIndex(args.collection, args.field);
       return null;
 
+    case 'createVectorIndex':
+      db.createVectorIndex(args.collection, args.field, args.dimensions, args.metric ?? null);
+      return null;
+
+    case 'dropVectorIndex':
+      db.dropVectorIndex(args.collection, args.field);
+      return null;
+
+    case 'findNearest':
+      return db.findNearest(
+        args.collection,
+        args.field,
+        args.queryJson,
+        args.topK,
+        args.filterJson ?? 'null',
+      );
+
     case 'close':
       db = null;
       return null;