@schlessera/brain-ui-react 0.34.1 → 0.36.0

package/dist/hooks/use-websocket.js

@@ -1,387 +1,16 @@
-import { useEffect, useRef, useCallback } from "react";
-import { BrainUiClient, } from "@schlessera/brain-ui-sdk/client";
-import { useConnectionStore } from "../stores/connection-store.js";
-import { useChatStore, activeChat } from "../stores/chat-store.js";
-import { useActivityStore, loadSessionActivityHistory } from "../stores/activity-store.js";
-import { getWsUrl } from "../lib/backend.js";
-import { dispatchServerMessage } from "./websocket-handlers/index.js";
-import { runStateForFrame } from "./websocket-handlers/chat.js";
-import { recheckVpnStatus } from "./use-vpn-status.js";
-import { REFUSAL_ATTEMPTS } from "../components/connectivity/connection-state.js";
+import { useEffect, useCallback } from "react";
+import { defaultRoot } from "../default-root.js";
+import { useBrainUiRoot } from "../root-context.js";
 export { runStateForFrame } from "./websocket-handlers/chat.js";
-/**
- * Does this frame announce the identity of the conversation THIS client just
- * started?
- *
- * The old test was `session_info || result` for any unknown session, which
- * adopted whichever arrived first — an older background turn finishing, or
- * another client's new session, would capture the user's draft and bind it to
- * a transcript they never wrote. `draftId` is minted per draft turn and echoed
- * on `session_info`, so a match is proof.
- *
- * A server too old to echo it sends no `draftId`, and the pre-existing
- * behaviour applies rather than the draft never binding at all: correctness
- * where the information exists, compatibility where it does not.
- */
-function isOurDraftAnnouncement(state, msg) {
-    if (msg.type !== "session_info" && msg.type !== "result")
-        return false;
-    const pending = state.pendingDraftId;
-    if (msg.type === "session_info" && msg.draftId)
-        return msg.draftId === pending;
-    // No echo to compare against.
-    return true;
-}
-/**
- * Lazily open a session-scoped activity subscription (live subagent rows,
- * server-stamped tool timing). Idempotent per connection; the subscribed set
- * resets when a new socket's server_hello arrives.
- */
-function ensureActivitySubscription(sessionId) {
-    if (!sessionId)
-        return;
-    const activity = useActivityStore.getState();
-    if (!activity.supported || activity.subscribed[sessionId])
-        return;
-    activity.markSubscribed(sessionId);
-    wsClient?.send({ type: "activity_subscribe", view: "session", sessionId });
-    // The subscription snapshot carries OPEN runs; finished runs (history
-    // timing, past fan-outs) come from the REST side.
-    void loadSessionActivityHistory(sessionId);
-}
-const pendingDeltas = new Map();
-let flushScheduled = false;
-/**
- * Apply every buffered delta, in arrival order, and clear the buffer.
- *
- * Exported so tests can force the frame rather than wait for one; production
- * code never needs to call it, because every path that reads the transcript
- * flushes first.
- */
-export function flushChatDeltas() {
-    flushDeltas();
-}
-function flushDeltas() {
-    flushScheduled = false;
-    if (pendingDeltas.size === 0)
-        return;
-    const batches = [...pendingDeltas.entries()];
-    pendingDeltas.clear();
-    const state = useChatStore.getState();
-    for (const [key, chunks] of batches) {
-        for (const chunk of chunks) {
-            if (chunk.kind === "text")
-                state.appendText(key, chunk.text);
-            else
-                state.appendThinking(key, chunk.text);
-        }
-    }
-}
-function enqueueDelta(key, kind, text) {
-    const chunks = pendingDeltas.get(key);
-    if (!chunks) {
-        pendingDeltas.set(key, [{ kind, text }]);
-    }
-    else {
-        const last = chunks[chunks.length - 1];
-        if (last.kind === kind)
-            last.text += text;
-        else
-            chunks.push({ kind, text });
-    }
-    // Outside a browser (unit tests, SSR) there is no frame to wait for, so apply
-    // straight away and keep the handler's behaviour synchronous.
-    if (typeof requestAnimationFrame !== "function") {
-        flushDeltas();
-        return;
-    }
-    if (!flushScheduled) {
-        flushScheduled = true;
-        requestAnimationFrame(flushDeltas);
-    }
-}
-export function handleServerMessage(msg) {
-    // Every frame that is not itself a delta must see the transcript fully
-    // applied: the demux below reads buffer state, and the store records parts in
-    // arrival order, so a tool call landing ahead of buffered text would reorder
-    // the message.
-    if (msg.type !== "text_delta" && msg.type !== "thinking_delta")
-        flushDeltas();
-    const state = useChatStore.getState();
-    // Activity stream frames feed their own store and never touch chat state.
-    if (msg.type === "server_hello" ||
-        msg.type === "activity_snapshot" ||
-        msg.type === "activity_delta") {
-        const key = state.activeSessionId;
-        dispatchServerMessage(msg, {
-            state,
-            frameSessionId: undefined,
-            key,
-            buffer: () => (key === null ? state.draft : state.buffers[key]),
-            enqueueDelta,
-            ensureActivitySubscription,
-            requestBrowserLocation,
-            resyncIfNeeded,
-            coldResumeIfNeeded,
-            markHistoryReplaced,
-        });
-        return;
-    }
-    // Parallel-session demux (per-session buffers). Every frame is scoped by
-    // sessionId; a scoped frame lands in ITS session's buffer, so a background
-    // session's transcript keeps accumulating while another one is in view.
-    // Frames without a sessionId (legacy single-session servers, or a new
-    // session's pre-binding frames) apply to the buffer in view.
-    const frameSessionId = msg.sessionId;
-    if (frameSessionId) {
-        // `detail` on a queued status is the host's queue-pressure note; it rides
-        // along so the session's badge can show WHY it is queued deep.
-        const queueNote = msg.type === "status" && msg.status === "queued" ? msg.detail : undefined;
-        state.setRunState(frameSessionId, runStateForFrame(msg), queueNote);
-    }
-    // Resolve the target buffer key for this frame.
-    let key;
-    if (!frameSessionId) {
-        key = state.activeSessionId; // null = the draft view
-    }
-    else if (state.buffers[frameSessionId]) {
-        key = frameSessionId;
-    }
-    else if (state.draft && isOurDraftAnnouncement(state, msg)) {
-        // A draft run just got its server identity: adopt the draft buffer.
-        state.bindDraftSession(frameSessionId);
-        key = frameSessionId;
-    }
-    else if (msg.type === "session_info" && frameSessionId === state.activeSessionId) {
-        // The ACTIVE session announced itself but has no buffer yet — the cold
-        // reattach case (PWA relaunch mid-turn; the server may skip the history
-        // frame when a first turn has no history yet). Materialize the buffer so
-        // this session_info and every following delta land instead of vanishing.
-        state.setActiveSession(frameSessionId);
-        key = frameSessionId;
-    }
-    else if (msg.type === "session_history") {
-        // History replay may create a buffer (cold resume / reattach).
-        key = frameSessionId;
-    }
-    else {
-        // Scoped frame for a session with no buffer (never opened, or evicted):
-        // the badge above is all we track. The transcript heals via
-        // session_resume when the user opens it.
-        return;
-    }
-    const buffer = () => {
-        const s = useChatStore.getState();
-        return key === null ? s.draft : s.buffers[key];
-    };
-    dispatchServerMessage(msg, {
-        state,
-        frameSessionId,
-        key,
-        buffer,
-        enqueueDelta,
-        ensureActivitySubscription,
-        requestBrowserLocation,
-        resyncIfNeeded,
-        coldResumeIfNeeded,
-        markHistoryReplaced,
-    });
-}
-/**
- * The server asked the browser for its location (the agent called
- * `get_current_location`). Read `navigator.geolocation` and reply over the same
- * requestId. The browser shows its own permission prompt on first use; a denial,
- * timeout, or unavailable position comes back as a `location_error` that the
- * server turns into a tool error for the agent.
- */
-function requestBrowserLocation(msg) {
-    const { requestId } = msg;
-    if (typeof navigator === "undefined" || !navigator.geolocation) {
-        wsClient?.send({
-            type: "location_error",
-            requestId,
-            code: 0,
-            message: "Geolocation is not available in this browser.",
-        });
-        return;
-    }
-    const opts = msg.options ?? {};
-    navigator.geolocation.getCurrentPosition((pos) => {
-        wsClient?.send({
-            type: "location_response",
-            requestId,
-            coords: {
-                latitude: pos.coords.latitude,
-                longitude: pos.coords.longitude,
-                accuracy: pos.coords.accuracy,
-                altitude: pos.coords.altitude,
-                altitudeAccuracy: pos.coords.altitudeAccuracy,
-                heading: pos.coords.heading,
-                speed: pos.coords.speed,
-            },
-            timestamp: pos.timestamp,
-        });
-    }, (err) => {
-        wsClient?.send({
-            type: "location_error",
-            requestId,
-            code: err.code,
-            message: err.message || "Failed to retrieve location.",
-        });
-    }, {
-        enableHighAccuracy: opts.enableHighAccuracy ?? false,
-        timeout: opts.timeoutMs ?? 15000,
-        maximumAge: opts.maximumAgeMs ?? 60000,
-    });
-}
-// Reconnection bookkeeping: after a WS drop, the rendered transcript may have
-// gaps (deltas streamed while offline are lost). We heal by replaying the
-// authoritative session history — immediately if nothing is streaming, or
-// once the in-flight turn settles (result / idle status). The pending resync
-// is keyed to the session that needs healing (the one that was active at the
-// reconnect), so a background session settling first cannot consume it.
-let wasDisconnected = false;
-let resyncSessionId = null;
-function handleSocketClose(close) {
-    const connection = useConnectionStore.getState();
-    connection.recordWsClose(close.opened, close.code);
-    if (!close.opened &&
-        useConnectionStore.getState().handshakeFailures >= REFUSAL_ATTEMPTS) {
-        recheckVpnStatus();
-    }
-}
-function resyncIfNeeded(sessionId) {
-    if (!resyncSessionId || !sessionId || sessionId !== resyncSessionId)
-        return;
-    resyncSessionId = null;
-    wsClient?.send({ type: "session_resume", sessionId });
-}
-// A cold page / PWA relaunch restores only the stored session id, not the
-// transcript. The server pushes a snapshot on connect only while a session is
-// actively running; when it's idle (the common "agent finished while the phone
-// slept" case) nothing arrives, so the client must fetch the finished
-// session's history itself. Guarded to fire at most once per session id per
-// connection — an empty session would otherwise re-trigger on every idle
-// status and loop.
-let coldResumedSessionId = null;
-function markHistoryReplaced(key) {
-    if (key !== null && key === resyncSessionId)
-        resyncSessionId = null;
-    coldResumedSessionId = key;
-}
-function coldResumeIfNeeded(sessionId, messageCount) {
-    if (!sessionId || messageCount > 0)
-        return;
-    if (coldResumedSessionId === sessionId)
-        return;
-    coldResumedSessionId = sessionId;
-    wsClient?.send({ type: "session_resume", sessionId });
-}
-function handleStatusChange(status) {
-    useConnectionStore.getState().setWsStatus(status);
-    if (status === "connected") {
-        useConnectionStore.getState().noteSocketOpen();
-    }
-    // A status change can be followed by a history replay that rewrites the
-    // buffer, so land whatever is still buffered before anything reads it. In a
-    // hidden tab requestAnimationFrame does not run at all, which is exactly the
-    // case where deltas could otherwise sit unapplied across a reconnect.
-    flushDeltas();
-    if (status === "disconnected") {
-        wasDisconnected = true;
-        // Allow a fresh cold-resume attempt after we reconnect.
-        coldResumedSessionId = null;
-    }
-    else if (status === "connected" && wasDisconnected) {
-        wasDisconnected = false;
-        const chat = useChatStore.getState();
-        const active = activeChat(chat);
-        if (chat.activeSessionId && active.messages.length > 0) {
-            resyncSessionId = chat.activeSessionId;
-            // Not mid-stream: replay right away. Mid-stream: the flag holds until
-            // the running turn finishes (result or idle status).
-            if (!active.isStreaming) {
-                resyncIfNeeded(chat.activeSessionId);
-            }
-        }
-    }
-}
-// Singleton client - survives React re-renders
-let wsClient = null;
-/**
- * Send on the live socket from outside a component.
- *
- * `useWebSocket()` OWNS the socket — its construction guard is a per-instance
- * ref, so a second caller would build a second client, overwrite this
- * singleton, and orphan the first; unmounting either one would then close the
- * socket for both. Anything that needs to send but not to own (the share
- * intake) goes through here instead of calling the hook again.
- *
- * Returns false when there is no open socket: `send` drops silently in that
- * case and a caller that just staged an upload needs to know.
- */
-export function sendClientMessage(msg) {
-    if (!wsClient)
-        return false;
-    if (useConnectionStore.getState().wsStatus !== "connected")
-        return false;
-    wsClient.send(msg);
-    return true;
-}
-/** Retry the singleton socket immediately, skipping any remaining backoff. */
-export function reconnectWebSocketNow() {
-    wsClient?.reconnectNow();
-}
+/** Default-app entry points for callers outside React. */
+export function handleServerMessage(message) { defaultRoot.connection.handleServerMessage(message); }
+export function flushChatDeltas() { defaultRoot.connection.flushChatDeltas(); }
+export function sendClientMessage(message) { return defaultRoot.connection.send(message); }
+export function reconnectWebSocketNow() { defaultRoot.connection.reconnectNow(); }
 export function useWebSocket() {
-    const initialized = useRef(false);
-    useEffect(() => {
-        if (initialized.current)
-            return;
-        initialized.current = true;
-        let disposed = false;
-        useConnectionStore.getState().noteSocketOpen();
-        wsClient = new BrainUiClient({
-            url: getWsUrl(),
-            // One handler with a shared preamble, rather than sixteen copies of the
-            // session-buffer demux — see handleServerMessage.
-            handlers: { onAny: handleServerMessage },
-            onStatusChange: handleStatusChange,
-            onClose: (close) => {
-                if (!disposed)
-                    handleSocketClose(close);
-            },
-            // A frame the SDK refused. Surfaced rather than logged into a console
-            // nobody is attached to; the server-side counterpart is the
-            // ws.frames.dropped counter.
-            onProtocolError: (err) => {
-                useConnectionStore
-                    .getState()
-                    .reportError("PROTOCOL_ERROR", err.frameType
-                    ? `Dropped a ${err.frameType} frame: ${err.detail}`
-                    : `Dropped an unreadable frame: ${err.detail}`);
-            },
-        });
-        wsClient.connect();
-        // Skip the exponential backoff when the network demonstrably returns.
-        const reconnectNow = () => wsClient?.reconnectNow();
-        const onVisible = () => {
-            if (document.visibilityState === "visible")
-                wsClient?.reconnectNow();
-        };
-        window.addEventListener("online", reconnectNow);
-        document.addEventListener("visibilitychange", onVisible);
-        return () => {
-            disposed = true;
-            window.removeEventListener("online", reconnectNow);
-            document.removeEventListener("visibilitychange", onVisible);
-            wsClient?.close();
-            wsClient = null;
-            initialized.current = false;
-        };
-    }, []);
-    const send = useCallback((msg) => {
-        wsClient?.send(msg);
-    }, []);
+    const root = useBrainUiRoot();
+    useEffect(() => root.connection.connect(), [root]);
+    const send = useCallback((message) => root.connection.send(message), [root]);
     return { send };
 }
 //# sourceMappingURL=use-websocket.js.map

