agent-working-memory 0.16.1 → 0.18.1

package/dist/core/model-daemon-server.js

@@ -0,0 +1,363 @@
+// Copyright 2026 Robert Winter / Complete Ideas
+// SPDX-License-Identifier: Apache-2.0
+/**
+ * The shared model daemon process (see model-daemon.ts for the design).
+ *
+ * It runs ONE inference at a time. onnxruntime-node's CPU arena grows to the largest
+ * tensor a session has run, so concurrent runs on the same session would multiply the
+ * high-water mark this daemon exists to bound. One at a time costs no speed: four
+ * processes recalling at once on one machine finished in 12.3 min with their own models
+ * and 11.6 min through one daemon, because they compete for the same cores anyway.
+ *
+ * Scheduling: interactive requests run before background ones. An embed is run one
+ * chunk at a time; an unfinished interactive embed goes to the back of its lane, so
+ * sessions take turns and one session's 100-memory write-batch cannot hold up another
+ * session's recall; an unfinished background embed goes to the front of its lane, so
+ * its chunks stay in order. A request whose connection has closed is dropped unrun.
+ * Each step reports `started` when it begins, and a job sent back to the queue between
+ * chunks reports `requeued`, so a client's run timer covers only time the daemon spends
+ * on its job: its timeout measures a stuck daemon, never a busy one. A step that runs
+ * longer than AWM_MODEL_DAEMON_STUCK_MS (default: the client's step timeout, 2 min; for
+ * a warm, which may download the models, the client's 10 min) means the daemon is wedged
+ * on a promise that will never settle: it exits, and the next client starts a fresh one.
+ * The two limits are the same so that a client that gave up on a step is never left
+ * reconnecting to the daemon still stuck on it.
+ *
+ * The daemon exits when no authenticated client has been connected for
+ * AWM_MODEL_DAEMON_IDLE_MS (default 5 min). Clients hold one connection each for the
+ * life of their process.
+ */
+import { createServer, connect } from 'node:net';
+import { randomBytes } from 'node:crypto';
+import { chmodSync, existsSync, mkdirSync, readFileSync, renameSync, unlinkSync, writeFileSync } from 'node:fs';
+import { setTimeout as delay } from 'node:timers/promises';
+import { PROTOCOL_VERSION, HELLO_MAX_BYTES, MAX_FRAME_BYTES, addressProblem, encodeFrame, envInt, FrameReader, modelDaemonPaths, proof, proofMatches, runDirProblem, REQUEST_TIMEOUT_MS, WARM_TIMEOUT_MS, } from './model-daemon.js';
+import { VERSION } from '../version.js';
+const HELLO_TIMEOUT_MS = 5_000;
+/** Windows refuses a rename over a file another process has open; clients open it briefly. */
+async function replaceFile(tmp, target) {
+    for (let attempt = 0;; attempt++) {
+        try {
+            renameSync(tmp, target);
+            return;
+        }
+        catch (err) {
+            const code = err.code;
+            if ((code !== 'EPERM' && code !== 'EBUSY' && code !== 'EACCES') || attempt >= 40) {
+                try {
+                    unlinkSync(tmp);
+                }
+                catch { /* nothing to clean */ }
+                throw err;
+            }
+            await delay(25);
+        }
+    }
+}
+/** A unix socket file left by a crashed daemon refuses connections; a live one accepts. */
+function removeStaleSocket(address) {
+    if (process.platform === 'win32' || !existsSync(address))
+        return Promise.resolve();
+    return new Promise(resolve => {
+        const probe = connect(address);
+        probe.once('connect', () => { probe.destroy(); resolve(); });
+        probe.once('error', () => { try {
+            unlinkSync(address);
+        }
+        catch { /* raced */ } resolve(); });
+    });
+}
+export async function startModelDaemon(opts) {
+    const { fingerprint } = opts;
+    const paths = modelDaemonPaths(fingerprint, opts.runDir);
+    const idleMs = opts.idleMs ?? envInt('AWM_MODEL_DAEMON_IDLE_MS', 300_000);
+    const log = opts.log ?? ((m) => console.error(`[model-daemon ${new Date().toISOString()}] ${m}`));
+    const h = opts.handlers;
+    const problem = addressProblem(paths);
+    if (problem)
+        throw new Error(problem);
+    mkdirSync(paths.runDir, { recursive: true, mode: 0o700 });
+    try {
+        chmodSync(paths.runDir, 0o700);
+    }
+    catch { /* not ours to change, or Windows */ }
+    const unsafe = runDirProblem(paths.runDir);
+    if (unsafe)
+        throw new Error(unsafe);
+    const stuckMs = opts.stuckMs ?? envInt('AWM_MODEL_DAEMON_STUCK_MS', envInt('AWM_MODEL_DAEMON_TIMEOUT_MS', REQUEST_TIMEOUT_MS));
+    await removeStaleSocket(paths.address);
+    const token = randomBytes(32).toString('hex');
+    const sockets = new Set(); // every connection, authenticated or not
+    /** Requests a client gave up on, per connection: never started, or never continued. */
+    const cancelled = new WeakMap();
+    const isCancelled = (job) => cancelled.get(job.socket)?.has(job.id) ?? false;
+    /** Until the token file is ours, a hello is told to retry, not proved with a token the
+     * file does not hold yet (the rename may be retried on Windows). */
+    let tokenReady = false;
+    const clients = new Set(); // authenticated ones: these hold the lease
+    const interactive = [];
+    const background = [];
+    const served = {};
+    const startedAt = Date.now();
+    let busy = false;
+    let current = null;
+    let idleTimer = null;
+    let closing = null;
+    const send = (socket, frame) => {
+        if (!socket.destroyed)
+            socket.write(encodeFrame(frame));
+    };
+    const armIdle = () => {
+        if (idleTimer || clients.size > 0 || idleMs <= 0 || closing)
+            return;
+        idleTimer = setTimeout(() => {
+            idleTimer = null;
+            if (clients.size > 0)
+                return;
+            // A step still running (a first-run warm downloading the models) finishes first.
+            if (current || interactive.length || background.length) {
+                armIdle();
+                return;
+            }
+            log(`no clients for ${idleMs} ms, exiting`);
+            void running.close().then(() => (opts.onIdle ?? (() => process.exit(0)))());
+        }, idleMs);
+        idleTimer.unref?.();
+    };
+    /** Runs one step of a job. Returns true when the job is finished. */
+    async function step(job) {
+        send(job.socket, { id: job.id, started: true });
+        if (job.vectors) {
+            const texts = job.args.texts;
+            const size = h.embedChunk > 0 ? h.embedChunk : texts.length;
+            const slice = texts.slice(job.offset, job.offset + size);
+            job.vectors.push(...await h.embed({ ...job.args, texts: slice }));
+            job.offset += slice.length;
+            if (job.offset < texts.length)
+                return false;
+            send(job.socket, { id: job.id, ok: true, result: job.vectors });
+            return true;
+        }
+        let result;
+        switch (job.op) {
+            case 'embed':
+                result = await h.embed(job.args);
+                break;
+            case 'rerank':
+                result = await h.rerank(job.args);
+                break;
+            case 'expand':
+                result = await h.expand(job.args);
+                break;
+            case 'warm':
+                await h.warm();
+                result = true;
+                break;
+            default: throw new Error(`unknown op ${String(job.op)}`);
+        }
+        send(job.socket, { id: job.id, ok: true, result });
+        return true;
+    }
+    async function pump() {
+        if (busy)
+            return;
+        busy = true;
+        try {
+            for (;;) {
+                const job = interactive.shift() ?? background.shift();
+                if (!job)
+                    break;
+                if (job.socket.destroyed || isCancelled(job))
+                    continue;
+                let done = true;
+                current = { op: job.op, since: Date.now() };
+                try {
+                    done = await step(job);
+                }
+                catch (err) {
+                    send(job.socket, { id: job.id, ok: false, error: err?.message ?? String(err) });
+                }
+                finally {
+                    current = null;
+                }
+                if (done) {
+                    served[job.op] = (served[job.op] ?? 0) + 1;
+                    continue;
+                }
+                // Back in the queue between chunks: the client's run timer stops until the next
+                // step starts, because waiting here is the daemon being busy, not stuck.
+                if (isCancelled(job))
+                    continue;
+                send(job.socket, { id: job.id, requeued: true });
+                if (job.lane === 'interactive')
+                    interactive.push(job);
+                else
+                    background.unshift(job);
+            }
+        }
+        finally {
+            busy = false;
+        }
+    }
+    const onRequest = (socket, frame) => {
+        if (typeof frame?.id !== 'number')
+            return;
+        if (frame.op === 'cancel') {
+            // The client gave up waiting: drop the job, or stop it after its running step.
+            const id = frame.args?.id;
+            if (typeof id !== 'number')
+                return;
+            let set = cancelled.get(socket);
+            if (!set)
+                cancelled.set(socket, (set = new Set()));
+            if (set.size >= 10_000)
+                set.clear();
+            set.add(id);
+            for (const q of [interactive, background]) {
+                const i = q.findIndex(j => j.socket === socket && j.id === id);
+                if (i >= 0)
+                    q.splice(i, 1);
+            }
+            return;
+        }
+        if (frame.op === 'stats') {
+            send(socket, {
+                id: frame.id, ok: true, result: {
+                    pid: process.pid, fingerprint, version: VERSION, uptimeMs: Date.now() - startedAt,
+                    clients: clients.size, queued: { interactive: interactive.length, background: background.length },
+                    served, rssBytes: process.memoryUsage().rss,
+                    current: current ? { op: current.op, runningMs: Date.now() - current.since } : null,
+                },
+            });
+            return;
+        }
+        const lane = frame.lane === 'background' ? 'background' : 'interactive';
+        const job = { socket, id: frame.id, op: frame.op, lane, args: frame.args };
+        if (frame.op === 'embed' && Array.isArray(frame.args?.texts)) {
+            job.offset = 0;
+            job.vectors = [];
+        }
+        (lane === 'background' ? background : interactive).push(job);
+        void pump();
+    };
+    const server = createServer(socket => {
+        sockets.add(socket);
+        socket.setEncoding('utf8');
+        let stage = 'hello';
+        let clientNonce = '';
+        let daemonNonce = '';
+        const helloTimer = setTimeout(() => socket.destroy(), HELLO_TIMEOUT_MS);
+        const refuse = (error) => { clearTimeout(helloTimer); socket.end(encodeFrame({ t: 'hello', ok: false, error })); };
+        const reader = new FrameReader(frame => {
+            if (stage === 'ready')
+                return onRequest(socket, frame);
+            if (closing)
+                return refuse('closing');
+            if (!tokenReady)
+                return refuse('starting');
+            if (stage === 'hello') {
+                const hello = frame;
+                if (hello?.t !== 'hello')
+                    return refuse('expected hello');
+                if (hello.protocol !== PROTOCOL_VERSION)
+                    return refuse(`protocol ${hello.protocol} != ${PROTOCOL_VERSION}`);
+                if (hello.fingerprint !== fingerprint)
+                    return refuse('fingerprint mismatch');
+                if (typeof hello.nonce !== 'string' || hello.nonce.length < 32)
+                    return refuse('bad nonce');
+                clientNonce = hello.nonce;
+                daemonNonce = randomBytes(16).toString('hex');
+                stage = 'auth';
+                socket.write(encodeFrame({
+                    t: 'challenge', nonce: daemonNonce, pid: process.pid, version: VERSION,
+                    proof: proof(token, 'daemon', clientNonce, daemonNonce, fingerprint),
+                }));
+                return;
+            }
+            const auth = frame;
+            if (auth?.t !== 'auth' || !proofMatches(proof(token, 'client', clientNonce, daemonNonce, fingerprint), auth.proof)) {
+                return refuse('bad token');
+            }
+            clearTimeout(helloTimer);
+            stage = 'ready';
+            reader.limit = MAX_FRAME_BYTES;
+            clients.add(socket);
+            if (idleTimer) {
+                clearTimeout(idleTimer);
+                idleTimer = null;
+            }
+            socket.write(encodeFrame({ t: 'hello', ok: true }));
+        }, HELLO_MAX_BYTES);
+        socket.on('data', (chunk) => { if (!reader.push(chunk))
+            socket.destroy(); });
+        socket.on('error', () => { });
+        socket.on('close', () => {
+            clearTimeout(helloTimer);
+            sockets.delete(socket);
+            if (clients.delete(socket))
+                armIdle();
+        });
+    });
+    await new Promise((resolve, reject) => {
+        server.once('error', reject);
+        server.listen(paths.address, () => { server.off('error', reject); resolve(); });
+    });
+    if (process.platform !== 'win32') {
+        try {
+            chmodSync(paths.address, 0o600);
+        }
+        catch { /* best effort */ }
+    }
+    // The token is written only once this daemon owns the address. A second daemon that
+    // lost the race (EADDRINUSE above) must not overwrite the live one's token. A client
+    // that connected in the gap read no token or an old one; it is refused and retries.
+    const tmp = `${paths.token}.${process.pid}`;
+    writeFileSync(tmp, token, { mode: 0o600 });
+    try {
+        await replaceFile(tmp, paths.token);
+    }
+    catch (err) {
+        server.close();
+        throw err;
+    }
+    tokenReady = true;
+    log(`listening on ${paths.address} (pid ${process.pid}, AWM ${VERSION}, fingerprint ${fingerprint})`);
+    armIdle();
+    const watchdog = stuckMs > 0 ? setInterval(() => {
+        if (!current)
+            return;
+        const limit = current.op === 'warm' ? Math.max(stuckMs, WARM_TIMEOUT_MS) : stuckMs;
+        if (Date.now() - current.since <= limit)
+            return;
+        log(`${current.op} has run for ${Date.now() - current.since} ms: wedged, exiting so a fresh daemon can start`);
+        if (watchdog)
+            clearInterval(watchdog);
+        void running.close().then(() => (opts.onStuck ?? (() => process.exit(1)))());
+    }, Math.min(5_000, Math.max(50, Math.floor(stuckMs / 4)))) : null;
+    watchdog?.unref?.();
+    const running = {
+        server, address: paths.address, fingerprint,
+        close: () => (closing ??= new Promise(resolve => {
+            if (idleTimer) {
+                clearTimeout(idleTimer);
+                idleTimer = null;
+            }
+            if (watchdog)
+                clearInterval(watchdog);
+            // Every connection, including one still in its handshake: server.close() waits for
+            // all of them, and a late hello must not authenticate against a closing daemon.
+            for (const s of sockets)
+                s.destroy();
+            server.close(() => {
+                try {
+                    if (readFileSync(paths.token, 'utf8') === token)
+                        unlinkSync(paths.token);
+                }
+                catch { /* already gone */ }
+                resolve();
+            });
+        })),
+    };
+    return running;
+}
+//# sourceMappingURL=model-daemon-server.js.map

