@nodetool-ai/websocket 0.7.0-rc.23 → 0.8.2

package/dist/system-stats.d.ts

@@ -0,0 +1,52 @@
+type GpuStatsSnapshot = {
+    gpu_percent: number;
+    vram_total_gb: number;
+    vram_used_gb: number;
+    vram_percent: number;
+};
+type SystemStatsSnapshot = {
+    cpu_percent: number;
+    memory_percent: number;
+    memory_used: number;
+    memory_total: number;
+    memory_used_gb: number;
+    memory_total_gb: number;
+} & Partial<GpuStatsSnapshot>;
+/**
+ * Parse `nvidia-smi --query-gpu=utilization.gpu,memory.total,memory.used
+ * --format=csv,noheader,nounits` output (one CSV line per GPU; utilization in
+ * %, memory in MiB). With several GPUs, report the one with the most memory —
+ * that's where models land, and a summed figure would suggest budgets no
+ * single card has. Returns null when no line parses to a usable total.
+ */
+export declare function parseNvidiaSmiGpuStats(stdout: string): GpuStatsSnapshot | null;
+/**
+ * Build a stateful sampler that returns whole-system stats on each call.
+ * CPU% is computed from the delta of `os.cpus()` times since the last call,
+ * so the first call after construction returns ~0% (no delta yet).
+ *
+ * GPU stats come from an async `nvidia-smi` probe kicked off by each call and
+ * read from a cache, so the sampler itself stays synchronous; the first call
+ * carries no GPU fields, later ones do. A machine without a working
+ * `nvidia-smi` is marked unavailable after the first failed probe so no
+ * further processes are spawned. (Apple Silicon has no discrete VRAM; the
+ * frontend budgets against unified system memory instead.)
+ */
+export declare function createSystemStatsSampler(): () => SystemStatsSnapshot;
+/**
+ * Whether this server should broadcast `system_stats` at all.
+ *
+ * A server that enforces auth is shared: its CPU and RAM belong to a container
+ * the user has no stake in, so the editor readout would describe the wrong
+ * machine and report the host's capacity to everyone connected. A local install
+ * is the machine the numbers describe, so it keeps sending them.
+ *
+ * `NODETOOL_SYSTEM_STATS=1|0` overrides both ways — `1` for a local server
+ * pointed at a hosted Supabase project, `0` to turn the broadcast off outright.
+ * `NODE_ENV` is deliberately not consulted: the desktop backend and the Docker
+ * image both set it to "production" while serving one user locally, which is
+ * how this broadcast went blank in every packaged build once before.
+ */
+export declare function systemStatsBroadcastEnabled(env?: Record<string, string | undefined>): boolean;
+export {};
+//# sourceMappingURL=system-stats.d.ts.map

package/dist/system-stats.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"system-stats.d.ts","sourceRoot":"","sources":["../src/system-stats.ts"],"names":[],"mappings":"AAIA,KAAK,gBAAgB,GAAG;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF,KAAK,mBAAmB,GAAG;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,eAAe,EAAE,MAAM,CAAC;CACzB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;AAuB9B;;;;;;GAMG;AACH,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,gBAAgB,GAAG,IAAI,CA+B9E;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,wBAAwB,IAAI,MAAM,mBAAmB,CA2CpE;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,2BAA2B,CACzC,GAAG,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAe,GACpD,OAAO,CAST"}

package/dist/system-stats.js