package/dist/hooks/use-websocket.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-websocket.js","sourceRoot":"","sources":["../../src/hooks/use-websocket.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACvD,OAAO,EACL,aAAa,GAEd,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,UAAU,EAAgB,MAAM,yBAAyB,CAAC;AACjF,OAAO,EAAE,gBAAgB,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AAC3F,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAM7C,OAAO,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gDAAgD,CAAC;AAElF,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAEhE;;;;;;;;;;;;;GAaG;AACH,SAAS,sBAAsB,CAC7B,KAA+C,EAC/C,GAAkB;IAElB,IAAI,GAAG,CAAC,IAAI,KAAK,cAAc,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACvE,MAAM,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC;IACrC,IAAI,GAAG,CAAC,IAAI,KAAK,cAAc,IAAI,GAAG,CAAC,OAAO;QAAE,OAAO,GAAG,CAAC,OAAO,KAAK,OAAO,CAAC;IAC/E,8BAA8B;IAC9B,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;GAIG;AACH,SAAS,0BAA0B,CAAC,SAAoC;IACtE,IAAI,CAAC,SAAS;QAAE,OAAO;IACvB,MAAM,QAAQ,GAAG,gBAAgB,CAAC,QAAQ,EAAE,CAAC;IAC7C,IAAI,CAAC,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO;IAClE,QAAQ,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IACnC,QAAQ,EAAE,IAAI,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;IAC3E,sEAAsE;IACtE,kDAAkD;IAClD,KAAK,0BAA0B,CAAC,SAAS,CAAC,CAAC;AAC7C,CAAC;AAgBD,MAAM,aAAa,GAAG,IAAI,GAAG,EAA2B,CAAC;AACzD,IAAI,cAAc,GAAG,KAAK,CAAC;AAE3B;;;;;;GAMG;AACH,MAAM,UAAU,eAAe;IAC7B,WAAW,EAAE,CAAC;AAChB,CAAC;AAED,SAAS,WAAW;IAClB,cAAc,GAAG,KAAK,CAAC;IACvB,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO;IACrC,MAAM,OAAO,GAAG,CAAC,GAAG,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC;IAC7C,aAAa,CAAC,KAAK,EAAE,CAAC;IACtB,MAAM,KAAK,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;IACtC,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;QACpC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;gBAAE,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;;gBACxD,KAAK,CAAC,cAAc,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,GAAY,EAAE,IAA0B,EAAE,IAAY;IAC1E,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACtC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,aAAa,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAC3C,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACvC,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI;YAAE,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC;;YACrC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACnC,CAAC;IACD,8EAA8E;IAC9E,8DAA8D;IAC9D,IAAI,OAAO,qBAAqB,KAAK,UAAU,EAAE,CAAC;QAChD,WAAW,EAAE,CAAC;QACd,OAAO;IACT,CAAC;IACD,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,cAAc,GAAG,IAAI,CAAC;QACtB,qBAAqB,CAAC,WAAW,CAAC,CAAC;IACrC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,GAAkB;IACpD,uEAAuE;IACvE,8EAA8E;IAC9E,6EAA6E;IAC7E,eAAe;IACf,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,IAAI,GAAG,CAAC,IAAI,KAAK,gBAAgB;QAAE,WAAW,EAAE,CAAC;IAE9E,MAAM,KAAK,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;IAEtC,0EAA0E;IAC1E,IACE,GAAG,CAAC,IAAI,KAAK,cAAc;QAC3B,GAAG,CAAC,IAAI,KAAK,mBAAmB;QAChC,GAAG,CAAC,IAAI,KAAK,gBAAgB,EAC7B,CAAC;QACD,MAAM,GAAG,GAAG,KAAK,CAAC,eAAe,CAAC;QAClC,qBAAqB,CAAC,GAAG,EAAE;YACzB,KAAK;YACL,cAAc,EAAE,SAAS;YACzB,GAAG;YACH,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAC/D,YAAY;YACZ,0BAA0B;YAC1B,sBAAsB;YACtB,cAAc;YACd,kBAAkB;YAClB,mBAAmB;SACpB,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,yEAAyE;IACzE,2EAA2E;IAC3E,wEAAwE;IACxE,sEAAsE;IACtE,6DAA6D;IAC7D,MAAM,cAAc,GAAI,GAA8B,CAAC,SAAS,CAAC;IACjE,IAAI,cAAc,EAAE,CAAC;QACnB,0EAA0E;QAC1E,+DAA+D;QAC/D,MAAM,SAAS,GACb,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;QAC5E,KAAK,CAAC,WAAW,CAAC,cAAc,EAAE,gBAAgB,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC;IACtE,CAAC;IAED,gDAAgD;IAChD,IAAI,GAAY,CAAC;IACjB,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,GAAG,GAAG,KAAK,CAAC,eAAe,CAAC,CAAC,wBAAwB;IACvD,CAAC;SAAM,IAAI,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QACzC,GAAG,GAAG,cAAc,CAAC;IACvB,CAAC;SAAM,IAAI,KAAK,CAAC,KAAK,IAAI,sBAAsB,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC;QAC7D,oEAAoE;QACpE,KAAK,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;QACvC,GAAG,GAAG,cAAc,CAAC;IACvB,CAAC;SAAM,IAAI,GAAG,CAAC,IAAI,KAAK,cAAc,IAAI,cAAc,KAAK,KAAK,CAAC,eAAe,EAAE,CAAC;QACnF,uEAAuE;QACvE,wEAAwE;QACxE,yEAAyE;QACzE,yEAAyE;QACzE,KAAK,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;QACvC,GAAG,GAAG,cAAc,CAAC;IACvB,CAAC;SAAM,IAAI,GAAG,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;QAC1C,+DAA+D;QAC/D,GAAG,GAAG,cAAc,CAAC;IACvB,CAAC;SAAM,CAAC;QACN,wEAAwE;QACxE,4DAA4D;QAC5D,yCAAyC;QACzC,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GAAG,GAAG,EAAE;QAClB,MAAM,CAAC,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;QAClC,OAAO,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACjD,CAAC,CAAC;IAEF,qBAAqB,CAAC,GAAG,EAAE;QACzB,KAAK;QACL,cAAc;QACd,GAAG;QACH,MAAM;QACN,YAAY;QACZ,0BAA0B;QAC1B,sBAAsB;QACtB,cAAc;QACd,kBAAkB;QAClB,mBAAmB;KACpB,CAAC,CAAC;AACL,CAAC;AAED;;;;;;GAMG;AACH,SAAS,sBAAsB,CAAC,GAA0B;IACxD,MAAM,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC;IAC1B,IAAI,OAAO,SAAS,KAAK,WAAW,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC;QAC/D,QAAQ,EAAE,IAAI,CAAC;YACb,IAAI,EAAE,gBAAgB;YACtB,SAAS;YACT,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,+CAA+C;SACzD,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IACD,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;IAC/B,SAAS,CAAC,WAAW,CAAC,kBAAkB,CACtC,CAAC,GAAG,EAAE,EAAE;QACN,QAAQ,EAAE,IAAI,CAAC;YACb,IAAI,EAAE,mBAAmB;YACzB,SAAS;YACT,MAAM,EAAE;gBACN,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,QAAQ;gBAC7B,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,SAAS;gBAC/B,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,QAAQ;gBAC7B,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,QAAQ;gBAC7B,gBAAgB,EAAE,GAAG,CAAC,MAAM,CAAC,gBAAgB;gBAC7C,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,OAAO;gBAC3B,KAAK,EAAE,GAAG,CAAC,MAAM,CAAC,KAAK;aACxB;YACD,SAAS,EAAE,GAAG,CAAC,SAAS;SACzB,CAAC,CAAC;IACL,CAAC,EACD,CAAC,GAAG,EAAE,EAAE;QACN,QAAQ,EAAE,IAAI,CAAC;YACb,IAAI,EAAE,gBAAgB;YACtB,SAAS;YACT,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,8BAA8B;SACvD,CAAC,CAAC;IACL,CAAC,EACD;QACE,kBAAkB,EAAE,IAAI,CAAC,kBAAkB,IAAI,KAAK;QACpD,OAAO,EAAE,IAAI,CAAC,SAAS,IAAI,KAAK;QAChC,UAAU,EAAE,IAAI,CAAC,YAAY,IAAI,KAAK;KACvC,CACF,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,0EAA0E;AAC1E,0EAA0E;AAC1E,6EAA6E;AAC7E,6EAA6E;AAC7E,wEAAwE;AACxE,IAAI,eAAe,GAAG,KAAK,CAAC;AAC5B,IAAI,eAAe,GAAkB,IAAI,CAAC;AAE1C,SAAS,iBAAiB,CAAC,KAAqB;IAC9C,MAAM,UAAU,GAAG,kBAAkB,CAAC,QAAQ,EAAE,CAAC;IACjD,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACnD,IACE,CAAC,KAAK,CAAC,MAAM;QACb,kBAAkB,CAAC,QAAQ,EAAE,CAAC,iBAAiB,IAAI,gBAAgB,EACnE,CAAC;QACD,gBAAgB,EAAE,CAAC;IACrB,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,SAAwB;IAC9C,IAAI,CAAC,eAAe,IAAI,CAAC,SAAS,IAAI,SAAS,KAAK,eAAe;QAAE,OAAO;IAC5E,eAAe,GAAG,IAAI,CAAC;IACvB,QAAQ,EAAE,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,SAAS,EAAE,CAAC,CAAC;AACxD,CAAC;AAED,0EAA0E;AAC1E,8EAA8E;AAC9E,+EAA+E;AAC/E,sEAAsE;AACtE,4EAA4E;AAC5E,yEAAyE;AACzE,mBAAmB;AACnB,IAAI,oBAAoB,GAAkB,IAAI,CAAC;AAE/C,SAAS,mBAAmB,CAAC,GAAY;IACvC,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,eAAe;QAAE,eAAe,GAAG,IAAI,CAAC;IACpE,oBAAoB,GAAG,GAAG,CAAC;AAC7B,CAAC;AAED,SAAS,kBAAkB,CAAC,SAAwB,EAAE,YAAoB;IACxE,IAAI,CAAC,SAAS,IAAI,YAAY,GAAG,CAAC;QAAE,OAAO;IAC3C,IAAI,oBAAoB,KAAK,SAAS;QAAE,OAAO;IAC/C,oBAAoB,GAAG,SAAS,CAAC;IACjC,QAAQ,EAAE,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,SAAS,EAAE,CAAC,CAAC;AACxD,CAAC;AAED,SAAS,kBAAkB,CAAC,MAAmD;IAC7E,kBAAkB,CAAC,QAAQ,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAElD,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;QAC3B,kBAAkB,CAAC,QAAQ,EAAE,CAAC,cAAc,EAAE,CAAC;IACjD,CAAC;IAED,wEAAwE;IACxE,4EAA4E;IAC5E,6EAA6E;IAC7E,sEAAsE;IACtE,WAAW,EAAE,CAAC;IAEd,IAAI,MAAM,KAAK,cAAc,EAAE,CAAC;QAC9B,eAAe,GAAG,IAAI,CAAC;QACvB,wDAAwD;QACxD,oBAAoB,GAAG,IAAI,CAAC;IAC9B,CAAC;SAAM,IAAI,MAAM,KAAK,WAAW,IAAI,eAAe,EAAE,CAAC;QACrD,eAAe,GAAG,KAAK,CAAC;QACxB,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;QACrC,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;QAChC,IAAI,IAAI,CAAC,eAAe,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvD,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;YACvC,sEAAsE;YACtE,qDAAqD;YACrD,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBACxB,cAAc,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,+CAA+C;AAC/C,IAAI,QAAQ,GAAyB,IAAI,CAAC;AAE1C;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,iBAAiB,CAAC,GAAkB;IAClD,IAAI,CAAC,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5B,IAAI,kBAAkB,CAAC,QAAQ,EAAE,CAAC,QAAQ,KAAK,WAAW;QAAE,OAAO,KAAK,CAAC;IACzE,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACnB,OAAO,IAAI,CAAC;AACd,CAAC;AAED,8EAA8E;AAC9E,MAAM,UAAU,qBAAqB;IACnC,QAAQ,EAAE,YAAY,EAAE,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAElC,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,WAAW,CAAC,OAAO;YAAE,OAAO;QAChC,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;QAC3B,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,kBAAkB,CAAC,QAAQ,EAAE,CAAC,cAAc,EAAE,CAAC;QAE/C,QAAQ,GAAG,IAAI,aAAa,CAAC;YAC3B,GAAG,EAAE,QAAQ,EAAE;YACf,wEAAwE;YACxE,kDAAkD;YAClD,QAAQ,EAAE,EAAE,KAAK,EAAE,mBAAmB,EAAE;YACxC,cAAc,EAAE,kBAAkB;YAClC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACjB,IAAI,CAAC,QAAQ;oBAAE,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAC1C,CAAC;YACD,sEAAsE;YACtE,4DAA4D;YAC5D,6BAA6B;YAC7B,eAAe,EAAE,CAAC,GAAG,EAAE,EAAE;gBACvB,kBAAkB;qBACf,QAAQ,EAAE;qBACV,WAAW,CACV,gBAAgB,EAChB,GAAG,CAAC,SAAS;oBACX,CAAC,CAAC,aAAa,GAAG,CAAC,SAAS,WAAW,GAAG,CAAC,MAAM,EAAE;oBACnD,CAAC,CAAC,gCAAgC,GAAG,CAAC,MAAM,EAAE,CACjD,CAAC;YACN,CAAC;SACF,CAAC,CAAC;QACH,QAAQ,CAAC,OAAO,EAAE,CAAC;QAEnB,sEAAsE;QACtE,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,CAAC;QACpD,MAAM,SAAS,GAAG,GAAG,EAAE;YACrB,IAAI,QAAQ,CAAC,eAAe,KAAK,SAAS;gBAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;QACvE,CAAC,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAChD,QAAQ,CAAC,gBAAgB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;QAEzD,OAAO,GAAG,EAAE;YACV,QAAQ,GAAG,IAAI,CAAC;YAChB,MAAM,CAAC,mBAAmB,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;YACnD,QAAQ,CAAC,mBAAmB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;YAC5D,QAAQ,EAAE,KAAK,EAAE,CAAC;YAClB,QAAQ,GAAG,IAAI,CAAC;YAChB,WAAW,CAAC,OAAO,GAAG,KAAK,CAAC;QAC9B,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,IAAI,GAAG,WAAW,CAAC,CAAC,GAAkB,EAAE,EAAE;QAC9C,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;IACtB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,EAAE,IAAI,EAAE,CAAC;AAClB,CAAC"}
+{"version":3,"file":"use-websocket.js","sourceRoot":"","sources":["../../src/hooks/use-websocket.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AAE/C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAEhE,0DAA0D;AAC1D,MAAM,UAAU,mBAAmB,CAAC,OAAsB,IAAI,WAAW,CAAC,UAAU,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AACpH,MAAM,UAAU,eAAe,KAAK,WAAW,CAAC,UAAU,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;AAC/E,MAAM,UAAU,iBAAiB,CAAC,OAAsB,IAAI,OAAO,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AAC1G,MAAM,UAAU,qBAAqB,KAAK,WAAW,CAAC,UAAU,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;AAElF,MAAM,UAAU,YAAY;IAC1B,MAAM,IAAI,GAAG,cAAc,EAAE,CAAC;IAC9B,SAAS,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;IACnD,MAAM,IAAI,GAAG,WAAW,CAAC,CAAC,OAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;IAC5F,OAAO,EAAE,IAAI,EAAE,CAAC;AAClB,CAAC"}

package/dist/hooks/websocket-handlers/activity.d.ts

@@ -1,5 +1,5 @@
 export declare const activityFrameHandlers: {
-    activity_snapshot: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerActivitySnapshot) => void;
-    activity_delta: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerActivityDelta) => void;
+    activity_snapshot: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerActivitySnapshot, context: import("./types.js").DispatchContext) => void;
+    activity_delta: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerActivityDelta, context: import("./types.js").DispatchContext) => void;
 };
 //# sourceMappingURL=activity.d.ts.map

