@schlessera/brain-ui-server 0.4.0

package/dist/ws/handler.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Composition root for the WebSocket side: one process-wide host instance
+ * wired to the default SQLite session catalog and the backend registry. The
+ * actual behavior lives in the focused modules next to this file —
+ * turns/bridge/run-session/dispatch/connection — which all take the host
+ * explicitly. createApp() configures this host from its options before
+ * building the upgrade handler.
+ */
+import type { ClientMessage } from "@schlessera/brain-ui-sdk/protocol";
+import type { WSContext } from "./clients.js";
+import { WsHost, type WsHostOptions } from "./host.js";
+import { websocket } from "./connection.js";
+export { websocket };
+export { WsHost, type WsHostOptions } from "./host.js";
+export { resolveTurnTarget } from "./routing.js";
+export { validateAttachments, estimateDecodedBase64Bytes } from "./attachments.js";
+export { createSessionCatalog, type SessionCatalog } from "./session-catalog.js";
+/** The process-default host. */
+export declare const defaultWsHost: WsHost;
+/** Apply embedder options to the default host (called by createApp). */
+export declare function configureWsHost(options: WsHostOptions): void;
+/** Hono upgrade handler bound to the default host. */
+export declare const wsUpgrade: import("hono").MiddlewareHandler<any, string, {
+    outputFormat: "ws";
+}>;
+/** True while any session has a running turn. */
+export declare function isTurnActive(): boolean;
+/** Cancel every running turn (used on shutdown). */
+export declare function cancelActiveTurn(): boolean;
+/** Drive a client message against the default host (also the test seam). */
+export declare function handleClientMessage(ws: WSContext, msg: ClientMessage): Promise<void>;
+/** Test seam: clear all turn/pending state on the default host. */
+export declare function resetForTests(): void;
+//# sourceMappingURL=handler.d.ts.map

package/dist/ws/handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../../src/ws/handler.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AACvE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,KAAK,aAAa,EAAE,MAAM,WAAW,CAAC;AACvD,OAAO,EAAmB,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAG7D,OAAO,EAAE,SAAS,EAAE,CAAC;AACrB,OAAO,EAAE,MAAM,EAAE,KAAK,aAAa,EAAE,MAAM,WAAW,CAAC;AACvD,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,0BAA0B,EAAE,MAAM,kBAAkB,CAAC;AACnF,OAAO,EAAE,oBAAoB,EAAE,KAAK,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEjF,gCAAgC;AAChC,eAAO,MAAM,aAAa,QAAe,CAAC;AAE1C,wEAAwE;AACxE,wBAAgB,eAAe,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI,CAE5D;AAED,sDAAsD;AACtD,eAAO,MAAM,SAAS;;EAAiC,CAAC;AAExD,iDAAiD;AACjD,wBAAgB,YAAY,IAAI,OAAO,CAEtC;AAED,oDAAoD;AACpD,wBAAgB,gBAAgB,IAAI,OAAO,CAE1C;AAED,4EAA4E;AAC5E,wBAAsB,mBAAmB,CAAC,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CAE1F;AAED,mEAAmE;AACnE,wBAAgB,aAAa,IAAI,IAAI,CAEpC"}

package/dist/ws/handler.js

@@ -0,0 +1,33 @@
+import { WsHost } from "./host.js";
+import { createWsUpgrade, websocket } from "./connection.js";
+import { handleClientMessage as dispatchClientMessage } from "./dispatch.js";
+export { websocket };
+export { WsHost } from "./host.js";
+export { resolveTurnTarget } from "./routing.js";
+export { validateAttachments, estimateDecodedBase64Bytes } from "./attachments.js";
+export { createSessionCatalog } from "./session-catalog.js";
+/** The process-default host. */
+export const defaultWsHost = new WsHost();
+/** Apply embedder options to the default host (called by createApp). */
+export function configureWsHost(options) {
+    defaultWsHost.configure(options);
+}
+/** Hono upgrade handler bound to the default host. */
+export const wsUpgrade = createWsUpgrade(defaultWsHost);
+/** True while any session has a running turn. */
+export function isTurnActive() {
+    return defaultWsHost.coordinator.isTurnActive();
+}
+/** Cancel every running turn (used on shutdown). */
+export function cancelActiveTurn() {
+    return defaultWsHost.coordinator.cancelAll("Server shutting down");
+}
+/** Drive a client message against the default host (also the test seam). */
+export async function handleClientMessage(ws, msg) {
+    return dispatchClientMessage(defaultWsHost, ws, msg);
+}
+/** Test seam: clear all turn/pending state on the default host. */
+export function resetForTests() {
+    defaultWsHost.coordinator.reset();
+}
+//# sourceMappingURL=handler.js.map

package/dist/ws/handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.js","sourceRoot":"","sources":["../../src/ws/handler.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,MAAM,EAAsB,MAAM,WAAW,CAAC;AACvD,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC7D,OAAO,EAAE,mBAAmB,IAAI,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAE7E,OAAO,EAAE,SAAS,EAAE,CAAC;AACrB,OAAO,EAAE,MAAM,EAAsB,MAAM,WAAW,CAAC;AACvD,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,0BAA0B,EAAE,MAAM,kBAAkB,CAAC;AACnF,OAAO,EAAE,oBAAoB,EAAuB,MAAM,sBAAsB,CAAC;AAEjF,gCAAgC;AAChC,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,MAAM,EAAE,CAAC;AAE1C,wEAAwE;AACxE,MAAM,UAAU,eAAe,CAAC,OAAsB;IACpD,aAAa,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC;AAED,sDAAsD;AACtD,MAAM,CAAC,MAAM,SAAS,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;AAExD,iDAAiD;AACjD,MAAM,UAAU,YAAY;IAC1B,OAAO,aAAa,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC;AAClD,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,gBAAgB;IAC9B,OAAO,aAAa,CAAC,WAAW,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;AACrE,CAAC;AAED,4EAA4E;AAC5E,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EAAa,EAAE,GAAkB;IACzE,OAAO,qBAAqB,CAAC,aAAa,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;AACvD,CAAC;AAED,mEAAmE;AACnE,MAAM,UAAU,aAAa;IAC3B,aAAa,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;AACpC,CAAC"}