@@ -0,0 +1,134 @@
+import os from "node:os";
+import { execFile } from "node:child_process";
+import { isAuthEnforced } from "@nodetool-ai/config";
+const BYTES_PER_GB = 1024 ** 3;
+const MIB_PER_GB = 1024;
+function snapshotCpu() {
+    let total = 0;
+    let idle = 0;
+    for (const cpu of os.cpus()) {
+        const t = cpu.times;
+        total += t.user + t.nice + t.sys + t.idle + t.irq;
+        idle += t.idle;
+    }
+    return { total, idle };
+}
+const NVIDIA_SMI_ARGS = [
+    "--query-gpu=utilization.gpu,memory.total,memory.used",
+    "--format=csv,noheader,nounits"
+];
+/**
+ * Parse `nvidia-smi --query-gpu=utilization.gpu,memory.total,memory.used
+ * --format=csv,noheader,nounits` output (one CSV line per GPU; utilization in
+ * %, memory in MiB). With several GPUs, report the one with the most memory —
+ * that's where models land, and a summed figure would suggest budgets no
+ * single card has. Returns null when no line parses to a usable total.
+ */
+export function parseNvidiaSmiGpuStats(stdout) {
+    let best = null;
+    for (const line of stdout.split("\n")) {
+        const parts = line.split(",").map((part) => part.trim());
+        if (parts.length < 3) {
+            continue;
+        }
+        // Unsupported fields print "[N/A]"; Number() turns those into NaN.
+        const util = Number(parts[0]);
+        const totalMib = Number(parts[1]);
+        const usedMib = Number(parts[2]);
+        if (!Number.isFinite(totalMib) || totalMib <= 0) {
+            continue;
+        }
+        if (!best || totalMib > best.totalMib) {
+            best = {
+                util: Number.isFinite(util) ? Math.max(0, Math.min(100, util)) : 0,
+                totalMib,
+                usedMib: Number.isFinite(usedMib) ? Math.max(0, usedMib) : 0
+            };
+        }
+    }
+    if (!best) {
+        return null;
+    }
+    return {
+        gpu_percent: best.util,
+        vram_total_gb: best.totalMib / MIB_PER_GB,
+        vram_used_gb: best.usedMib / MIB_PER_GB,
+        vram_percent: Math.max(0, Math.min(100, (best.usedMib / best.totalMib) * 100))
+    };
+}
+/**
+ * Build a stateful sampler that returns whole-system stats on each call.
+ * CPU% is computed from the delta of `os.cpus()` times since the last call,
+ * so the first call after construction returns ~0% (no delta yet).
+ *
+ * GPU stats come from an async `nvidia-smi` probe kicked off by each call and
+ * read from a cache, so the sampler itself stays synchronous; the first call
+ * carries no GPU fields, later ones do. A machine without a working
+ * `nvidia-smi` is marked unavailable after the first failed probe so no
+ * further processes are spawned. (Apple Silicon has no discrete VRAM; the
+ * frontend budgets against unified system memory instead.)
+ */
+export function createSystemStatsSampler() {
+    let previous = snapshotCpu();
+    let gpu = null;
+    let gpuProbe = "idle";
+    const probeGpu = () => {
+        gpuProbe = "probing";
+        execFile("nvidia-smi", NVIDIA_SMI_ARGS, { timeout: 3000 }, (error, stdout) => {
+            if (error) {
+                gpuProbe = "unavailable";
+                gpu = null;
+                return;
+            }
+            gpu = parseNvidiaSmiGpuStats(stdout);
+            gpuProbe = gpu ? "idle" : "unavailable";
+        });
+    };
+    return () => {
+        if (gpuProbe === "idle") {
+            probeGpu();
+        }
+        const current = snapshotCpu();
+        const totalDelta = current.total - previous.total;
+        const idleDelta = current.idle - previous.idle;
+        const cpuPercent = totalDelta > 0 ? (1 - idleDelta / totalDelta) * 100 : 0;
+        previous = current;
+        const totalMem = os.totalmem();
+        const freeMem = os.freemem();
+        const usedMem = totalMem - freeMem;
+        return {
+            cpu_percent: Math.max(0, Math.min(100, cpuPercent)),
+            memory_percent: (usedMem / totalMem) * 100,
+            memory_used: usedMem,
+            memory_total: totalMem,
+            memory_used_gb: usedMem / BYTES_PER_GB,
+            memory_total_gb: totalMem / BYTES_PER_GB,
+            ...(gpu ?? {})
+        };
+    };
+}
+/**
+ * Whether this server should broadcast `system_stats` at all.
+ *
+ * A server that enforces auth is shared: its CPU and RAM belong to a container
+ * the user has no stake in, so the editor readout would describe the wrong
+ * machine and report the host's capacity to everyone connected. A local install
+ * is the machine the numbers describe, so it keeps sending them.
+ *
+ * `NODETOOL_SYSTEM_STATS=1|0` overrides both ways — `1` for a local server
+ * pointed at a hosted Supabase project, `0` to turn the broadcast off outright.
+ * `NODE_ENV` is deliberately not consulted: the desktop backend and the Docker
+ * image both set it to "production" while serving one user locally, which is
+ * how this broadcast went blank in every packaged build once before.
+ */
+export function systemStatsBroadcastEnabled(env = process.env) {
+    const override = env["NODETOOL_SYSTEM_STATS"]?.trim();
+    if (override === "1" || override === "true") {
+        return true;
+    }
+    if (override === "0" || override === "false") {
+        return false;
+    }
+    return !isAuthEnforced(env);
+}
+//# sourceMappingURL=system-stats.js.map