package/dist/core/model-daemon-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-daemon-server.js","sourceRoot":"","sources":["../../src/core/model-daemon-server.ts"],"names":[],"mappings":"AAAA,gDAAgD;AAChD,sCAAsC;AACtC;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AAEH,OAAO,EAAE,YAAY,EAAE,OAAO,EAA4B,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,UAAU,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAChH,OAAO,EAAE,UAAU,IAAI,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EACL,gBAAgB,EAAE,eAAe,EAAE,eAAe,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,EACpG,gBAAgB,EAAE,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,kBAAkB,EAAE,eAAe,GAE1F,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AA2CxC,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAE/B,8FAA8F;AAC9F,KAAK,UAAU,WAAW,CAAC,GAAW,EAAE,MAAc;IACpD,KAAK,IAAI,OAAO,GAAG,CAAC,GAAI,OAAO,EAAE,EAAE,CAAC;QAClC,IAAI,CAAC;YAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAAC,OAAO;QAAC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACpD,MAAM,IAAI,GAAI,GAA6B,CAAC,IAAI,CAAC;YACjD,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,QAAQ,CAAC,IAAI,OAAO,IAAI,EAAE,EAAE,CAAC;gBACjF,IAAI,CAAC;oBAAC,UAAU,CAAC,GAAG,CAAC,CAAC;gBAAC,CAAC;gBAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC;gBACzD,MAAM,GAAG,CAAC;YACZ,CAAC;YACD,MAAM,KAAK,CAAC,EAAE,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;AACH,CAAC;AAED,2FAA2F;AAC3F,SAAS,iBAAiB,CAAC,OAAe;IACxC,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;QAAE,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IACnF,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;QAC3B,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/B,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7D,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;YAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/F,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,IAAmB;IACxD,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;IAC7B,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACzD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,0BAA0B,EAAE,OAAO,CAAC,CAAC;IAC1E,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1G,MAAM,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC;IAExB,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;IACtC,IAAI,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;IACtC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1D,IAAI,CAAC;QAAC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAAC,CAAC;IAAC,MAAM,CAAC,CAAC,oCAAoC,CAAC,CAAC;IACtF,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC3C,IAAI,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;IACpC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC,2BAA2B,EAAE,MAAM,CAAC,6BAA6B,EAAE,kBAAkB,CAAC,CAAC,CAAC;IAC/H,MAAM,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAEvC,MAAM,KAAK,GAAG,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC9C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC,CAAG,yCAAyC;IAC9E,uFAAuF;IACvF,MAAM,SAAS,GAAG,IAAI,OAAO,EAAuB,CAAC;IACrD,MAAM,WAAW,GAAG,CAAC,GAAQ,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,KAAK,CAAC;IAClF;wEACoE;IACpE,IAAI,UAAU,GAAG,KAAK,CAAC;IACvB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC,CAAG,2CAA2C;IAChF,MAAM,WAAW,GAAU,EAAE,CAAC;IAC9B,MAAM,UAAU,GAAU,EAAE,CAAC;IAC7B,MAAM,MAAM,GAA2B,EAAE,CAAC;IAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,IAAI,IAAI,GAAG,KAAK,CAAC;IACjB,IAAI,OAAO,GAAyC,IAAI,CAAC;IACzD,IAAI,SAAS,GAAyC,IAAI,CAAC;IAC3D,IAAI,OAAO,GAAyB,IAAI,CAAC;IAEzC,MAAM,IAAI,GAAG,CAAC,MAAc,EAAE,KAAoB,EAAE,EAAE;QACpD,IAAI,CAAC,MAAM,CAAC,SAAS;YAAE,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,IAAI,SAAS,IAAI,OAAO,CAAC,IAAI,GAAG,CAAC,IAAI,MAAM,IAAI,CAAC,IAAI,OAAO;YAAE,OAAO;QACpE,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;YAC1B,SAAS,GAAG,IAAI,CAAC;YACjB,IAAI,OAAO,CAAC,IAAI,GAAG,CAAC;gBAAE,OAAO;YAC7B,iFAAiF;YACjF,IAAI,OAAO,IAAI,WAAW,CAAC,MAAM,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;gBAAC,OAAO,EAAE,CAAC;gBAAC,OAAO;YAAC,CAAC;YAC9E,GAAG,CAAC,kBAAkB,MAAM,cAAc,CAAC,CAAC;YAC5C,KAAK,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC9E,CAAC,EAAE,MAAM,CAAC,CAAC;QACX,SAAS,CAAC,KAAK,EAAE,EAAE,CAAC;IACtB,CAAC,CAAC;IAEF,qEAAqE;IACrE,KAAK,UAAU,IAAI,CAAC,GAAQ;QAC1B,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAChD,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChB,MAAM,KAAK,GAAa,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC;YACvC,MAAM,IAAI,GAAG,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;YAC5D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,MAAO,EAAE,GAAG,CAAC,MAAO,GAAG,IAAI,CAAC,CAAC;YAC3D,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YAClE,GAAG,CAAC,MAAO,IAAI,KAAK,CAAC,MAAM,CAAC;YAC5B,IAAI,GAAG,CAAC,MAAO,GAAG,KAAK,CAAC,MAAM;gBAAE,OAAO,KAAK,CAAC;YAC7C,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YAChE,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,MAAe,CAAC;QACpB,QAAQ,GAAG,CAAC,EAAE,EAAE,CAAC;YACf,KAAK,OAAO;gBAAE,MAAM,GAAG,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAAC,MAAM;YACtD,KAAK,QAAQ;gBAAE,MAAM,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAAC,MAAM;YACxD,KAAK,QAAQ;gBAAE,MAAM,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAAC,MAAM;YACxD,KAAK,MAAM;gBAAE,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;gBAAC,MAAM,GAAG,IAAI,CAAC;gBAAC,MAAM;YAClD,OAAO,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,cAAc,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QACnD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,UAAU,IAAI;QACjB,IAAI,IAAI;YAAE,OAAO;QACjB,IAAI,GAAG,IAAI,CAAC;QACZ,IAAI,CAAC;YACH,SAAS,CAAC;gBACR,MAAM,GAAG,GAAG,WAAW,CAAC,KAAK,EAAE,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBACtD,IAAI,CAAC,GAAG;oBAAE,MAAM;gBAChB,IAAI,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,WAAW,CAAC,GAAG,CAAC;oBAAE,SAAS;gBACvD,IAAI,IAAI,GAAG,IAAI,CAAC;gBAChB,OAAO,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;gBAC5C,IAAI,CAAC;oBACH,IAAI,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;gBACzB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAG,GAAa,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC7F,CAAC;wBAAS,CAAC;oBACT,OAAO,GAAG,IAAI,CAAC;gBACjB,CAAC;gBACD,IAAI,IAAI,EAAE,CAAC;oBAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;oBAAC,SAAS;gBAAC,CAAC;gBACnE,gFAAgF;gBAChF,yEAAyE;gBACzE,IAAI,WAAW,CAAC,GAAG,CAAC;oBAAE,SAAS;gBAC/B,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBACjD,IAAI,GAAG,CAAC,IAAI,KAAK,aAAa;oBAAE,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;;oBACjD,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,GAAG,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GAAG,CAAC,MAAc,EAAE,KAAmB,EAAE,EAAE;QACxD,IAAI,OAAO,KAAK,EAAE,EAAE,KAAK,QAAQ;YAAE,OAAO;QAC1C,IAAI,KAAK,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;YAC1B,+EAA+E;YAC/E,MAAM,EAAE,GAAI,KAAK,CAAC,IAAY,EAAE,EAAE,CAAC;YACnC,IAAI,OAAO,EAAE,KAAK,QAAQ;gBAAE,OAAO;YACnC,IAAI,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,GAAG;gBAAE,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,GAAG,GAAG,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC;YACnD,IAAI,GAAG,CAAC,IAAI,IAAI,MAAM;gBAAE,GAAG,CAAC,KAAK,EAAE,CAAC;YACpC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACZ,KAAK,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,EAAE,CAAC;gBAC1C,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC/D,IAAI,CAAC,IAAI,CAAC;oBAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC7B,CAAC;YACD,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,EAAE,KAAK,OAAO,EAAE,CAAC;YACzB,IAAI,CAAC,MAAM,EAAE;gBACX,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;oBAC9B,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;oBACjF,OAAO,EAAE,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,WAAW,EAAE,WAAW,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,MAAM,EAAE;oBACjG,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC,GAAG;oBAC3C,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI;iBACpF;aACF,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QACD,MAAM,IAAI,GAAS,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,aAAa,CAAC;QAC9E,MAAM,GAAG,GAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;QAChF,IAAI,KAAK,CAAC,EAAE,KAAK,OAAO,IAAI,KAAK,CAAC,OAAO,CAAE,KAAK,CAAC,IAAY,EAAE,KAAK,CAAC,EAAE,CAAC;YACtE,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC;YACf,GAAG,CAAC,OAAO,GAAG,EAAE,CAAC;QACnB,CAAC;QACD,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7D,KAAK,IAAI,EAAE,CAAC;IACd,CAAC,CAAC;IAEF,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC,EAAE;QACnC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpB,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC3B,IAAI,KAAK,GAA+B,OAAO,CAAC;QAChD,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,gBAAgB,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,CAAC,KAAa,EAAE,EAAE,GAAG,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3H,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,EAAE;YACrC,IAAI,KAAK,KAAK,OAAO;gBAAE,OAAO,SAAS,CAAC,MAAM,EAAE,KAAqB,CAAC,CAAC;YACvE,IAAI,OAAO;gBAAE,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC;YACtC,IAAI,CAAC,UAAU;gBAAE,OAAO,MAAM,CAAC,UAAU,CAAC,CAAC;YAC3C,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;gBACtB,MAAM,KAAK,GAAG,KAAmB,CAAC;gBAClC,IAAI,KAAK,EAAE,CAAC,KAAK,OAAO;oBAAE,OAAO,MAAM,CAAC,gBAAgB,CAAC,CAAC;gBAC1D,IAAI,KAAK,CAAC,QAAQ,KAAK,gBAAgB;oBAAE,OAAO,MAAM,CAAC,YAAY,KAAK,CAAC,QAAQ,OAAO,gBAAgB,EAAE,CAAC,CAAC;gBAC5G,IAAI,KAAK,CAAC,WAAW,KAAK,WAAW;oBAAE,OAAO,MAAM,CAAC,sBAAsB,CAAC,CAAC;gBAC7E,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,EAAE;oBAAE,OAAO,MAAM,CAAC,WAAW,CAAC,CAAC;gBAC3F,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC;gBAC1B,WAAW,GAAG,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;gBAC9C,KAAK,GAAG,MAAM,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC;oBACvB,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,OAAO;oBACtE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;iBACrE,CAAC,CAAC,CAAC;gBACJ,OAAO;YACT,CAAC;YACD,MAAM,IAAI,GAAG,KAAkB,CAAC;YAChC,IAAI,IAAI,EAAE,CAAC,KAAK,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACnH,OAAO,MAAM,CAAC,WAAW,CAAC,CAAC;YAC7B,CAAC;YACD,YAAY,CAAC,UAAU,CAAC,CAAC;YACzB,KAAK,GAAG,OAAO,CAAC;YAChB,MAAM,CAAC,KAAK,GAAG,eAAe,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACpB,IAAI,SAAS,EAAE,CAAC;gBAAC,YAAY,CAAC,SAAS,CAAC,CAAC;gBAAC,SAAS,GAAG,IAAI,CAAC;YAAC,CAAC;YAC7D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC,EAAE,eAAe,CAAC,CAAC;QACpB,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACrF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,GAAqC,CAAC,CAAC,CAAC;QAChE,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACtB,YAAY,CAAC,UAAU,CAAC,CAAC;YACzB,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACvB,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;gBAAE,OAAO,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;IACH,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QAAC,IAAI,CAAC;YAAC,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC;IAAC,CAAC;IAC1G,oFAAoF;IACpF,qFAAqF;IACrF,oFAAoF;IACpF,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC,KAAK,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IAC5C,aAAa,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC3C,IAAI,CAAC;QACH,MAAM,WAAW,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,CAAC;IACZ,CAAC;IACD,UAAU,GAAG,IAAI,CAAC;IAClB,GAAG,CAAC,gBAAgB,KAAK,CAAC,OAAO,SAAS,OAAO,CAAC,GAAG,SAAS,OAAO,iBAAiB,WAAW,GAAG,CAAC,CAAC;IACtG,OAAO,EAAE,CAAC;IAEV,MAAM,QAAQ,GAAG,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE;QAC9C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,KAAK,GAAG,OAAO,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QACnF,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,KAAK,IAAI,KAAK;YAAE,OAAO;QAChD,GAAG,CAAC,GAAG,OAAO,CAAC,EAAE,gBAAgB,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,KAAK,kDAAkD,CAAC,CAAC;QAC/G,IAAI,QAAQ;YAAE,aAAa,CAAC,QAAQ,CAAC,CAAC;QACtC,KAAK,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC/E,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAClE,QAAQ,EAAE,KAAK,EAAE,EAAE,CAAC;IAEpB,MAAM,OAAO,GAAkB;QAC7B,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW;QAC3C,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,OAAO,KAAK,IAAI,OAAO,CAAO,OAAO,CAAC,EAAE;YACpD,IAAI,SAAS,EAAE,CAAC;gBAAC,YAAY,CAAC,SAAS,CAAC,CAAC;gBAAC,SAAS,GAAG,IAAI,CAAC;YAAC,CAAC;YAC7D,IAAI,QAAQ;gBAAE,aAAa,CAAC,QAAQ,CAAC,CAAC;YACtC,mFAAmF;YACnF,gFAAgF;YAChF,KAAK,MAAM,CAAC,IAAI,OAAO;gBAAE,CAAC,CAAC,OAAO,EAAE,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;gBAChB,IAAI,CAAC;oBAAC,IAAI,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,KAAK;wBAAE,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAAC,CAAC;gBAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;gBAC9G,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;KACJ,CAAC;IACF,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/core/model-daemon.d.ts