package/dist/ws/history.d.ts

@@ -0,0 +1,17 @@
+import type { SessionHistoryMessage } from "@schlessera/brain-ui-sdk/protocol";
+import { type WSContext } from "./clients.js";
+/**
+ * Send a structured history to one socket as one or more `session_history`
+ * frames. Each message is size-bounded, then batched into byte-limited chunks
+ * so no single frame exceeds the socket's per-message limit (which would drop
+ * the connection and wedge the reconnect loop). The first frame replaces the
+ * client transcript; continuation frames carry `append: true`. An empty
+ * history still emits one (empty, replacing) frame so the client settles.
+ *
+ * Every frame carries `sessionId` so a client with multiple sessions open can
+ * demux it: a history load for a background session must never replace the
+ * transcript in view (a reconnect or an out-of-order getHistory otherwise
+ * silently overwrites the wrong pane).
+ */
+export declare function sendSessionHistory(ws: WSContext, sessionId: string, messages: SessionHistoryMessage[]): void;
+//# sourceMappingURL=history.d.ts.map

package/dist/ws/history.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"history.d.ts","sourceRoot":"","sources":["../../src/ws/history.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,mCAAmC,CAAC;AAC/E,OAAO,EAAE,KAAK,SAAS,EAAU,MAAM,cAAc,CAAC;AAGtD;;;;;;;;;;;;GAYG;AACH,wBAAgB,kBAAkB,CAChC,EAAE,EAAE,SAAS,EACb,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,qBAAqB,EAAE,GAChC,IAAI,CA6BN"}

package/dist/ws/history.js

@@ -0,0 +1,45 @@
+import { sendTo } from "./clients.js";
+import { HISTORY_CHUNK_BYTES, shrinkForReplication } from "./shrink.js";
+/**
+ * Send a structured history to one socket as one or more `session_history`
+ * frames. Each message is size-bounded, then batched into byte-limited chunks
+ * so no single frame exceeds the socket's per-message limit (which would drop
+ * the connection and wedge the reconnect loop). The first frame replaces the
+ * client transcript; continuation frames carry `append: true`. An empty
+ * history still emits one (empty, replacing) frame so the client settles.
+ *
+ * Every frame carries `sessionId` so a client with multiple sessions open can
+ * demux it: a history load for a background session must never replace the
+ * transcript in view (a reconnect or an out-of-order getHistory otherwise
+ * silently overwrites the wrong pane).
+ */
+export function sendSessionHistory(ws, sessionId, messages) {
+    const bounded = messages.map((m) => shrinkForReplication(m));
+    if (bounded.length === 0) {
+        sendTo(ws, { type: "session_history", sessionId, messages: [] });
+        return;
+    }
+    let batch = [];
+    let batchBytes = 0;
+    let isFirst = true;
+    const flush = () => {
+        sendTo(ws, {
+            type: "session_history",
+            sessionId,
+            messages: batch,
+            ...(isFirst ? {} : { append: true }),
+        });
+        isFirst = false;
+        batch = [];
+        batchBytes = 0;
+    };
+    for (const message of bounded) {
+        const size = JSON.stringify(message).length;
+        if (batch.length > 0 && batchBytes + size > HISTORY_CHUNK_BYTES)
+            flush();
+        batch.push(message);
+        batchBytes += size;
+    }
+    flush();
+}
+//# sourceMappingURL=history.js.map

package/dist/ws/history.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"history.js","sourceRoot":"","sources":["../../src/ws/history.ts"],"names":[],"mappings":"AACA,OAAO,EAAkB,MAAM,EAAE,MAAM,cAAc,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAExE;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,kBAAkB,CAChC,EAAa,EACb,SAAiB,EACjB,QAAiC;IAEjC,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QACjE,OAAO;IACT,CAAC;IAED,IAAI,KAAK,GAA4B,EAAE,CAAC;IACxC,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,OAAO,GAAG,IAAI,CAAC;IACnB,MAAM,KAAK,GAAG,GAAG,EAAE;QACjB,MAAM,CAAC,EAAE,EAAE;YACT,IAAI,EAAE,iBAAiB;YACvB,SAAS;YACT,QAAQ,EAAE,KAAK;YACf,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SACrC,CAAC,CAAC;QACH,OAAO,GAAG,KAAK,CAAC;QAChB,KAAK,GAAG,EAAE,CAAC;QACX,UAAU,GAAG,CAAC,CAAC;IACjB,CAAC,CAAC;IAEF,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;QAC5C,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,GAAG,IAAI,GAAG,mBAAmB;YAAE,KAAK,EAAE,CAAC;QACzE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpB,UAAU,IAAI,IAAI,CAAC;IACrB,CAAC;IACD,KAAK,EAAE,CAAC;AACV,CAAC"}

package/dist/ws/host.d.ts

