@kotoshu/wasm 0.2.0 → 0.3.0

package/kotoshu_wasm.d.ts

@@ -65,3 +65,13 @@ export function loadModel(model_bytes: Uint8Array, vocab_bytes: Uint8Array): Kot
  * candidates as the gem's context boost does; see the module docs.
  */
 export function rerank(model: KotoshuModel, word: string, context: string): number;
+
+/**
+ * The `k` nearest vocabulary words to `word` by cosine — semantic
+ * candidate GENERATION (see `Int8Model::semantic_neighbors`): one plain
+ * `{ word, score }` row per neighbor, score-descending, the query word
+ * itself excluded. Empty when `k` is 0 or the word embeds nowhere (out
+ * of vocabulary with no in-vocabulary character n-gram). `k` defaults
+ * to 5 when omitted.
+ */
+export function semanticSuggest(model: KotoshuModel, word: string, k?: number | null): Array<any>;

package/kotoshu_wasm.js

@@ -5,5 +5,5 @@ import { __wbg_set_wasm } from "./kotoshu_wasm_bg.js";
 __wbg_set_wasm(wasm);
 wasm.__wbindgen_start();
 export {
-    KotoshuModel, KotoshuWasm, loadModel, rerank
+    KotoshuModel, KotoshuWasm, loadModel, rerank, semanticSuggest
 } from "./kotoshu_wasm_bg.js";

package/kotoshu_wasm_bg.js

@@ -151,6 +151,26 @@ export function rerank(model, word, context) {
     const ret = wasm.rerank(model.__wbg_ptr, ptr0, len0, ptr1, len1);
     return ret;
 }
+
+/**
+ * The `k` nearest vocabulary words to `word` by cosine — semantic
+ * candidate GENERATION (see `Int8Model::semantic_neighbors`): one plain
+ * `{ word, score }` row per neighbor, score-descending, the query word
+ * itself excluded. Empty when `k` is 0 or the word embeds nowhere (out
+ * of vocabulary with no in-vocabulary character n-gram). `k` defaults
+ * to 5 when omitted.
+ * @param {KotoshuModel} model
+ * @param {string} word
+ * @param {number | null} [k]
+ * @returns {Array<any>}
+ */
+export function semanticSuggest(model, word, k) {
+    _assertClass(model, KotoshuModel);
+    const ptr0 = passStringToWasm0(word, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.semanticSuggest(model.__wbg_ptr, ptr0, len0, isLikeNone(k) ? Number.MAX_SAFE_INTEGER : (k) >>> 0);
+    return ret;
+}
 export function __wbg_Error_408e67f47ca7b58b(arg0, arg1) {
     const ret = Error(getStringFromWasm0(arg0, arg1));
     return ret;

package/package.json

@@ -5,7 +5,7 @@
     "Ribose Inc. <open.source@ribose.com>"
   ],
   "description": "npm @kotoshu/wasm packaging shim: wasm-pack build of the kotoshu engine's wasm-bindgen surface",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "license": "BSD-2-Clause",
   "repository": {
     "type": "git",