@@ -0,0 +1,137 @@
+/**
+ * Shared model daemon: what the client and the daemon agree on.
+ *
+ * Every AWM MCP process used to load its own three ONNX models (embedder, cross-encoder,
+ * query expander): about 1.1 GB per process even with the input-shape caps, so six
+ * Claude Code sessions held 6+ GB of identical weights and arenas. With
+ * AWM_MODEL_DAEMON=1, one per-user process holds the models and every AWM process on the
+ * machine sends inference to it over a local socket. Store, engine and recall stay in
+ * each process; only the three dispatch calls in ml-worker.ts cross the socket.
+ *
+ * This module has no model imports, so a client that only talks to the daemon never
+ * loads @huggingface/transformers' models itself.
+ *
+ * - The daemon is keyed by a FINGERPRINT of the inference config the model modules
+ *   actually captured (model ids, dtype, dimensions, pooling, shape caps), the model
+ *   cache dir, the AWM version and package root, the user and the run dir. Two installs,
+ *   two versions or two configs get separate daemons and can never mix vectors. The
+ *   spawned daemon is given the client's captured config as env, so a value applied
+ *   after the modules loaded (a .env file) cannot make the two disagree; the handshake
+ *   checks the fingerprint again.
+ * - Transport: a named pipe on Windows, a unix socket in ~/.awm/run (0700) elsewhere.
+ * - Authentication is mutual, by HMAC over two nonces with the token the daemon wrote to
+ *   the run dir (0600). The token itself never crosses the socket, so a process that
+ *   took the pipe name first learns nothing and is refused by the client.
+ * - Frames are newline-delimited JSON. JSON.stringify escapes every newline inside a
+ *   value, so a newline only ever ends a frame. Before authentication a connection may
+ *   send HELLO_MAX_BYTES; after it, MAX_FRAME_BYTES.
+ */
+export declare const PROTOCOL_VERSION = 2;
+export declare const MAX_FRAME_BYTES: number;
+export declare const HELLO_MAX_BYTES: number;
+/** A step must finish within this (client run timer and daemon watchdog both). */
+export declare const REQUEST_TIMEOUT_MS = 120000;
+/** The first warm can include a first-run model download (client and daemon both). */
+export declare const WARM_TIMEOUT_MS: number;
+/** Exit code of a daemon whose own fingerprint is not the one its spawner expected. */
+export declare const EXIT_FINGERPRINT_DRIFT = 3;
+export type ModelOp = 'embed' | 'rerank' | 'expand' | 'warm' | 'stats' | 'cancel';
+/** Interactive work (prime hook, recall, writes) runs ahead of background backfill. */
+export type Lane = 'interactive' | 'background';
+/** What the model modules captured at import; registered with ml-worker. */
+export type InferenceConfig = Record<string, string | number>;
+/** Client → daemon, first frame. */
+export interface HelloFrame {
+    t: 'hello';
+    protocol: number;
+    fingerprint: string;
+    nonce: string;
+}
+/** Daemon → client: its proof, and a refusal carries `error`. */
+export interface ChallengeFrame {
+    t: 'challenge';
+    nonce: string;
+    proof: string;
+    pid: number;
+    version: string;
+}
+/** Client → daemon: the client's proof. */
+export interface AuthFrame {
+    t: 'auth';
+    proof: string;
+}
+export interface HelloResult {
+    t: 'hello';
+    ok: boolean;
+    error?: string;
+}
+export interface RequestFrame {
+    id: number;
+    op: ModelOp;
+    lane?: Lane;
+    args?: unknown;
+}
+/** `started` starts the client's run timer and `requeued` stops it; the final frame has `ok`. */
+export interface ResponseFrame {
+    id: number;
+    ok?: boolean;
+    started?: true;
+    requeued?: true;
+    result?: unknown;
+    error?: string;
+}
+export declare function configEnv(config: InferenceConfig): Record<string, string>;
+export declare function packageRoot(): string;
+export declare function defaultRunDir(env?: NodeJS.ProcessEnv): string;
+/**
+ * Size and mtime of the modules whose code decides what inference returns. A rebuild or
+ * a dev edit without a version bump is a different daemon, not the old one serving old
+ * code to new clients. The client and the daemon run from the same files, so they agree.
+ */
+export declare function codeStamp(): string;
+/** 16 hex chars. Everything is a parameter so tests can compare configs in one process. */
+export declare function modelDaemonFingerprint(config: InferenceConfig, o?: {
+    env?: NodeJS.ProcessEnv;
+    version?: string;
+    root?: string;
+    code?: string;
+}): string;
+export interface DaemonPaths {
+    runDir: string;
+    /** Pipe name on Windows, socket path elsewhere. */
+    address: string;
+    token: string;
+    lock: string;
+    log: string;
+}
+export declare function modelDaemonPaths(fingerprint: string, runDir?: string): DaemonPaths;
+/** A unix socket path the OS will refuse, or null. */
+export declare function addressProblem(paths: DaemonPaths): string | null;
+/**
+ * A run dir another user can write to, or that this user does not own, is refused off
+ * Windows: the token and the log are written there, and another user could plant a
+ * symlink in their place. (On Windows the dir sits under the user's profile, whose ACL
+ * already excludes other users.)
+ */
+export declare function runDirProblem(runDir: string): string | null;
+export declare function proof(token: string, role: 'daemon' | 'client', clientNonce: string, daemonNonce: string, fingerprint: string): string;
+export declare function proofMatches(expected: string, got: unknown): boolean;
+/**
+ * Splits a byte stream into frames. push() returns false when the unterminated tail is
+ * over `limit`. Chunks without a newline are kept as a list and only the new chunk is
+ * searched, so a long frame arriving in many chunks costs linear time: appending to one
+ * string and searching it again would copy the whole buffer on every chunk.
+ */
+export declare class FrameReader {
+    private readonly onFrame;
+    limit: number;
+    private parts;
+    private size;
+    constructor(onFrame: (frame: unknown) => void, limit?: number);
+    push(chunk: string): boolean;
+}
+export declare function encodeFrame(frame: unknown): string;
+export declare function isModelDaemonEnabled(): boolean;
+/** Non-negative integer env var; anything else falls back (an explicit 0 is kept). */
+export declare function envInt(name: string, fallback: number): number;
+//# sourceMappingURL=model-daemon.d.ts.map