@@ -0,0 +1,40 @@
+import type { ServerMessage } from "@schlessera/brain-ui-sdk/protocol";
+import { type WSContext } from "./clients.js";
+import { TurnCoordinator } from "./turns.js";
+import { type SessionCatalog } from "./session-catalog.js";
+/** Cap on host-side follow-up queue depth per session (backends without followUp). */
+export declare const MAX_SESSION_QUEUE = 5;
+export interface WsHostOptions {
+    /** Session persistence seam; defaults to the package's SQLite catalog. */
+    catalog?: SessionCatalog;
+    /** Display name used in connection/status copy. */
+    appName?: string;
+    /** Per-turn timeout in ms (default 10 minutes). */
+    turnTimeoutMs?: number;
+    /**
+     * Deployment-time cap on concurrent RUNNING sessions. Each running turn is
+     * roughly one CLI subprocess, so this bounds memory/CPU. Read per request so
+     * the deploy-time env is honored without a restart.
+     */
+    maxConcurrentSessions?: () => number;
+}
+/**
+ * Everything one WebSocket coordinator instance owns: turn state, the session
+ * catalog, branding copy, and the frame senders. Handlers receive this host
+ * explicitly instead of reaching for module globals.
+ */
+export declare class WsHost {
+    readonly coordinator: TurnCoordinator;
+    catalog: SessionCatalog;
+    appName: string;
+    turnTimeoutMs: number;
+    maxConcurrentSessions: () => number;
+    constructor(options?: WsHostOptions);
+    /** Re-apply embedder options (createApp configures the default host). */
+    configure(options: WsHostOptions): void;
+    /** Fan a frame out to every attached client (size-bounded per frame). */
+    sendToClients(msg: ServerMessage): void;
+    /** Send a frame to one specific socket (size-bounded). */
+    sendMessage(ws: WSContext, msg: ServerMessage): void;
+}
+//# sourceMappingURL=host.d.ts.map

package/dist/ws/host.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"host.d.ts","sourceRoot":"","sources":["../../src/ws/host.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AACvE,OAAO,EAAqB,KAAK,SAAS,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAwB,KAAK,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAKjF,sFAAsF;AACtF,eAAO,MAAM,iBAAiB,IAAI,CAAC;AAEnC,MAAM,WAAW,aAAa;IAC5B,0EAA0E;IAC1E,OAAO,CAAC,EAAE,cAAc,CAAC;IACzB,mDAAmD;IACnD,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,mDAAmD;IACnD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB;;;;OAIG;IACH,qBAAqB,CAAC,EAAE,MAAM,MAAM,CAAC;CACtC;AAMD;;;;GAIG;AACH,qBAAa,MAAM;IACjB,QAAQ,CAAC,WAAW,kBAAyB;IAC7C,OAAO,EAAE,cAAc,CAAC;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,CAAC;IACtB,qBAAqB,EAAE,MAAM,MAAM,CAAC;gBAExB,OAAO,GAAE,aAAkB;IAOvC,yEAAyE;IACzE,SAAS,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI;IASvC,yEAAyE;IACzE,aAAa,CAAC,GAAG,EAAE,aAAa,GAAG,IAAI;IAIvC,0DAA0D;IAC1D,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,aAAa,GAAG,IAAI;CAGrD"}

package/dist/ws/host.js

@@ -0,0 +1,49 @@
+import { broadcast, sendTo } from "./clients.js";
+import { TurnCoordinator } from "./turns.js";
+import { createSessionCatalog } from "./session-catalog.js";
+/** Host-side turn timeout. The backend no longer times out — the host owns it. */
+const DEFAULT_TIMEOUT_MS = 10 * 60 * 1000; // 10 minutes
+/** Cap on host-side follow-up queue depth per session (backends without followUp). */
+export const MAX_SESSION_QUEUE = 5;
+function envMaxConcurrentSessions() {
+    return Math.max(1, Number(process.env.MAX_CONCURRENT_SESSIONS) || 3);
+}
+/**
+ * Everything one WebSocket coordinator instance owns: turn state, the session
+ * catalog, branding copy, and the frame senders. Handlers receive this host
+ * explicitly instead of reaching for module globals.
+ */
+export class WsHost {
+    coordinator = new TurnCoordinator();
+    catalog;
+    appName;
+    turnTimeoutMs;
+    maxConcurrentSessions;
+    constructor(options = {}) {
+        this.catalog = options.catalog ?? createSessionCatalog();
+        this.appName = options.appName ?? "Brain UI";
+        this.turnTimeoutMs = options.turnTimeoutMs ?? DEFAULT_TIMEOUT_MS;
+        this.maxConcurrentSessions = options.maxConcurrentSessions ?? envMaxConcurrentSessions;
+    }
+    /** Re-apply embedder options (createApp configures the default host). */
+    configure(options) {
+        if (options.catalog)
+            this.catalog = options.catalog;
+        if (options.appName)
+            this.appName = options.appName;
+        if (options.turnTimeoutMs)
+            this.turnTimeoutMs = options.turnTimeoutMs;
+        if (options.maxConcurrentSessions) {
+            this.maxConcurrentSessions = options.maxConcurrentSessions;
+        }
+    }
+    /** Fan a frame out to every attached client (size-bounded per frame). */
+    sendToClients(msg) {
+        broadcast(msg);
+    }
+    /** Send a frame to one specific socket (size-bounded). */
+    sendMessage(ws, msg) {
+        sendTo(ws, msg);
+    }
+}
+//# sourceMappingURL=host.js.map