package/dist/hooks/websocket-handlers/activity.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"activity.d.ts","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/activity.ts"],"names":[],"mappings":"AAKA,eAAO,MAAM,qBAAqB;;;CAOgB,CAAC"}
+{"version":3,"file":"activity.d.ts","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/activity.ts"],"names":[],"mappings":"AAIA,eAAO,MAAM,qBAAqB;;;CAOgB,CAAC"}

package/dist/hooks/websocket-handlers/activity.js

@@ -1,10 +1,9 @@
-import { useActivityStore } from "../../stores/activity-store.js";
 export const activityFrameHandlers = {
-    activity_snapshot: (msg) => {
-        useActivityStore.getState().applySnapshot(msg);
+    activity_snapshot: (msg, context) => {
+        context.stores.activity.getState().applySnapshot(msg);
     },
-    activity_delta: (msg) => {
-        useActivityStore.getState().applyDelta(msg);
+    activity_delta: (msg, context) => {
+        context.stores.activity.getState().applyDelta(msg);
     },
 };
 //# sourceMappingURL=activity.js.map

package/dist/hooks/websocket-handlers/activity.js.map

@@ -1 +1 @@
-{"version":3,"file":"activity.js","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/activity.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAKlE,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,iBAAiB,EAAE,CAAC,GAAG,EAAE,EAAE;QACzB,gBAAgB,CAAC,QAAQ,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IACjD,CAAC;IACD,cAAc,EAAE,CAAC,GAAG,EAAE,EAAE;QACtB,gBAAgB,CAAC,QAAQ,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IAC9C,CAAC;CAC+C,CAAC"}
+{"version":3,"file":"activity.js","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/activity.ts"],"names":[],"mappings":"AAIA,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,iBAAiB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAClC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IACxD,CAAC;IACD,cAAc,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC/B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IACrD,CAAC;CAC+C,CAAC"}

package/dist/hooks/websocket-handlers/chat.d.ts

@@ -11,6 +11,7 @@ export declare const chatFrameHandlers: {
     tool_result: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerToolResult, context: import("./types.js").DispatchContext) => void;
     ask_user_request: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerAskUserRequest, context: import("./types.js").DispatchContext) => void;
     result: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerResultMessage, context: import("./types.js").DispatchContext) => void;