package/dist/core/model-daemon.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-daemon.d.ts","sourceRoot":"","sources":["../../src/core/model-daemon.ts"],"names":[],"mappings":"AAEA;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AAUH,eAAO,MAAM,gBAAgB,IAAI,CAAC;AAClC,eAAO,MAAM,eAAe,QAAmB,CAAC;AAChD,eAAO,MAAM,eAAe,QAAW,CAAC;AACxC,kFAAkF;AAClF,eAAO,MAAM,kBAAkB,SAAU,CAAC;AAC1C,sFAAsF;AACtF,eAAO,MAAM,eAAe,QAAc,CAAC;AAC3C,uFAAuF;AACvF,eAAO,MAAM,sBAAsB,IAAI,CAAC;AAIxC,MAAM,MAAM,OAAO,GAAG,OAAO,GAAG,QAAQ,GAAG,QAAQ,GAAG,MAAM,GAAG,OAAO,GAAG,QAAQ,CAAC;AAClF,uFAAuF;AACvF,MAAM,MAAM,IAAI,GAAG,aAAa,GAAG,YAAY,CAAC;AAEhD,4EAA4E;AAC5E,MAAM,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,CAAC;AAE9D,oCAAoC;AACpC,MAAM,WAAW,UAAU;IAAG,CAAC,EAAE,OAAO,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE;AAChG,iEAAiE;AACjE,MAAM,WAAW,cAAc;IAAG,CAAC,EAAE,WAAW,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE;AAC9G,2CAA2C;AAC3C,MAAM,WAAW,SAAS;IAAG,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE;AACvD,MAAM,WAAW,WAAW;IAAG,CAAC,EAAE,OAAO,CAAC;IAAC,EAAE,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE;AACxE,MAAM,WAAW,YAAY;IAAG,EAAE,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,OAAO,CAAC;IAAC,IAAI,CAAC,EAAE,IAAI,CAAC;IAAC,IAAI,CAAC,EAAE,OAAO,CAAA;CAAE;AACtF,iGAAiG;AACjG,MAAM,WAAW,aAAa;IAAG,EAAE,EAAE,MAAM,CAAC;IAAC,EAAE,CAAC,EAAE,OAAO,CAAC;IAAC,OAAO,CAAC,EAAE,IAAI,CAAC;IAAC,QAAQ,CAAC,EAAE,IAAI,CAAC;IAAC,MAAM,CAAC,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE;AAS9H,wBAAgB,SAAS,CAAC,MAAM,EAAE,eAAe,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAIzE;AAMD,wBAAgB,WAAW,IAAI,MAAM,CAGpC;AAED,wBAAgB,aAAa,CAAC,GAAG,GAAE,MAAM,CAAC,UAAwB,GAAG,MAAM,CAE1E;AAED;;;;GAIG;AACH,wBAAgB,SAAS,IAAI,MAAM,CAoBlC;AAED,2FAA2F;AAC3F,wBAAgB,sBAAsB,CACpC,MAAM,EAAE,eAAe,EACvB,CAAC,GAAE;IAAE,GAAG,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAO,GAClF,MAAM,CAWR;AAED,MAAM,WAAW,WAAW;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,mDAAmD;IACnD,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;CACb;AAED,wBAAgB,gBAAgB,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,GAAE,MAAwB,GAAG,WAAW,CAWnG;AAED,sDAAsD;AACtD,wBAAgB,cAAc,CAAC,KAAK,EAAE,WAAW,GAAG,MAAM,GAAG,IAAI,CAKhE;AAED;;;;;GAKG;AACH,wBAAgB,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAgB3D;AAED,wBAAgB,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,GAAG,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,CAErI;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,GAAG,OAAO,CAGpE;AAED;;;;;GAKG;AACH,qBAAa,WAAW;IAGV,OAAO,CAAC,QAAQ,CAAC,OAAO;IAAmC,KAAK;IAF5E,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,IAAI,CAAK;gBACY,OAAO,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,EAAS,KAAK,SAAkB;IAE9F,IAAI,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;CAe7B;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAElD;AAED,wBAAgB,oBAAoB,IAAI,OAAO,CAE9C;AAED,sFAAsF;AACtF,wBAAgB,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAG7D"}