package/dist/ws/host.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"host.js","sourceRoot":"","sources":["../../src/ws/host.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,EAAkB,MAAM,cAAc,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,oBAAoB,EAAuB,MAAM,sBAAsB,CAAC;AAEjF,kFAAkF;AAClF,MAAM,kBAAkB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;AAExD,sFAAsF;AACtF,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC;AAiBnC,SAAS,wBAAwB;IAC/B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,CAAC;AACvE,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,MAAM;IACR,WAAW,GAAG,IAAI,eAAe,EAAE,CAAC;IAC7C,OAAO,CAAiB;IACxB,OAAO,CAAS;IAChB,aAAa,CAAS;IACtB,qBAAqB,CAAe;IAEpC,YAAY,UAAyB,EAAE;QACrC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,oBAAoB,EAAE,CAAC;QACzD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,UAAU,CAAC;QAC7C,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,kBAAkB,CAAC;QACjE,IAAI,CAAC,qBAAqB,GAAG,OAAO,CAAC,qBAAqB,IAAI,wBAAwB,CAAC;IACzF,CAAC;IAED,yEAAyE;IACzE,SAAS,CAAC,OAAsB;QAC9B,IAAI,OAAO,CAAC,OAAO;YAAE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QACpD,IAAI,OAAO,CAAC,OAAO;YAAE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QACpD,IAAI,OAAO,CAAC,aAAa;YAAE,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;QACtE,IAAI,OAAO,CAAC,qBAAqB,EAAE,CAAC;YAClC,IAAI,CAAC,qBAAqB,GAAG,OAAO,CAAC,qBAAqB,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,yEAAyE;IACzE,aAAa,CAAC,GAAkB;QAC9B,SAAS,CAAC,GAAG,CAAC,CAAC;IACjB,CAAC;IAED,0DAA0D;IAC1D,WAAW,CAAC,EAAa,EAAE,GAAkB;QAC3C,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;IAClB,CAAC;CACF"}

package/dist/ws/routing.d.ts

@@ -0,0 +1,12 @@
+import type { AgentBackend } from "@schlessera/brain-ui-sdk/server";
+import type { SessionCatalog } from "./session-catalog.js";
+/**
+ * Resolve one turn slot's pinned backend and profile. Resumed sessions route by
+ * their stored backend id; new sessions route by the requested profile owner.
+ */
+export declare function resolveTurnTarget(catalog: SessionCatalog, sessionId: string | undefined, requested: string | undefined): Promise<{
+    profileId?: string;
+    backend: AgentBackend;
+    droppedPin?: string;
+}>;
+//# sourceMappingURL=routing.d.ts.map

package/dist/ws/routing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"routing.d.ts","sourceRoot":"","sources":["../../src/ws/routing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAOpE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAE3D;;;GAGG;AACH,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,cAAc,EACvB,SAAS,EAAE,MAAM,GAAG,SAAS,EAC7B,SAAS,EAAE,MAAM,GAAG,SAAS,GAC5B,OAAO,CAAC;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,YAAY,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAyB7E"}

package/dist/ws/routing.js

@@ -0,0 +1,31 @@
+import { getBackendForProfile, getBackendForSession, getDefaultBackend, listAllProviders, } from "../agent/backend.js";
+/**
+ * Resolve one turn slot's pinned backend and profile. Resumed sessions route by
+ * their stored backend id; new sessions route by the requested profile owner.
+ */
+export async function resolveTurnTarget(catalog, sessionId, requested) {
+    if (sessionId) {
+        const backend = await getBackendForSession(catalog.getStoredBackendId(sessionId));
+        const stored = catalog.getStoredProviderId(sessionId);
+        const available = await backend.listProfiles();
+        if (stored && available.some((profile) => profile.id === stored)) {
+            return { backend, profileId: stored };
+        }
+        // A stored pin that no longer resolves: fall back to the default profile
+        // but report it (droppedPin) so the caller can surface the switch instead
+        // of silently changing model/endpoint/billing with nothing in the logs.
+        return { backend, ...(stored ? { droppedPin: stored } : {}) };
+    }
+    if (requested) {
+        const available = await listAllProviders();
+        if (available.some((profile) => profile.id === requested)) {
+            const backend = await getBackendForProfile(requested);
+            if (backend)
+                return { backend, profileId: requested };
+        }
+    }
+    const backend = await getDefaultBackend();
+    const profileId = (await backend.listProfiles())[0]?.id;
+    return { backend, ...(profileId ? { profileId } : {}) };
+}
+//# sourceMappingURL=routing.js.map

package/dist/ws/routing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"routing.js","sourceRoot":"","sources":["../../src/ws/routing.ts"],"names":[],"mappings":"AACA,OAAO,EACL,oBAAoB,EACpB,oBAAoB,EACpB,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,qBAAqB,CAAC;AAG7B;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAuB,EACvB,SAA6B,EAC7B,SAA6B;IAE7B,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,OAAO,GAAG,MAAM,oBAAoB,CAAC,OAAO,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC;QAClF,MAAM,MAAM,GAAG,OAAO,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACtD,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,YAAY,EAAE,CAAC;QAC/C,IAAI,MAAM,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,MAAM,CAAC,EAAE,CAAC;YACjE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;QACxC,CAAC;QACD,yEAAyE;QACzE,0EAA0E;QAC1E,wEAAwE;QACxE,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;IAChE,CAAC;IAED,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,SAAS,GAAG,MAAM,gBAAgB,EAAE,CAAC;QAC3C,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,SAAS,CAAC,EAAE,CAAC;YAC1D,MAAM,OAAO,GAAG,MAAM,oBAAoB,CAAC,SAAS,CAAC,CAAC;YACtD,IAAI,OAAO;gBAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;QACxD,CAAC;IACH,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,iBAAiB,EAAE,CAAC;IAC1C,MAAM,SAAS,GAAG,CAAC,MAAM,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IACxD,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;AAC1D,CAAC"}

package/dist/ws/run-session.d.ts

@@ -0,0 +1,17 @@
+import type { ChatImageAttachment } from "@schlessera/brain-ui-sdk/protocol";
+import type { WSContext } from "./clients.js";
+import { type WsHost } from "./host.js";
+/**
+ * Run one session slot: the initial turn, then any queued follow-up turns in
+ * order. The host owns per-turn cancellation (AbortController + timeout). Never
+ * throws — runtime failures are emitted as error frames.
+ */
+export declare function runSession(host: WsHost, initial: {
+    text: string;
+    sessionId?: string;
+    attachments: ChatImageAttachment[];
+    providerId?: string;
+}): Promise<void>;
+/** Dispatch a chat_message: follow-up to a running session, or a new session. */
+export declare function handleChatMessage(host: WsHost, ws: WSContext, text: string, sessionId: string | undefined, attachments: ChatImageAttachment[], requestedProviderId: string | undefined): Promise<void>;
+//# sourceMappingURL=run-session.d.ts.map