+    message_blocks: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerMessageBlocks, context: import("./types.js").DispatchContext) => void;
     session_history: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerSessionHistory, context: import("./types.js").DispatchContext) => void;
     status: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerStatus, context: import("./types.js").DispatchContext) => void;
 };

package/dist/hooks/websocket-handlers/chat.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat.d.ts","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/chat.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAEV,aAAa,EAEd,MAAM,mCAAmC,CAAC;AAkG3C,gFAAgF;AAChF,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,aAAa,GAAG,WAAW,GAAG,QAAQ,GAAG,MAAM,CAOpF;AAED,eAAO,MAAM,iBAAiB;;;;;;;;;;;;CA+FgB,CAAC"}
+{"version":3,"file":"chat.d.ts","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/chat.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAEV,aAAa,EAEd,MAAM,mCAAmC,CAAC;AA6G3C,gFAAgF;AAChF,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,aAAa,GAAG,WAAW,GAAG,QAAQ,GAAG,MAAM,CAOpF;AAED,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;CAyGgB,CAAC"}

package/dist/hooks/websocket-handlers/chat.js

@@ -1,4 +1,4 @@
-import { activeChat, useChatStore } from "../../stores/chat-store.js";
+import { activeChat } from "../../stores/chat-state.js";
 import { isAskUserTool } from "../../lib/tool-names.js";
 let historyMessageCounter = 0;
 function historyId() {
@@ -33,9 +33,19 @@ function convertHistoryMessage(msg) {
         timestamp: Date.now(),
         ...(askUserExchanges ? { askUserExchanges } : {}),
         ...(msg.attachmentCount ? { attachmentCount: msg.attachmentCount } : {}),
+        ...(msg.blocks && msg.blocks.length > 0 ? { blocks: msg.blocks } : {}),
     };
 }
