entroly-wasm 1.0.23 → 1.0.25

package/js/server.js

@@ -4,7 +4,7 @@
 //
 // Architecture: MCP Client → JSON-RPC (stdio) → Node.js → Wasm Engine → Results
 
-const { WasmEntrolyEngine } = require('../pkg/entroly_wasm');
+const { WasmEntrolyEngine, qccr_select } = require('../pkg/entroly_wasm');
 const { EntrolyConfig } = require('./config');
 const { CheckpointManager, persistIndex, loadIndex } = require('./checkpoint');
 const { autoIndex, startIncrementalWatcher } = require('./auto_index');
@@ -226,7 +226,35 @@ class EntrolyMCPServer {
         const budget = args.token_budget || this.config.defaultTokenBudget;
         const query = args.query || '';
         const profile = this.taskProfiles.applyToEngine(this.engine, query);
-        const result = this.engine.optimize(budget, query);
+        // ── Query-conditioned selection via QCCR (Rust SSOT) ──
+        // Routes npm's optimize_context through the SAME ranking + sentence
+        // selection as pip/MCP/SDK (the shared entroly-qccr crate). Any failure
+        // falls back to the native knapsack so optimize never breaks.
+        let result;
+        if (query.trim()) {
+          try {
+            const candidates = this.engine.export_fragments();
+            const slim = JSON.stringify(candidates.map(f => ({ source: f.source || '', content: f.content || '' })));
+            const selected = JSON.parse(qccr_select(slim, budget, query, '{}', '[]'));
+            const tok = (f) => f.token_count || Math.floor(((f.content || '').length) / 4);
+            const tokensUsed = selected.reduce((a, f) => a + tok(f), 0);
+            const totalInput = candidates.reduce((a, f) => a + tok(f), 0);
+            result = {
+              selected_fragments: selected,
+              selected,
+              selected_count: selected.length,
+              tokens_used: tokensUsed,
+              tokens_saved: Math.max(0, totalInput - tokensUsed),
+              total_fragments: candidates.length,
+              selector: 'qccr',
+            };
+          } catch (_) {
+            result = this.engine.optimize(budget, query);
+            result.selector = 'knapsack_fallback';
+          }
+        } else {
+          result = this.engine.optimize(budget, query);
+        }
         result._taskProfile = { taskType: profile.taskType, confidence: profile.confidence };
         // Record value to the shared sink (mirrors entroly/server.py).
         try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "entroly-wasm",
-  "version": "1.0.23",
+  "version": "1.0.25",
   "description": "Auditable context control plane for AI agents: context receipts, local optimization, MCP, and app SDK helpers. Pure WebAssembly, no Python dependency.",
   "main": "index.js",
   "types": "index.d.ts",

package/pkg/entroly_wasm.d.ts

@@ -207,4 +207,21 @@ export function classify_query_transition(prev_hash: bigint, current_query: stri
 
 export function optimize_task_profiles(episodes_json: string): any;
 
+/**
+ * Expand a query into its retrieval vocabulary; returns a sorted JSON array.
+ */
+export function qccr_expand_query(query: string): string;
+
+/**
+ * Rank files; returns a JSON array of `[index, score]` pairs, best-first.
+ */
+export function qccr_rank_files(sources: string[], texts: string[], query: string, overrides_json: string): string;
+
+/**
+ * Full QCCR selection (rank + sentence-MMR + emit + trim). `fragments_json`
+ * is a JSON array of `{source, content}`; `preferred_json` is the optional
+ * file order (or "[]"); returns a JSON array of selected fragments.
+ */
+export function qccr_select(fragments_json: string, token_budget: number, query: string, overrides_json: string, preferred_json: string): string;
+
 export function reward_weighted_optimize(episodes_json: string, current_weights_json: string): any;

package/pkg/entroly_wasm.js

@@ -488,6 +488,90 @@ function optimize_task_profiles(episodes_json) {
 }
 exports.optimize_task_profiles = optimize_task_profiles;
 
+/**
+ * Expand a query into its retrieval vocabulary; returns a sorted JSON array.
+ * @param {string} query
+ * @returns {string}
+ */
+function qccr_expand_query(query) {
+    let deferred2_0;
+    let deferred2_1;
+    try {
+        const ptr0 = passStringToWasm0(query, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ret = wasm.qccr_expand_query(ptr0, len0);
+        deferred2_0 = ret[0];
+        deferred2_1 = ret[1];
+        return getStringFromWasm0(ret[0], ret[1]);
+    } finally {
+        wasm.__wbindgen_free(deferred2_0, deferred2_1, 1);
+    }
+}
+exports.qccr_expand_query = qccr_expand_query;
+
+/**
+ * Rank files; returns a JSON array of `[index, score]` pairs, best-first.
+ * @param {string[]} sources
+ * @param {string[]} texts
+ * @param {string} query
+ * @param {string} overrides_json
+ * @returns {string}
+ */
+function qccr_rank_files(sources, texts, query, overrides_json) {
+    let deferred5_0;
+    let deferred5_1;
+    try {
+        const ptr0 = passArrayJsValueToWasm0(sources, wasm.__wbindgen_malloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passArrayJsValueToWasm0(texts, wasm.__wbindgen_malloc);
+        const len1 = WASM_VECTOR_LEN;
+        const ptr2 = passStringToWasm0(query, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len2 = WASM_VECTOR_LEN;
+        const ptr3 = passStringToWasm0(overrides_json, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len3 = WASM_VECTOR_LEN;
+        const ret = wasm.qccr_rank_files(ptr0, len0, ptr1, len1, ptr2, len2, ptr3, len3);
+        deferred5_0 = ret[0];
+        deferred5_1 = ret[1];
+        return getStringFromWasm0(ret[0], ret[1]);
+    } finally {
+        wasm.__wbindgen_free(deferred5_0, deferred5_1, 1);
+    }
+}
+exports.qccr_rank_files = qccr_rank_files;
+
+/**
+ * Full QCCR selection (rank + sentence-MMR + emit + trim). `fragments_json`
+ * is a JSON array of `{source, content}`; `preferred_json` is the optional
+ * file order (or "[]"); returns a JSON array of selected fragments.
+ * @param {string} fragments_json
+ * @param {number} token_budget
+ * @param {string} query
+ * @param {string} overrides_json
+ * @param {string} preferred_json
+ * @returns {string}
+ */
+function qccr_select(fragments_json, token_budget, query, overrides_json, preferred_json) {
+    let deferred5_0;
+    let deferred5_1;
+    try {
+        const ptr0 = passStringToWasm0(fragments_json, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passStringToWasm0(query, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len1 = WASM_VECTOR_LEN;
+        const ptr2 = passStringToWasm0(overrides_json, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len2 = WASM_VECTOR_LEN;
+        const ptr3 = passStringToWasm0(preferred_json, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len3 = WASM_VECTOR_LEN;
+        const ret = wasm.qccr_select(ptr0, len0, token_budget, ptr1, len1, ptr2, len2, ptr3, len3);
+        deferred5_0 = ret[0];
+        deferred5_1 = ret[1];
+        return getStringFromWasm0(ret[0], ret[1]);
+    } finally {
+        wasm.__wbindgen_free(deferred5_0, deferred5_1, 1);
+    }
+}
+exports.qccr_select = qccr_select;
+
 /**
  * @param {string} episodes_json
  * @param {string} current_weights_json
@@ -509,6 +593,14 @@ exports.reward_weighted_optimize = reward_weighted_optimize;
 function __wbg_get_imports() {
     const import0 = {
         __proto__: null,
+        __wbg___wbindgen_string_get_914df97fcfa788f2: function(arg0, arg1) {
+            const obj = arg1;
+            const ret = typeof(obj) === 'string' ? obj : undefined;
+            var ptr1 = isLikeNone(ret) ? 0 : passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+            var len1 = WASM_VECTOR_LEN;
+            getDataViewMemory0().setInt32(arg0 + 4 * 1, len1, true);
+            getDataViewMemory0().setInt32(arg0 + 4 * 0, ptr1, true);
+        },
         __wbg___wbindgen_throw_81fc77679af83bc6: function(arg0, arg1) {
             throw new Error(getStringFromWasm0(arg0, arg1));
         },
@@ -547,6 +639,14 @@ function addToExternrefTable0(obj) {
     return idx;
 }
 
+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;
+}
+
 function getStringFromWasm0(ptr, len) {
     ptr = ptr >>> 0;
     return decodeText(ptr, len);
@@ -569,6 +669,20 @@ function handleError(f, args) {
     }
 }
 
+function isLikeNone(x) {
+    return x === undefined || x === null;
+}
+
+function passArrayJsValueToWasm0(array, malloc) {
+    const ptr = malloc(array.length * 4, 4) >>> 0;
+    for (let i = 0; i < array.length; i++) {
+        const add = addToExternrefTable0(array[i]);
+        getDataViewMemory0().setUint32(ptr + 4 * i, add, true);
+    }
+    WASM_VECTOR_LEN = array.length;
+    return ptr;
+}
+
 function passStringToWasm0(arg, malloc, realloc) {
     if (realloc === undefined) {
         const buf = cachedTextEncoder.encode(arg);