package/dist/ws/run-session.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-session.d.ts","sourceRoot":"","sources":["../../src/ws/run-session.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAC7E,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAK9C,OAAO,EAAqB,KAAK,MAAM,EAAE,MAAM,WAAW,CAAC;AAE3D;;;;GAIG;AACH,wBAAsB,UAAU,CAC9B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE;IACP,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,mBAAmB,EAAE,CAAC;IACnC,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GACA,OAAO,CAAC,IAAI,CAAC,CAwGf;AAED,iFAAiF;AACjF,wBAAsB,iBAAiB,CACrC,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,SAAS,EACb,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,MAAM,GAAG,SAAS,EAC7B,WAAW,EAAE,mBAAmB,EAAE,EAClC,mBAAmB,EAAE,MAAM,GAAG,SAAS,GACtC,OAAO,CAAC,IAAI,CAAC,CAwDf"}

package/dist/ws/run-session.js

@@ -0,0 +1,160 @@
+import { withSessionId, withTurnScope } from "./frames.js";
+import { makeBridge, emitTurnError } from "./bridge.js";
+import { resolveTurnTarget } from "./routing.js";
+import { MAX_SESSION_QUEUE } from "./host.js";
+/**
+ * Run one session slot: the initial turn, then any queued follow-up turns in
+ * order. The host owns per-turn cancellation (AbortController + timeout). Never
+ * throws — runtime failures are emitted as error frames.
+ */
+export async function runSession(host, initial) {
+    const { coordinator } = host;
+    coordinator.startingSessions += 1;
+    let target;
+    try {
+        target = await resolveTurnTarget(host.catalog, initial.sessionId, initial.providerId);
+    }
+    catch (err) {
+        host.sendToClients({
+            type: "error",
+            code: "BACKEND_ERROR",
+            message: err instanceof Error ? err.message : String(err),
+            ...(initial.sessionId ? { sessionId: initial.sessionId } : {}),
+        });
+        return;
+    }
+    finally {
+        coordinator.startingSessions = Math.max(0, coordinator.startingSessions - 1);
+    }
+    const { backend, profileId: initialProfileId } = target;
+    if (target.droppedPin) {
+        console.warn(`[agent] session ${initial.sessionId} pinned profile "${target.droppedPin}" ` +
+            `is unavailable; running on the default profile.`);
+        host.sendToClients({
+            type: "status",
+            status: "thinking",
+            detail: `Pinned model "${target.droppedPin}" is unavailable — running on the default.`,
+            ...(initial.sessionId ? { sessionId: initial.sessionId } : {}),
+        });
+    }
+    const turn = {
+        sessionId: initial.sessionId ?? null,
+        turnId: crypto.randomUUID(),
+        providerId: initialProfileId ?? null,
+        backend,
+        abortController: new AbortController(),
+        timeoutHandle: setTimeout(() => { }, 0),
+        queue: [],
+        cancelled: false,
+    };
+    clearTimeout(turn.timeoutHandle);
+    coordinator.running.add(turn);
+    if (turn.sessionId)
+        coordinator.bySession.set(turn.sessionId, turn);
+    let profileId = initialProfileId;
+    let resumeId = initial.sessionId;
+    let next = {
+        text: initial.text,
+        attachments: initial.attachments,
+    };
+    try {
+        while (next && !turn.cancelled) {
+            const { text, attachments } = next;
+            next = null;
+            const abortController = new AbortController();
+            turn.abortController = abortController;
+            const timeoutHandle = setTimeout(() => {
+                console.log("[agent] Turn timed out after", host.turnTimeoutMs, "ms");
+                abortController.abort();
+                // Reject any pending interactive request for this turn too. A bridge
+                // that awaits askUser/approval without racing the abort signal would
+                // otherwise leave startTurn parked forever after abort(), leaking a
+                // concurrency slot until restart. Draining unblocks the unwind.
+                coordinator.drainPendingForTurn(turn, "Turn timed out");
+            }, host.turnTimeoutMs);
+            turn.timeoutHandle = timeoutHandle;
+            const bridge = makeBridge(host, turn, text, backend.id);
+            try {
+                await backend.startTurn({
+                    prompt: text,
+                    attachments,
+                    sessionId: resumeId,
+                    profileId,
+                    signal: abortController.signal,
+                    bridge,
+                });
+            }
+            catch (err) {
+                emitTurnError(host, turn, err);
+            }
+            finally {
+                clearTimeout(timeoutHandle);
+            }
+            // Subsequent (queued) turns resume the now-known session and keep its
+            // pin. turn.providerId is the resolved profile (set from initial routing
+            // and re-pinned by session_info); carrying it forward stops a queued
+            // follow-up from silently dropping onto the default profile/model/billing.
+            resumeId = turn.sessionId ?? resumeId;
+            profileId = turn.providerId ?? profileId;
+            if (!turn.cancelled && turn.queue.length > 0) {
+                next = turn.queue.shift();
+                // A queued follow-up is its own turn — give it a fresh identity.
+                turn.turnId = crypto.randomUUID();
+            }
+        }
+    }
+    finally {
+        if (turn.sessionId)
+            coordinator.bySession.delete(turn.sessionId);
+        coordinator.running.delete(turn);
+        coordinator.drainPendingForTurn(turn, "Session ended");
+    }
+}
+/** Dispatch a chat_message: follow-up to a running session, or a new session. */
+export async function handleChatMessage(host, ws, text, sessionId, attachments, requestedProviderId) {
+    const { coordinator } = host;
+    const runningTurn = sessionId ? coordinator.bySession.get(sessionId) : undefined;
+    if (runningTurn && sessionId) {
+        const backend = runningTurn.backend;
+        // Message to a session whose turn is running = a follow-up.
+        if (backend.capabilities.followUp && backend.followUp) {
+            // Inject into the running turn; frames flow through its bridge.
+            backend.followUp({ sessionId, prompt: text, attachments }).catch((err) => {
+                host.sendToClients(withTurnScope({ type: "error", code: "FOLLOWUP_FAILED", message: err instanceof Error ? err.message : String(err) }, runningTurn));
+            });
+        }
+        else if (runningTurn.queue.length >= MAX_SESSION_QUEUE) {
+            host.sendMessage(ws, 
+            // Session-scoped only: the rejected message would have become a
+            // FUTURE turn in this slot, not the one currently running.
+            withSessionId({ type: "error", code: "SESSION_QUEUE_FULL", message: `This session's queue is full (max ${MAX_SESSION_QUEUE}). Wait for it to catch up.` }, sessionId));
+        }
+        else {
+            // Queue it as the session's next turn; report queued immediately.
+            runningTurn.queue.push({ text, attachments });
+            host.sendToClients(withSessionId({ type: "status", status: "queued" }, sessionId));
+        }
+        return;
+    }
+    // New session (or resume of an idle one) — gated by the concurrency cap.
+    const cap = host.maxConcurrentSessions();
+    if (coordinator.running.size + coordinator.startingSessions >= cap) {
+        host.sendMessage(ws, {
+            type: "error",
+            code: "SESSION_LIMIT",
+            message: `Too many concurrent sessions (max ${cap}). Wait for one to finish.`,
+            ...(sessionId ? { sessionId } : {}),
+        });
+        return;
+    }
+    if (sessionId) {
+        host.sendToClients(withSessionId({ type: "status", status: "thinking" }, sessionId));
+    }
+    void runSession(host, {
+        text,
+        sessionId,
+        attachments,
+        providerId: requestedProviderId,
+    });
+}
+//# sourceMappingURL=run-session.js.map