-/** Rebuild ask_user exchanges from a resumed message's tool calls. */
+/**
+ * Rebuild ask_user exchanges from a resumed message's tool calls.
+ *
+ * The persisted tool output is `{answers, annotations}` (Claude) or the bare
+ * answers map (pi). Neither carries whether the answer was chosen from the
+ * options or typed into the composer (`AskUserExchange.typed`), nor when it
+ * was given, so a resumed exchange comes back as a plain `answered` one with
+ * no time. Carrying `typed` would mean widening the `ask_user` response —
+ * a `CONTRACT:` change — not guessing here.
+ */
 function isBareAnswersMap(v) {
     return (!!v &&
         typeof v === "object" &&
@@ -90,14 +100,18 @@ export const chatFrameHandlers = {
         // Opening the bubble stays immediate — the first token should show a
         // message starting, and every later delta needs isStreaming to be true.
         if (!context.buffer()?.isStreaming) {
-            context.state.startAssistantMessage(context.key);
+            context.state.startAssistantMessage(context.key, context.frameTurnId);
         }
+        if (context.frameTurnId)
+            context.state.stampTurn(context.key, context.frameTurnId);
         context.enqueueDelta(context.key, "text", msg.text);
     },
     thinking_delta: (msg, context) => {
         if (!context.buffer()?.isStreaming) {
-            context.state.startAssistantMessage(context.key);
+            context.state.startAssistantMessage(context.key, context.frameTurnId);
         }
+        if (context.frameTurnId)
+            context.state.stampTurn(context.key, context.frameTurnId);
         context.enqueueDelta(context.key, "thinking", msg.text);
     },
     tool_use_start: (msg, context) => {
@@ -137,6 +151,14 @@ export const chatFrameHandlers = {
         context.state.finishAssistantMessage(context.key);
         context.resyncIfNeeded(msg.sessionId);
     },
+    message_blocks: (msg, context) => {
+        // Arrives after `result`, for the turn's own assistant message. An
+        // answer never waits on it: the markdown is already on screen, and this
+        // only says which spans to draw as blocks (D42). Targeted by the turn
+        // the frame belongs to: a queued follow-up may already have opened a
+        // newer assistant message by the time the pass returns.
+        context.state.setMessageBlocks(context.key, msg.blocks, context.frameTurnId);
+    },
     session_history: (msg, context) => {
         const converted = msg.messages.map(convertHistoryMessage);
         if (msg.append) {
@@ -161,7 +183,7 @@ export const chatFrameHandlers = {
         // THIS frame's session: a background session settling must not consume
         // (or trigger) the active session's pending resync.
         if (msg.status === "idle" || msg.status === "cancelled") {
-            const current = useChatStore.getState();
+            const current = context.stores.chat.getState();
             context.resyncIfNeeded(context.frameSessionId ?? current.activeSessionId);
             // Cold load: we hold a stored session id but an empty transcript and
             // the server is idle (so it sent no snapshot). Fetch the history.

package/dist/hooks/websocket-handlers/chat.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat.js","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAOtE,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAgBxD,IAAI,qBAAqB,GAAG,CAAC,CAAC;AAC9B,SAAS,SAAS;IAChB,OAAO,QAAQ,EAAE,qBAAqB,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,qBAAqB,CAAC,GAA0B;IACvD,0EAA0E;IAC1E,yCAAyC;IACzC,MAAM,KAAK,GACT,GAAG,CAAC,KAAK;QACT;YACE,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,CAAC,QAAQ,EAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,CAAU,CAAC;YACzE,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;SACvE,CAAC;IACJ,MAAM,gBAAgB,GAAG,2BAA2B,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACpE,OAAO;QACL,EAAE,EAAE,SAAS,EAAE;QACf,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAY,EAAE,CAAC,CAAC;YAC9C,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,KAAK,EAAE,EAAE,CAAC,KAAK;YACf,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5C,MAAM,EAAE,EAAE,CAAC,MAAM;YACjB,OAAO,EAAE,EAAE,CAAC,OAAO;YACnB,MAAM,EAAE,UAAU;SACnB,CAAC,CAAC;QACH,KAAK;QACL,WAAW,EAAE,KAAK;QAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACjD,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,GAAG,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACzE,CAAC;AACJ,CAAC;AAED,sEAAsE;AACtE,SAAS,gBAAgB,CAAC,CAAU;IAClC,OAAO,CACL,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,KAAK,QAAQ;QACrB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QACjB,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CACrD,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B,CAClC,SAA6C;IAE7C,MAAM,SAAS,GAAsB,EAAE,CAAC;IACxC,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;QAC3B,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,IAAI,CAAC;YAAE,SAAS;QACtC,MAAM,SAAS,GAAI,EAAE,CAAC,KAAK,EAAE,SAA+B,IAAI,EAAE,CAAC;QACnE,MAAM,QAAQ,GAAoB,EAAE,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,SAAS,EAAE,CAAC;QAClE,IAAI,EAAE,CAAC,MAAM,EAAE,CAAC;YACd,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CAGnC,CAAC;gBACF,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAC9D,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;oBACnC,QAAQ,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;gBAC7C,CAAC;qBAAM,IAAI,gBAAgB,CAAC,OAAO,CAAC,EAAE,CAAC;oBACrC,4DAA4D;oBAC5D,+CAA+C;oBAC/C,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;gBAC7B,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,mEAAmE;gBACnE,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC;YAC5B,CAAC;QACH,CAAC;QACD,IAAI,EAAE,CAAC,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO;YAAE,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC;QAC/D,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC3B,CAAC;IACD,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;AAClD,CAAC;AAED,gFAAgF;AAChF,MAAM,UAAU,gBAAgB,CAAC,GAAkB;IACjD,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,MAAM,CAAC;IACjE,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC1B,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAC7C,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW;YAAE,OAAO,MAAM,CAAC;IACzE,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,UAAU,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC3B,qEAAqE;QACrE,wEAAwE;QACxE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,qBAAqB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC;IACD,cAAc,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC/B,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,qBAAqB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IAC1D,CAAC;IACD,cAAc,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC/B,sEAAsE;QACtE,qEAAqE;QACrE,sDAAsD;QACtD,IAAI,GAAG,CAAC,eAAe;YAAE,OAAO;QAChC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxE,CAAC;IACD,gBAAgB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACjC,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC;IAC9D,CAAC;IACD,iBAAiB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAClC,oEAAoE;QACpE,yDAAyD;QACzD,IAAI,GAAG,CAAC,eAAe;YAAE,OAAO;QAChC,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;IACtF,CAAC;IACD,qBAAqB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACtC,OAAO,CAAC,KAAK,CAAC,mBAAmB,CAC/B,OAAO,CAAC,GAAG,EACX,GAAG,CAAC,SAAS,EACb,GAAG,CAAC,QAAQ,EACZ,GAAG,CAAC,KAAK,EACT,GAAG,CAAC,WAAW,EACf,GAAG,CAAC,IAAI,CACT,CAAC;IACJ,CAAC;IACD,WAAW,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC5B,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;QACjF,oEAAoE;QACpE,kDAAkD;QAClD,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QAC9B,IAAI,IAAI,EAAE,OAAO,EAAE,OAAO,IAAI,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;YACvD,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IACD,gBAAgB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACjC,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7E,CAAC;IACD,MAAM,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACvB,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClD,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IACD,eAAe,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAChC,MAAM,SAAS,GAAG,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QAC1D,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,uEAAuE;YACvE,oEAAoE;YACpE,uEAAuE;YACvE,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YAClD,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IACD,MAAM,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACvB,2DAA2D;QAC3D,IACE,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,CAAC;YACrD,OAAO,CAAC,MAAM,EAAE,EAAE,WAAW,EAC7B,CAAC;YACD,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACpD,CAAC;QACD,mEAAmE;QACnE,yEAAyE;QACzE,uEAAuE;QACvE,oDAAoD;QACpD,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;YACxC,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,eAAe,CAAC,CAAC;YAC1E,qEAAqE;YACrE,kEAAkE;YAClE,qEAAqE;YACrE,IAAI,CAAC,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,cAAc,KAAK,OAAO,CAAC,eAAe,EAAE,CAAC;gBAClF,OAAO,CAAC,kBAAkB,CACxB,OAAO,CAAC,eAAe,EACvB,UAAU,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CACpC,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;CAC2C,CAAC"}
+{"version":3,"file":"chat.js","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/chat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAOxD,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAiBxD,IAAI,qBAAqB,GAAG,CAAC,CAAC;AAC9B,SAAS,SAAS;IAChB,OAAO,QAAQ,EAAE,qBAAqB,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,qBAAqB,CAAC,GAA0B;IACvD,0EAA0E;IAC1E,yCAAyC;IACzC,MAAM,KAAK,GACT,GAAG,CAAC,KAAK;QACT;YACE,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,CAAC,QAAQ,EAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,CAAU,CAAC;YACzE,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;SACvE,CAAC;IACJ,MAAM,gBAAgB,GAAG,2BAA2B,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACpE,OAAO;QACL,EAAE,EAAE,SAAS,EAAE;QACf,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAY,EAAE,CAAC,CAAC;YAC9C,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,KAAK,EAAE,EAAE,CAAC,KAAK;YACf,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5C,MAAM,EAAE,EAAE,CAAC,MAAM;YACjB,OAAO,EAAE,EAAE,CAAC,OAAO;YACnB,MAAM,EAAE,UAAU;SACnB,CAAC,CAAC;QACH,KAAK;QACL,WAAW,EAAE,KAAK;QAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACjD,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,GAAG,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACxE,GAAG,CAAC,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvE,CAAC;AACJ,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,gBAAgB,CAAC,CAAU;IAClC,OAAO,CACL,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,KAAK,QAAQ;QACrB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QACjB,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CACrD,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B,CAClC,SAA6C;IAE7C,MAAM,SAAS,GAAsB,EAAE,CAAC;IACxC,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;QAC3B,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,IAAI,CAAC;YAAE,SAAS;QACtC,MAAM,SAAS,GAAI,EAAE,CAAC,KAAK,EAAE,SAA+B,IAAI,EAAE,CAAC;QACnE,MAAM,QAAQ,GAAoB,EAAE,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,SAAS,EAAE,CAAC;QAClE,IAAI,EAAE,CAAC,MAAM,EAAE,CAAC;YACd,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CAGnC,CAAC;gBACF,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAC9D,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;oBACnC,QAAQ,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;gBAC7C,CAAC;qBAAM,IAAI,gBAAgB,CAAC,OAAO,CAAC,EAAE,CAAC;oBACrC,4DAA4D;oBAC5D,+CAA+C;oBAC/C,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;gBAC7B,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,mEAAmE;gBACnE,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC;YAC5B,CAAC;QACH,CAAC;QACD,IAAI,EAAE,CAAC,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO;YAAE,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC;QAC/D,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC3B,CAAC;IACD,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;AAClD,CAAC;AAED,gFAAgF;AAChF,MAAM,UAAU,gBAAgB,CAAC,GAAkB;IACjD,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,MAAM,CAAC;IACjE,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC1B,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAC7C,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW;YAAE,OAAO,MAAM,CAAC;IACzE,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,UAAU,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC3B,qEAAqE;QACrE,wEAAwE;QACxE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,qBAAqB,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,OAAO,CAAC,WAAW;YAAE,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACnF,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC;IACD,cAAc,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC/B,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,qBAAqB,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,OAAO,CAAC,WAAW;YAAE,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACnF,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IAC1D,CAAC;IACD,cAAc,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC/B,sEAAsE;QACtE,qEAAqE;QACrE,sDAAsD;QACtD,IAAI,GAAG,CAAC,eAAe;YAAE,OAAO;QAChC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxE,CAAC;IACD,gBAAgB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACjC,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC;IAC9D,CAAC;IACD,iBAAiB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAClC,oEAAoE;QACpE,yDAAyD;QACzD,IAAI,GAAG,CAAC,eAAe;YAAE,OAAO;QAChC,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;IACtF,CAAC;IACD,qBAAqB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACtC,OAAO,CAAC,KAAK,CAAC,mBAAmB,CAC/B,OAAO,CAAC,GAAG,EACX,GAAG,CAAC,SAAS,EACb,GAAG,CAAC,QAAQ,EACZ,GAAG,CAAC,KAAK,EACT,GAAG,CAAC,WAAW,EACf,GAAG,CAAC,IAAI,CACT,CAAC;IACJ,CAAC;IACD,WAAW,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC5B,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;QACjF,oEAAoE;QACpE,kDAAkD;QAClD,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QAC9B,IAAI,IAAI,EAAE,OAAO,EAAE,OAAO,IAAI,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;YACvD,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IACD,gBAAgB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACjC,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7E,CAAC;IACD,MAAM,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACvB,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClD,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IACD,cAAc,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC/B,mEAAmE;QACnE,wEAAwE;QACxE,sEAAsE;QACtE,qEAAqE;QACrE,wDAAwD;QACxD,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAC/E,CAAC;IACD,eAAe,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAChC,MAAM,SAAS,GAAG,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QAC1D,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,uEAAuE;YACvE,oEAAoE;YACpE,uEAAuE;YACvE,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YAClD,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IACD,MAAM,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACvB,2DAA2D;QAC3D,IACE,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,CAAC;YACrD,OAAO,CAAC,MAAM,EAAE,EAAE,WAAW,EAC7B,CAAC;YACD,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACpD,CAAC;QACD,mEAAmE;QACnE,yEAAyE;QACzE,uEAAuE;QACvE,oDAAoD;QACpD,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/C,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,eAAe,CAAC,CAAC;YAC1E,qEAAqE;YACrE,kEAAkE;YAClE,qEAAqE;YACrE,IAAI,CAAC,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,cAAc,KAAK,OAAO,CAAC,eAAe,EAAE,CAAC;gBAClF,OAAO,CAAC,kBAAkB,CACxB,OAAO,CAAC,eAAe,EACvB,UAAU,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CACpC,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;CAC2C,CAAC"}

package/dist/hooks/websocket-handlers/connection.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"connection.d.ts","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/connection.ts"],"names":[],"mappings":"AAMA,eAAO,MAAM,uBAAuB;;;;CAuBgB,CAAC"}
+{"version":3,"file":"connection.d.ts","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/connection.ts"],"names":[],"mappings":"AAIA,eAAO,MAAM,uBAAuB;;;;CAuBgB,CAAC"}

package/dist/hooks/websocket-handlers/connection.js

@@ -1,12 +1,10 @@
-import { useActivityStore } from "../../stores/activity-store.js";
-import { useConnectionStore } from "../../stores/connection-store.js";
 export const connectionFrameHandlers = {
     server_hello: (msg, context) => {
-        useActivityStore.getState().setSupported(msg.capabilities?.activity === true);
+        context.stores.activity.getState().setSupported(msg.capabilities?.activity === true);
         // A new hello means a new connection: server-side subscriptions are gone.
-        useActivityStore.getState().resetSubscriptions();
+        context.stores.activity.getState().resetSubscriptions();
         // …and view-owned subscriptions (the Activity index) must re-send too.
-        useActivityStore.getState().bumpConnectionEpoch();
+        context.stores.activity.getState().bumpConnectionEpoch();
         context.ensureActivitySubscription(context.state.activeSessionId);
     },
     location_request: (msg, context) => {
@@ -17,7 +15,7 @@ export const connectionFrameHandlers = {
         // message is streaming, and this used to be the whole handler — so an
         // error arriving between turns (a rejected frame, a failed resume) was
         // dropped as silently on the client as it was on the server.
-        useConnectionStore.getState().reportError(msg.code, msg.message);
+        context.stores.connection.getState().reportError(msg.code, msg.message);
         if (context.buffer()?.isStreaming) {
             context.state.appendText(context.key, `\n\n**Error:** ${msg.message}`);
             context.state.finishAssistantMessage(context.key);

package/dist/hooks/websocket-handlers/connection.js.map

@@ -1 +1 @@
-{"version":3,"file":"connection.js","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/connection.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AAKtE,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACrC,YAAY,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC7B,gBAAgB,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,KAAK,IAAI,CAAC,CAAC;QAC9E,0EAA0E;QAC1E,gBAAgB,CAAC,QAAQ,EAAE,CAAC,kBAAkB,EAAE,CAAC;QACjD,uEAAuE;QACvE,gBAAgB,CAAC,QAAQ,EAAE,CAAC,mBAAmB,EAAE,CAAC;QAClD,OAAO,CAAC,0BAA0B,CAAC,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;IACpE,CAAC;IACD,gBAAgB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACjC,OAAO,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;IACtC,CAAC;IACD,KAAK,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACtB,kEAAkE;QAClE,sEAAsE;QACtE,uEAAuE;QACvE,6DAA6D;QAC7D,kBAAkB,CAAC,QAAQ,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;QACjE,IAAI,OAAO,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC;YAClC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,kBAAkB,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACvE,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;CACiD,CAAC"}
+{"version":3,"file":"connection.js","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/connection.ts"],"names":[],"mappings":"AAIA,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACrC,YAAY,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAC7B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,KAAK,IAAI,CAAC,CAAC;QACrF,0EAA0E;QAC1E,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,kBAAkB,EAAE,CAAC;QACxD,uEAAuE;QACvE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,mBAAmB,EAAE,CAAC;QACzD,OAAO,CAAC,0BAA0B,CAAC,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;IACpE,CAAC;IACD,gBAAgB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACjC,OAAO,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;IACtC,CAAC;IACD,KAAK,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QACtB,kEAAkE;QAClE,sEAAsE;QACtE,uEAAuE;QACvE,6DAA6D;QAC7D,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;QACxE,IAAI,OAAO,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC;YAClC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,kBAAkB,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACvE,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;CACiD,CAAC"}

package/dist/hooks/websocket-handlers/mask.d.ts

@@ -1,4 +1,4 @@
 export declare const maskFrameHandlers: {
-    mask_request: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerMaskRequest) => void;
+    mask_request: (msg: import("@schlessera/brain-ui-sdk/protocol").ServerMaskRequest, context: import("./types.js").DispatchContext) => void;
 };
 //# sourceMappingURL=mask.d.ts.map

package/dist/hooks/websocket-handlers/mask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mask.d.ts","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/mask.ts"],"names":[],"mappings":"AAKA,eAAO,MAAM,iBAAiB;;CAWgB,CAAC"}
+{"version":3,"file":"mask.d.ts","sourceRoot":"","sources":["../../../src/hooks/websocket-handlers/mask.ts"],"names":[],"mappings":"AAIA,eAAO,MAAM,iBAAiB;;CAWgB,CAAC"}