package/dist/system-stats.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"system-stats.js","sourceRoot":"","sources":["../src/system-stats.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAoBrD,MAAM,YAAY,GAAG,IAAI,IAAI,CAAC,CAAC;AAC/B,MAAM,UAAU,GAAG,IAAI,CAAC;AAExB,SAAS,WAAW;IAClB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,KAAK,MAAM,GAAG,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;QAC5B,MAAM,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC;QACpB,KAAK,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC;QAClD,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC;IACjB,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AACzB,CAAC;AAED,MAAM,eAAe,GAAG;IACtB,sDAAsD;IACtD,+BAA+B;CAChC,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAc;IACnD,IAAI,IAAI,GAA+D,IAAI,CAAC;IAC5E,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACzD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,SAAS;QACX,CAAC;QACD,mEAAmE;QACnE,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACjC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;YAChD,SAAS;QACX,CAAC;QACD,IAAI,CAAC,IAAI,IAAI,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtC,IAAI,GAAG;gBACL,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClE,QAAQ;gBACR,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;aAC7D,CAAC;QACJ,CAAC;IACH,CAAC;IACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO;QACL,WAAW,EAAE,IAAI,CAAC,IAAI;QACtB,aAAa,EAAE,IAAI,CAAC,QAAQ,GAAG,UAAU;QACzC,YAAY,EAAE,IAAI,CAAC,OAAO,GAAG,UAAU;QACvC,YAAY,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,GAAG,CAAC,CAAC;KAC/E,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,wBAAwB;IACtC,IAAI,QAAQ,GAAG,WAAW,EAAE,CAAC;IAC7B,IAAI,GAAG,GAA4B,IAAI,CAAC;IACxC,IAAI,QAAQ,GAAuC,MAAM,CAAC;IAE1D,MAAM,QAAQ,GAAG,GAAG,EAAE;QACpB,QAAQ,GAAG,SAAS,CAAC;QACrB,QAAQ,CAAC,YAAY,EAAE,eAAe,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;YAC3E,IAAI,KAAK,EAAE,CAAC;gBACV,QAAQ,GAAG,aAAa,CAAC;gBACzB,GAAG,GAAG,IAAI,CAAC;gBACX,OAAO;YACT,CAAC;YACD,GAAG,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;YACrC,QAAQ,GAAG,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,OAAO,GAAG,EAAE;QACV,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;YACxB,QAAQ,EAAE,CAAC;QACb,CAAC;QACD,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;QAC9B,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAClD,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;QAC/C,MAAM,UAAU,GACd,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,GAAG,UAAU,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1D,QAAQ,GAAG,OAAO,CAAC;QAEnB,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG,QAAQ,GAAG,OAAO,CAAC;QAEnC,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;YACnD,cAAc,EAAE,CAAC,OAAO,GAAG,QAAQ,CAAC,GAAG,GAAG;YAC1C,WAAW,EAAE,OAAO;YACpB,YAAY,EAAE,QAAQ;YACtB,cAAc,EAAE,OAAO,GAAG,YAAY;YACtC,eAAe,EAAE,QAAQ,GAAG,YAAY;YACxC,GAAG,CAAC,GAAG,IAAI,EAAE,CAAC;SACf,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,2BAA2B,CACzC,GAAG,GAAuC,OAAO,CAAC,GAAG;IAErD,MAAM,QAAQ,GAAG,GAAG,CAAC,uBAAuB,CAAC,EAAE,IAAI,EAAE,CAAC;IACtD,IAAI,QAAQ,KAAK,GAAG,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;QAC5C,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,QAAQ,KAAK,GAAG,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QAC7C,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;AAC9B,CAAC"}