package/dist/ws/run-session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-session.js","sourceRoot":"","sources":["../../src/ws/run-session.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAEjD,OAAO,EAAE,iBAAiB,EAAe,MAAM,WAAW,CAAC;AAE3D;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,IAAY,EACZ,OAKC;IAED,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;IAC7B,WAAW,CAAC,gBAAgB,IAAI,CAAC,CAAC;IAClC,IAAI,MAAqD,CAAC;IAC1D,IAAI,CAAC;QACH,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IACxF,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,aAAa,CAAC;YACjB,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,eAAe;YACrB,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;YACzD,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC/D,CAAC,CAAC;QACH,OAAO;IACT,CAAC;YAAS,CAAC;QACT,WAAW,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,CAAC,gBAAgB,GAAG,CAAC,CAAC,CAAC;IAC/E,CAAC;IAED,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,GAAG,MAAM,CAAC;IACxD,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QACtB,OAAO,CAAC,IAAI,CACV,mBAAmB,OAAO,CAAC,SAAS,oBAAoB,MAAM,CAAC,UAAU,IAAI;YAC3E,iDAAiD,CACpD,CAAC;QACF,IAAI,CAAC,aAAa,CAAC;YACjB,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,UAAU;YAClB,MAAM,EAAE,iBAAiB,MAAM,CAAC,UAAU,4CAA4C;YACtF,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IACD,MAAM,IAAI,GAAgB;QACxB,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;QACpC,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE;QAC3B,UAAU,EAAE,gBAAgB,IAAI,IAAI;QACpC,OAAO;QACP,eAAe,EAAE,IAAI,eAAe,EAAE;QACtC,aAAa,EAAE,UAAU,CAAC,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,CAAC;QACtC,KAAK,EAAE,EAAE;QACT,SAAS,EAAE,KAAK;KACjB,CAAC;IACF,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACjC,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC9B,IAAI,IAAI,CAAC,SAAS;QAAE,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAEpE,IAAI,SAAS,GAAG,gBAAgB,CAAC;IACjC,IAAI,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC;IACjC,IAAI,IAAI,GAA0B;QAChC,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,WAAW,EAAE,OAAO,CAAC,WAAW;KACjC,CAAC;IAEF,IAAI,CAAC;QACH,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YAC/B,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;YACnC,IAAI,GAAG,IAAI,CAAC;YAEZ,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;YAC9C,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;YACvC,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;gBACpC,OAAO,CAAC,GAAG,CAAC,8BAA8B,EAAE,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;gBACtE,eAAe,CAAC,KAAK,EAAE,CAAC;gBACxB,qEAAqE;gBACrE,qEAAqE;gBACrE,oEAAoE;gBACpE,gEAAgE;gBAChE,WAAW,CAAC,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;YAC1D,CAAC,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;YACvB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;YAEnC,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;YACxD,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,SAAS,CAAC;oBACtB,MAAM,EAAE,IAAI;oBACZ,WAAW;oBACX,SAAS,EAAE,QAAQ;oBACnB,SAAS;oBACT,MAAM,EAAE,eAAe,CAAC,MAAM;oBAC9B,MAAM;iBACP,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,aAAa,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACjC,CAAC;oBAAS,CAAC;gBACT,YAAY,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC;YAED,sEAAsE;YACtE,yEAAyE;YACzE,qEAAqE;YACrE,2EAA2E;YAC3E,QAAQ,GAAG,IAAI,CAAC,SAAS,IAAI,QAAQ,CAAC;YACtC,SAAS,GAAG,IAAI,CAAC,UAAU,IAAI,SAAS,CAAC;YAEzC,IAAI,CAAC,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7C,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAG,CAAC;gBAC3B,iEAAiE;gBACjE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;YACpC,CAAC;QACH,CAAC;IACH,CAAC;YAAS,CAAC;QACT,IAAI,IAAI,CAAC,SAAS;YAAE,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACjC,WAAW,CAAC,mBAAmB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;IACzD,CAAC;AACH,CAAC;AAED,iFAAiF;AACjF,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,IAAY,EACZ,EAAa,EACb,IAAY,EACZ,SAA6B,EAC7B,WAAkC,EAClC,mBAAuC;IAEvC,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;IAC7B,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAEjF,IAAI,WAAW,IAAI,SAAS,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC;QACpC,4DAA4D;QAC5D,IAAI,OAAO,CAAC,YAAY,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACtD,gEAAgE;YAChE,OAAO,CAAC,QAAQ,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACvE,IAAI,CAAC,aAAa,CAChB,aAAa,CACX,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EACrG,WAAW,CACZ,CACF,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,WAAW,CAAC,KAAK,CAAC,MAAM,IAAI,iBAAiB,EAAE,CAAC;YACzD,IAAI,CAAC,WAAW,CACd,EAAE;YACF,gEAAgE;YAChE,2DAA2D;YAC3D,aAAa,CACX,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,oBAAoB,EAAE,OAAO,EAAE,qCAAqC,iBAAiB,6BAA6B,EAAE,EAC3I,SAAS,CACV,CACF,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,kEAAkE;YAClE,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,SAAS,CAAC,CAAC,CAAC;QACrF,CAAC;QACD,OAAO;IACT,CAAC;IAED,yEAAyE;IACzE,MAAM,GAAG,GAAG,IAAI,CAAC,qBAAqB,EAAE,CAAC;IACzC,IAAI,WAAW,CAAC,OAAO,CAAC,IAAI,GAAG,WAAW,CAAC,gBAAgB,IAAI,GAAG,EAAE,CAAC;QACnE,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE;YACnB,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,eAAe;YACrB,OAAO,EAAE,qCAAqC,GAAG,4BAA4B;YAC7E,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACpC,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,IAAI,SAAS,EAAE,CAAC;QACd,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,SAAS,CAAC,CAAC,CAAC;IACvF,CAAC;IACD,KAAK,UAAU,CAAC,IAAI,EAAE;QACpB,IAAI;QACJ,SAAS;QACT,WAAW;QACX,UAAU,EAAE,mBAAmB;KAChC,CAAC,CAAC;AACL,CAAC"}

package/dist/ws/session-catalog.d.ts

@@ -0,0 +1,25 @@
+import type { Database } from "bun:sqlite";
+import type { ServerResultMessage } from "@schlessera/brain-ui-sdk/protocol";
+/**
+ * Persistence seam for session ownership + accounting. The ws coordinator only
+ * talks to this interface — swapping the store means implementing four
+ * methods, not editing the turn loop.
+ */
+export interface SessionCatalog {
+    /** The provider/profile a stored session is pinned to, or null. */
+    getStoredProviderId(sessionId: string): string | null;
+    /** The backend that owns a stored session, or null for legacy/unknown rows. */
+    getStoredBackendId(sessionId: string): string | null;
+    /**
+     * Persist session ownership as soon as `session_info` names the session,
+     * with zero cost/turn deltas — the terminal `result` adds the accounting.
+     * Without this, a failed or cancelled first turn left the transcript
+     * unowned.
+     */
+    persistSessionStub(sessionId: string, promptText: string, providerId: string | null, backendId: string): void;
+    /** Persist session accounting on the terminal `result` frame. */
+    persistSession(msg: ServerResultMessage, promptText: string, providerId: string | null, backendId: string): void;
+}
+/** SQLite-backed catalog over the package's session database. */
+export declare function createSessionCatalog(db?: () => Database): SessionCatalog;
+//# sourceMappingURL=session-catalog.d.ts.map

package/dist/ws/session-catalog.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-catalog.d.ts","sourceRoot":"","sources":["../../src/ws/session-catalog.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAG7E;;;;GAIG;AACH,MAAM,WAAW,cAAc;IAC7B,mEAAmE;IACnE,mBAAmB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;IACtD,+EAA+E;IAC/E,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;IACrD;;;;;OAKG;IACH,kBAAkB,CAChB,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,EAClB,UAAU,EAAE,MAAM,GAAG,IAAI,EACzB,SAAS,EAAE,MAAM,GAChB,IAAI,CAAC;IACR,iEAAiE;IACjE,cAAc,CACZ,GAAG,EAAE,mBAAmB,EACxB,UAAU,EAAE,MAAM,EAClB,UAAU,EAAE,MAAM,GAAG,IAAI,EACzB,SAAS,EAAE,MAAM,GAChB,IAAI,CAAC;CACT;AAWD,iEAAiE;AACjE,wBAAgB,oBAAoB,CAAC,EAAE,GAAE,MAAM,QAAgB,GAAG,cAAc,CAmD/E"}

package/dist/ws/session-catalog.js