package/dist/test-ui-server.d.ts

@@ -1,9 +1,55 @@
 import { type IncomingMessage, type ServerResponse } from "node:http";
+import { NodeRegistry } from "@nodetool-ai/node-sdk";
+import { WebSocketClientSession, type WebSocketClientSessionOptions } from "./websocket-client-session.js";
 import { type HttpApiOptions } from "./http-api.js";
 export interface TestUiServerOptions extends HttpApiOptions {
     port?: number;
     host?: string;
     examplesDir?: string;
+    /**
+     * Called after built-in nodes are registered, before the server starts.
+     * Lets callers register extra node classes / metadata (e.g. an e2e harness
+     * registering test-only node types) on the same registry the runner uses.
+     */
+    configureRegistry?: (registry: NodeRegistry) => void;
+    /**
+     * Override LLM provider resolution. Defaults to a deterministic
+     * ScriptedProvider so agent/LLM workflows run without API keys. An e2e
+     * harness can pass a resolver that returns real providers when keys exist.
+     */
+    resolveProvider?: WebSocketClientSessionOptions["resolveProvider"];
+    /**
+     * Override executor resolution for unknown node types. Receives the registry
+     * so the default (registry.resolve) can be reused for known types.
+     */
+    resolveExecutor?: WebSocketClientSessionOptions["resolveExecutor"];
+    /**
+     * When true, unknown node types resolve to a permissive passthrough instead
+     * of failing graph hydration/execution. Lets the harness run CLI fixtures
+     * that use test-only node types such as `test.Input`.
+     */
+    passthroughUnknownNodes?: boolean;
+    /**
+     * Extends node-TYPE resolution (property/output shape used during graph
+     * hydration — distinct from `resolveExecutor`, which only resolves how a
+     * node EXECUTES) for a type the base registry can't resolve. Consulted
+     * after the registry and before `passthroughUnknownNodes`'s permissive
+     * fallback. Needed because `Graph.loadFromDict` (node-sdk) silently DROPS a
+     * node (and its edges) whose type this resolver can't answer for at all —
+     * so a caller pairing a custom `resolveExecutor` with a node type the
+     * registry doesn't know (e.g. the reliability harness's Python-bridge-fake
+     * node, task D3) must also answer here, or the node never reaches
+     * `resolveExecutor` in the first place.
+     */
+    resolveUnknownNodeType?: (nodeType: string) => Promise<import("@nodetool-ai/kernel").ResolvedNodeType | null>;
+    /**
+     * Called with each per-connection `WebSocketClientSession` right after it is
+     * constructed. The runner is otherwise private to the upgrade handler; the
+     * reliability harness needs it to read `jobs.slotCounters` for leak accounting.
+     */
+    onRunnerCreated?: (runner: WebSocketClientSession) => void;
+    /** Reset a hermetic test server to its initial fixture state. */
+    resetDatabase?: () => Promise<void>;
 }
 export declare function createTestUiServer(options?: TestUiServerOptions): {
     server: import("http").Server<typeof IncomingMessage, typeof ServerResponse>;

package/dist/test-ui-server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"test-ui-server.d.ts","sourceRoot":"","sources":["../src/test-ui-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,eAAe,EACpB,KAAK,cAAc,EACpB,MAAM,WAAW,CAAC;AAenB,OAAO,EAAyB,KAAK,cAAc,EAAE,MAAM,eAAe,CAAC;AAk2B3E,MAAM,WAAW,mBAAoB,SAAQ,cAAc;IACzD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AA8QD,wBAAgB,kBAAkB,CAAC,OAAO,GAAE,mBAAwB;;;;;;;;;;EA2JnE"}
+{"version":3,"file":"test-ui-server.d.ts","sourceRoot":"","sources":["../src/test-ui-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,eAAe,EACpB,KAAK,cAAc,EACpB,MAAM,WAAW,CAAC;AAcnB,OAAO,EACL,YAAY,EAEb,MAAM,uBAAuB,CAAC;AAK/B,OAAO,EACL,sBAAsB,EAEtB,KAAK,6BAA6B,EACnC,MAAM,+BAA+B,CAAC;AAIvC,OAAO,EAAyB,KAAK,cAAc,EAAE,MAAM,eAAe,CAAC;AA45B3E,MAAM,WAAW,mBAAoB,SAAQ,cAAc;IACzD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;OAIG;IACH,iBAAiB,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,KAAK,IAAI,CAAC;IACrD;;;;OAIG;IACH,eAAe,CAAC,EAAE,6BAA6B,CAAC,iBAAiB,CAAC,CAAC;IACnE;;;OAGG;IACH,eAAe,CAAC,EAAE,6BAA6B,CAAC,iBAAiB,CAAC,CAAC;IACnE;;;;OAIG;IACH,uBAAuB,CAAC,EAAE,OAAO,CAAC;IAClC;;;;;;;;;;;OAWG;IACH,sBAAsB,CAAC,EAAE,CACvB,QAAQ,EAAE,MAAM,KACb,OAAO,CAAC,OAAO,qBAAqB,EAAE,gBAAgB,GAAG,IAAI,CAAC,CAAC;IACpE;;;;OAIG;IACH,eAAe,CAAC,EAAE,CAAC,MAAM,EAAE,sBAAsB,KAAK,IAAI,CAAC;IAC3D,iEAAiE;IACjE,aAAa,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CACrC;AA8QD,wBAAgB,kBAAkB,CAAC,OAAO,GAAE,mBAAwB;;;;;;;;;;EA4SnE"}

package/dist/test-ui-server.js

@@ -1,18 +1,34 @@
 import { createServer } from "node:http";
-import { createLogger, getDefaultDbPath } from "@nodetool-ai/config";
+import { createLogger, getDefaultDbPath, getPostgresDatabaseUrl, loadEnvironment } from "@nodetool-ai/config";
+import { createHTTPHandler } from "@trpc/server/adapters/standalone";
+import { TRPC_MAX_BATCH_SIZE } from "@nodetool-ai/protocol";
 import { existsSync, readdirSync, readFileSync } from "node:fs";
 import { spawnSync } from "node:child_process";
 import path from "node:path";
+import { fileURLToPath } from "node:url";
 import { WebSocketServer } from "ws";
 import { NodeRegistry, createGraphNodeTypeResolver } from "@nodetool-ai/node-sdk";
-import { registerBaseNodes } from "@nodetool-ai/base-nodes";
-import { registerElevenLabsNodes } from "@nodetool-ai/elevenlabs-nodes";
-import { registerTransformersJsNodes } from "@nodetool-ai/transformers-js-nodes";
-import { UnifiedWebSocketRunner } from "./unified-websocket-runner.js";
+import { registerBuiltInNodes, applyProductionNodePolicy } from "./node-registry-setup.js";
+import { WebSocketClientSession } from "./websocket-client-session.js";
+import { appRouter } from "./trpc/router.js";
+import { createContextFactory } from "./trpc/context.js";
 import { ScriptedProvider, autoScript } from "@nodetool-ai/runtime";
 import { handleNodeHttpRequest } from "./http-api.js";
-import { initDb } from "@nodetool-ai/models";
+import { createTestUiRunRoutes } from "./lib/test-ui-run-routes.js";
+import { initDb, initPostgresDb, migrateSqliteDb } from "@nodetool-ai/models";
+import { isNonEmptyString, isString } from "./lib/wire-values.js";
+loadEnvironment(path.resolve(path.dirname(fileURLToPath(import.meta.url)), "../../.."));
 const log = createLogger("nodetool.websocket.server");
+const DEMO_IMAGE_PNG = Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVQIHWP4////fwAJ+wP9KobjigAAAABJRU5ErkJggg==", "base64");
+// 1×1 solid-color PNGs served for the seeded entity assets, so entity avatars
+// and cards render as clean color swatches (scaled by object-fit) instead of
+// broken images in documentation screenshots.
+const ENTITY_IMAGE_PNGS = {
+    "entity-marta": Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAIAAACQd1PeAAAADElEQVR4nGNY5msGAALGASqvcSOBAAAAAElFTkSuQmCC", "base64"),
+    "entity-harbor": Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAIAAACQd1PeAAAADElEQVR4nGMw8YwDAAGQANyfP6qhAAAAAElFTkSuQmCC", "base64"),
+    "entity-noir": Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAIAAACQd1PeAAAADElEQVR4nGOYrVUDAAKlAULCKC3dAAAAAElFTkSuQmCC", "base64"),
+    "entity-umbrella": Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAIAAACQd1PeAAAADElEQVR4nGPYpGQCAAKSAQkGlyO/AAAAAElFTkSuQmCC", "base64")
+};
 function htmlPage() {
     return `<!doctype html>
 <html lang="en">
@@ -836,6 +852,12 @@ class WsAdapter {
             const waiter = this.waiters.shift();
             if (waiter)
                 waiter({ type: "websocket.disconnect" });
+            // If no receiver is parked (the runner is busy handling a command), queue
+            // the disconnect so the next receive() resolves it. Dropping it here left
+            // receive() waiting on a frame that could never arrive — the runner loop
+            // hung forever and its disconnect()/cleanup never ran.
+            else
+                this.queue.push({ type: "websocket.disconnect" });
         });
     }
     async accept() { }
@@ -843,6 +865,11 @@ class WsAdapter {
         const next = this.queue.shift();
         if (next)
             return next;
+        // Never park a waiter on an already-closed socket — no further 'message' or
+        // 'close' event will fire to resolve it.
+        if (this.clientState === "disconnected") {
+            return { type: "websocket.disconnect" };
+        }
         return await new Promise((resolve) => this.waiters.push(resolve));
     }
     async sendBytes(data) {
@@ -925,7 +952,7 @@ print(json.dumps(sorted(roots)))
             const roots = JSON.parse(proc.stdout.trim());
             if (!Array.isArray(roots))
                 continue;
-            return roots.filter((p) => typeof p === "string" && p.length > 0 && existsSync(p));
+            return roots.filter((p) => isNonEmptyString(p) && existsSync(p));
         }
         catch {
             // try next python executable
@@ -1055,12 +1082,12 @@ function listExampleWorkflows(examplesDir) {
             const parsed = JSON.parse(raw);
             items.push({
                 id: file,
-                name: typeof parsed.name === "string"
+                name: isString(parsed.name)
                     ? parsed.name
                     : file.replace(/\.json$/i, ""),
-                description: typeof parsed.description === "string" ? parsed.description : "",
+                description: isString(parsed.description) ? parsed.description : "",
                 tags: Array.isArray(parsed.tags)
-                    ? parsed.tags.filter((t) => typeof t === "string")
+                    ? parsed.tags.filter((t) => isString(t))
                     : []
             });
         }
@@ -1084,12 +1111,12 @@ function readExampleWorkflow(examplesDir, id) {
         return null;
     return {
         id: safeId,
-        name: typeof parsed.name === "string"
+        name: isString(parsed.name)
             ? parsed.name
             : safeId.replace(/\.json$/i, ""),
-        description: typeof parsed.description === "string" ? parsed.description : "",
+        description: isString(parsed.description) ? parsed.description : "",
         tags: Array.isArray(parsed.tags)
-            ? parsed.tags.filter((t) => typeof t === "string")
+            ? parsed.tags.filter((t) => isString(t))
             : [],
         graph
     };
@@ -1104,20 +1131,124 @@ export function createTestUiServer(options = {}) {
         roots: metadataRoots,
         maxDepth: options.metadataMaxDepth ?? 8
     });
-    registerBaseNodes(registry);
-    registerElevenLabsNodes(registry);
-    registerTransformersJsNodes(registry);
+    registerBuiltInNodes(registry);
+    applyProductionNodePolicy(registry);
+    options.configureRegistry?.(registry);
     const resolvedApiOptions = {
         ...options,
         metadataRoots,
-        registry,
-        // Pass the resolved examples directory so handleWorkflowExamples can serve
-        // examples directly from the filesystem without requiring Python metadata.
-        ...(examplesDir ? { examplesDir } : {})
+        registry
+    };
+    const fontsDir = path.resolve(path.dirname(fileURLToPath(import.meta.url)), "../../timeline/fonts");
+    if (!resolvedApiOptions.bundledFontsDir && existsSync(fontsDir)) {
+        resolvedApiOptions.bundledFontsDir = fontsDir;
+    }
+    // Pass the resolved examples directory so handleWorkflowExamples can serve
+    // examples directly from the filesystem without requiring Python metadata.
+    if (examplesDir) {
+        resolvedApiOptions.examplesDir = examplesDir;
+    }
+    const baseGraphNodeTypeResolver = createGraphNodeTypeResolver(registry);
+    // When `passthroughUnknownNodes` is set, unknown node types (e.g. the
+    // `test.Input` placeholder used by CLI fixtures) resolve to a permissive
+    // synthetic type so graph hydration succeeds. The kernel already treats
+    // `test.Input` / `nodetool.input.*` as external inputs and dispatches the
+    // run params by node name; declaring `name` keeps that property through
+    // hydration.
+    const graphNodeTypeResolver = options.resolveUnknownNodeType || options.passthroughUnknownNodes
+        ? {
+            resolveNodeType: async (nodeType) => {
+                const resolved = await baseGraphNodeTypeResolver.resolveNodeType(nodeType);
+                if (resolved)
+                    return resolved;
+                const extended = await options.resolveUnknownNodeType?.(nodeType);
+                if (extended)
+                    return extended;
+                if (!options.passthroughUnknownNodes)
+                    return null;
+                return {
+                    nodeType,
+                    propertyTypes: { name: "str", value: "any" },
+                    outputs: {},
+                    supportsDynamicInputs: true,
+                    descriptorDefaults: {}
+                };
+            }
+        }
+        : baseGraphNodeTypeResolver;
+    // Passthrough executor for unknown node types: echo inputs back so values
+    // flow downstream. For `test.Input` the kernel calls process({ value }) and
+    // dispatches the echoed `value` to each outgoing edge (falling back to the
+    // raw param value per handle when the source handle isn't echoed).
+    const passthroughExecutor = {
+        async process(inputs) {
+            return inputs;
+        }
     };
-    const graphNodeTypeResolver = createGraphNodeTypeResolver(registry);
+    const resolveExecutor = options.resolveExecutor ??
+        ((node) => {
+            if (options.passthroughUnknownNodes && !registry.has(node.type)) {
+                return passthroughExecutor;
+            }
+            return registry.resolve(node);
+        });
+    const trpcContextFactory = createContextFactory({
+        registry,
+        apiOptions: resolvedApiOptions,
+        pythonBridge: {},
+        getPythonBridgeReady: () => false
+    });
+    const trpcHandler = createHTTPHandler({
+        router: appRouter,
+        // Match the production server (see server.ts): the web client forces all
+        // batches to POST via httpBatchLink `methodOverride: "POST"`. Without this
+        // flag the handler rejects POST-to-query, leaving the chat/threads panels
+        // empty behind a "threads.list" error.
+        allowMethodOverride: true,
+        maxBatchSize: TRPC_MAX_BATCH_SIZE,
+        createContext: ({ req }) => {
+            // SAFETY: the test-UI server has no auth plugin, so it attaches the
+            // single-user id the context reads directly onto the request.
+            const authedReq = req;
+            authedReq.userId = "1";
+            return trpcContextFactory({ req: authedReq });
+        }
+    });
+    const runRoutes = createTestUiRunRoutes();
     const server = createServer((req, res) => {
         const url = new URL(req.url ?? "/", `http://${host}:${port}`);
+        if (runRoutes.matches(url.pathname)) {
+            void runRoutes.handle(req, res).catch((error) => {
+                log.error("UI run route failed", error);
+                if (!res.headersSent) {
+                    res.statusCode = 500;
+                }
+                res.end();
+            });
+            return;
+        }
+        if (url.pathname === "/api/test/reset") {
+            if (req.method !== "POST" || !options.resetDatabase) {
+                res.statusCode = 404;
+                res.setHeader("content-type", "application/json");
+                res.end(JSON.stringify({ detail: "Not found" }));
+                return;
+            }
+            void options
+                .resetDatabase()
+                .then(() => {
+                res.statusCode = 204;
+                res.end();
+            })
+                .catch((error) => {
+                res.statusCode = 500;
+                res.setHeader("content-type", "application/json");
+                res.end(JSON.stringify({
+                    detail: error instanceof Error ? error.message : String(error)
+                }));
+            });
+            return;
+        }
         if (url.pathname === "/" || url.pathname === "/test-ui") {
             res.statusCode = 200;
             res.setHeader("content-type", "text/html; charset=utf-8");
@@ -1157,6 +1288,32 @@ export function createTestUiServer(options = {}) {
             res.end(JSON.stringify(example));
             return;
         }
+        if (/^\/api\/storage\/(?:[^/]+\/)?(?:asset-photo[1-4](?:_thumb)?\.(?:jpg|jpeg|png)|asset-doc1_thumb\.jpg)$/i.test(url.pathname)) {
+            res.statusCode = 200;
+            res.setHeader("content-type", "image/png");
+            res.setHeader("cache-control", "no-store");
+            res.end(DEMO_IMAGE_PNG);
+            return;
+        }
+        {
+            // Entity reference images (color swatches) for the seeded entities.
+            const entityMatch = url.pathname.match(/^\/api\/storage\/(?:[^/]+\/)?(entity-[a-z]+)(?:_thumb)?\.(?:jpg|jpeg|png)$/i);
+            const entityPng = entityMatch ? ENTITY_IMAGE_PNGS[entityMatch[1]] : null;
+            if (entityPng) {
+                res.statusCode = 200;
+                res.setHeader("content-type", "image/png");
+                res.setHeader("cache-control", "no-store");
+                res.end(entityPng);
+                return;
+            }
+        }
+        if (url.pathname.startsWith("/trpc")) {
+            const originalUrl = req.url;
+            req.url = (req.url ?? "/").replace(/^\/trpc/, "") || "/";
+            void trpcHandler(req, res);
+            req.url = originalUrl;
+            return;
+        }
         if (url.pathname.startsWith("/api/") ||
             url.pathname.startsWith("/v1/") ||
             url.pathname.startsWith("/admin/")) {
@@ -1181,26 +1338,28 @@ export function createTestUiServer(options = {}) {
             ws.on("error", (error) => {
                 log.error("WebSocket client error", error);
             });
-            const runner = new UnifiedWebSocketRunner({
-                resolveExecutor: (node) => registry.resolve(node),
+            const runner = new WebSocketClientSession({
+                resolveExecutor,
                 resolveNodeType: graphNodeTypeResolver,
-                resolveProvider: async (_providerId) => new ScriptedProvider([
-                    autoScript({
-                        plan: {
-                            title: "Agent task",
-                            steps: [
-                                {
-                                    id: "s1",
-                                    instructions: "Complete the objective",
-                                    depends_on: []
-                                }
-                            ]
-                        },
-                        text: "fake agent response from server"
-                    })
-                ]),
+                resolveProvider: options.resolveProvider ??
+                    (async (_providerId) => new ScriptedProvider([
+                        autoScript({
+                            plan: {
+                                title: "Agent task",
+                                steps: [
+                                    {
+                                        id: "s1",
+                                        instructions: "Complete the objective",
+                                        depends_on: []
+                                    }
+                                ]
+                            },
+                            text: "fake agent response from server"
+                        })
+                    ])),
                 getNodeMetadata: (nodeType) => registry.getMetadata(nodeType)
             });
+            options.onRunnerCreated?.(runner);
             log.info("WebSocket client connected");
             void runner.run(new WsAdapter(ws)).catch((error) => {
                 log.error("Runner crashed", error instanceof Error ? error : new Error(String(error)));
@@ -1225,7 +1384,7 @@ export function createTestUiServer(options = {}) {
                     });
                 }
             });
-        }),
+        }).then(() => runRoutes.close()),
         info: {
             host,
             port,
@@ -1235,10 +1394,16 @@ export function createTestUiServer(options = {}) {
     };
 }
 if (import.meta.url === `file://${process.argv[1]}`) {
-    // Initialize SQLite adapter pointing at the same DB as the Python side
-    const dbPath = getDefaultDbPath();
     try {
-        initDb(dbPath);
+        const postgresDatabaseUrl = getPostgresDatabaseUrl();
+        if (postgresDatabaseUrl) {
+            await initPostgresDb(postgresDatabaseUrl);
+        }
+        else {
+            const dbPath = getDefaultDbPath();
+            await migrateSqliteDb(dbPath);
+            initDb(dbPath);
+        }
     }
     catch {
         // DB unavailable — secrets will appear unconfigured