@@ -0,0 +1,45 @@
+import { getDb } from "../db/client.js";
+const UPSERT_SESSION_SQL = `INSERT INTO sessions (id, title, created_at, last_active_at, total_cost_usd, num_turns, provider_id, backend_id)
+   VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+   ON CONFLICT(id) DO UPDATE SET
+     last_active_at = excluded.last_active_at,
+     total_cost_usd = total_cost_usd + excluded.total_cost_usd,
+     num_turns = num_turns + excluded.num_turns,
+     provider_id = COALESCE(excluded.provider_id, provider_id),
+     backend_id = COALESCE(excluded.backend_id, backend_id)`;
+/** SQLite-backed catalog over the package's session database. */
+export function createSessionCatalog(db = getDb) {
+    return {
+        getStoredProviderId(sessionId) {
+            const row = db()
+                .query("SELECT provider_id AS providerId FROM sessions WHERE id = ?")
+                .get(sessionId);
+            return row?.providerId ?? null;
+        },
+        getStoredBackendId(sessionId) {
+            const row = db()
+                .query("SELECT backend_id AS backendId FROM sessions WHERE id = ?")
+                .get(sessionId);
+            return row?.backendId ?? null;
+        },
+        persistSessionStub(sessionId, promptText, providerId, backendId) {
+            try {
+                db()
+                    .prepare(UPSERT_SESSION_SQL)
+                    .run(sessionId, promptText.slice(0, 100), Date.now(), Date.now(), 0, 0, providerId, backendId);
+            }
+            catch { }
+        },
+        persistSession(msg, promptText, providerId, backendId) {
+            try {
+                db()
+                    .prepare(UPSERT_SESSION_SQL)
+                    .run(msg.sessionId, promptText.slice(0, 100), Date.now(), Date.now(), 
+                // costUsd is optional (rev 2): absent = unknown, accounted as 0.
+                msg.costUsd ?? 0, msg.numTurns, providerId, backendId);
+            }
+            catch { }
+        },
+    };
+}
+//# sourceMappingURL=session-catalog.js.map

package/dist/ws/session-catalog.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-catalog.js","sourceRoot":"","sources":["../../src/ws/session-catalog.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAiCxC,MAAM,kBAAkB,GAAG;;;;;;;4DAOiC,CAAC;AAE7D,iEAAiE;AACjE,MAAM,UAAU,oBAAoB,CAAC,KAAqB,KAAK;IAC7D,OAAO;QACL,mBAAmB,CAAC,SAAS;YAC3B,MAAM,GAAG,GAAG,EAAE,EAAE;iBACb,KAAK,CAAC,6DAA6D,CAAC;iBACpE,GAAG,CAAC,SAAS,CAAyC,CAAC;YAC1D,OAAO,GAAG,EAAE,UAAU,IAAI,IAAI,CAAC;QACjC,CAAC;QAED,kBAAkB,CAAC,SAAS;YAC1B,MAAM,GAAG,GAAG,EAAE,EAAE;iBACb,KAAK,CAAC,2DAA2D,CAAC;iBAClE,GAAG,CAAC,SAAS,CAAwC,CAAC;YACzD,OAAO,GAAG,EAAE,SAAS,IAAI,IAAI,CAAC;QAChC,CAAC;QAED,kBAAkB,CAAC,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS;YAC7D,IAAI,CAAC;gBACH,EAAE,EAAE;qBACD,OAAO,CAAC,kBAAkB,CAAC;qBAC3B,GAAG,CACF,SAAS,EACT,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EACxB,IAAI,CAAC,GAAG,EAAE,EACV,IAAI,CAAC,GAAG,EAAE,EACV,CAAC,EACD,CAAC,EACD,UAAU,EACV,SAAS,CACV,CAAC;YACN,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;QAED,cAAc,CAAC,GAAG,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS;YACnD,IAAI,CAAC;gBACH,EAAE,EAAE;qBACD,OAAO,CAAC,kBAAkB,CAAC;qBAC3B,GAAG,CACF,GAAG,CAAC,SAAS,EACb,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EACxB,IAAI,CAAC,GAAG,EAAE,EACV,IAAI,CAAC,GAAG,EAAE;gBACV,iEAAiE;gBACjE,GAAG,CAAC,OAAO,IAAI,CAAC,EAChB,GAAG,CAAC,QAAQ,EACZ,UAAU,EACV,SAAS,CACV,CAAC;YACN,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/ws/shrink.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Size-bounding for outbound WebSocket frames.
+ *
+ * A single oversized frame — typically a multi-megabyte tool result (a big
+ * `Read`/`Bash`/`Grep` output), a tool_use with a large input, or a tool
+ * result whose content array holds thousands of blocks — can exceed the
+ * socket's per-frame limit, drop the connection, and (because the client
+ * reconnects and re-requests the same history) wedge into a reconnect loop.
+ *
+ * This helper bounds any JSON-serializable payload below
+ * {@link MAX_WS_MESSAGE_BYTES} by head-truncating long strings AND
+ * head-clipping long arrays, leaving small metadata (ids, types, statuses)
+ * untouched. Both axes matter: string truncation alone leaves the cap
+ * unenforced for a payload built of many short strings. The model already
+ * received the full tool output server-side — this only bounds what the
+ * browser renders.
+ *
+ * Ported from oh-my-pi's collab replication-shrink.
+ */
+/** Per-frame ceiling for server → client messages. Well under Bun's 16 MB
+ * default and typical reverse-proxy WebSocket limits. */
+export declare const MAX_WS_MESSAGE_BYTES = 512000;
+/** Target byte size for one `session_history` chunk. Below the frame cap so a
+ * batch plus its wrapper stays comfortably under {@link MAX_WS_MESSAGE_BYTES}. */
+export declare const HISTORY_CHUNK_BYTES = 400000;
+/**
+ * Return `value` unchanged when its JSON serialization already fits
+ * {@link MAX_WS_MESSAGE_BYTES}; otherwise return a deep-cloned shadow shrunk
+ * along both string and array axes until it fits. The wire shape is preserved:
+ * only string leaves and array tails change — discriminator fields, ids, and
+ * other small metadata pass through untouched.
+ */
+export declare function shrinkForReplication<T>(value: T): T;
+//# sourceMappingURL=shrink.d.ts.map

package/dist/ws/shrink.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"shrink.d.ts","sourceRoot":"","sources":["../../src/ws/shrink.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AAEH;yDACyD;AACzD,eAAO,MAAM,oBAAoB,SAAU,CAAC;AAE5C;kFACkF;AAClF,eAAO,MAAM,mBAAmB,SAAU,CAAC;AAqD3C;;;;;;GAMG;AACH,wBAAgB,oBAAoB,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,CAQnD"}