@openma/common 0.0.0-stage → 0.9.0

package/dist/agent-ui/index.js

@@ -0,0 +1,853 @@
+import { reduceWorkItemEvent, } from "../session-events/openma.js";
+export const AGENT_UI_STATE_VERSION = "oma.agent-ui.v1";
+export function createAgentUIState(sessionId) {
+    return {
+        version: AGENT_UI_STATE_VERSION,
+        sessionId,
+        status: "unknown",
+        sessionInfo: {},
+        commands: [],
+        capabilities: {},
+        turnOrder: [],
+        turns: {},
+        workItemOrder: [],
+        workItems: {},
+        planOrder: [],
+        plans: {},
+        callbackOrder: [],
+        callbacks: {},
+        outcomeOrder: [],
+        outcomes: {},
+        seenEventIds: {},
+    };
+}
+function eventData(event) {
+    return event.data && typeof event.data === "object"
+        ? event.data
+        : {};
+}
+function cloneState(state) {
+    return {
+        ...state,
+        sessionInfo: { ...state.sessionInfo },
+        commands: [...state.commands],
+        capabilities: { ...state.capabilities },
+        turnOrder: [...state.turnOrder],
+        turns: Object.fromEntries(Object.entries(state.turns).map(([id, turn]) => [
+            id,
+            {
+                ...turn,
+                items: turn.items.map((item) => item.kind === "tool"
+                    ? {
+                        ...item,
+                        outputs: item.outputs.map((output) => ({ ...output })),
+                        ...(item.content ? { content: [...item.content] } : {}),
+                        ...(item.locations
+                            ? { locations: item.locations.map((location) => ({ ...location })) }
+                            : {}),
+                        ...(item.adapterMeta
+                            ? { adapterMeta: { ...item.adapterMeta } }
+                            : {}),
+                    }
+                    : { ...item }),
+            },
+        ])),
+        workItemOrder: [...state.workItemOrder],
+        workItems: Object.fromEntries(Object.entries(state.workItems).map(([id, item]) => [
+            id,
+            { ...item, output: [...item.output] },
+        ])),
+        planOrder: [...state.planOrder],
+        plans: Object.fromEntries(Object.entries(state.plans).map(([id, plan]) => [
+            id,
+            { ...plan, entries: plan.entries.map((entry) => ({ ...entry })) },
+        ])),
+        callbackOrder: [...state.callbackOrder],
+        callbacks: Object.fromEntries(Object.entries(state.callbacks).map(([id, callback]) => [
+            id,
+            { ...callback },
+        ])),
+        ...(state.usage ? { usage: { ...state.usage } } : {}),
+        outcomeOrder: [...state.outcomeOrder],
+        outcomes: Object.fromEntries(Object.entries(state.outcomes).map(([id, outcome]) => [id, { ...outcome }])),
+        seenEventIds: { ...state.seenEventIds },
+    };
+}
+function toWorkItemSnapshot(item) {
+    return {
+        id: item.id,
+        kind: item.kind,
+        status: item.status,
+        output: [...item.output],
+        ...(item.title !== undefined ? { title: item.title } : {}),
+        ...(item.progress !== undefined ? { progress: item.progress } : {}),
+        ...(item.result !== undefined ? { result: item.result } : {}),
+        ...(item.error !== undefined ? { error: item.error } : {}),
+        ...(item.reason !== undefined ? { reason: item.reason } : {}),
+        ...(item.startedAt !== undefined ? { started_at: item.startedAt } : {}),
+        ...(item.endedAt !== undefined ? { ended_at: item.endedAt } : {}),
+        ...(item.missingStart !== undefined ? { missing_start: item.missingStart } : {}),
+        ...(item.missingTerminal !== undefined
+            ? { missing_terminal: item.missingTerminal }
+            : {}),
+    };
+}
+function toAgentUIWorkItem(item) {
+    return {
+        id: item.id,
+        kind: item.kind,
+        status: item.status,
+        output: [...item.output],
+        ...(item.title !== undefined ? { title: item.title } : {}),
+        ...(item.progress !== undefined ? { progress: item.progress } : {}),
+        ...(item.result !== undefined ? { result: item.result } : {}),
+        ...(item.error !== undefined ? { error: item.error } : {}),
+        ...(item.reason !== undefined ? { reason: item.reason } : {}),
+        ...(item.started_at !== undefined ? { startedAt: item.started_at } : {}),
+        ...(item.ended_at !== undefined ? { endedAt: item.ended_at } : {}),
+        ...(item.missing_start !== undefined ? { missingStart: item.missing_start } : {}),
+        ...(item.missing_terminal !== undefined
+            ? { missingTerminal: item.missing_terminal }
+            : {}),
+    };
+}
+function applyWorkItemProjection(state, previous, event) {
+    const registry = reduceWorkItemEvent({
+        items: new Map(Object.entries(previous.workItems).map(([id, item]) => [
+            id,
+            toWorkItemSnapshot(item),
+        ])),
+        seen_event_ids: new Set(Object.keys(previous.seenEventIds)),
+    }, event);
+    const ids = [...registry.items.keys()];
+    const present = new Set(ids);
+    state.workItemOrder = previous.workItemOrder.filter((id) => present.has(id));
+    for (const id of ids) {
+        if (!state.workItemOrder.includes(id))
+            state.workItemOrder.push(id);
+    }
+    state.workItems = Object.fromEntries([...registry.items].map(([id, item]) => [id, toAgentUIWorkItem(item)]));
+}
+function planId(event) {
+    if (!event.data || typeof event.data !== "object")
+        return undefined;
+    const value = event.data.plan_id;
+    return typeof value === "string" && value.length > 0
+        ? value
+        : event.turn_id;
+}
+function mergePlanEntries(previous, incoming) {
+    const merged = previous.map((entry) => ({ ...entry }));
+    for (const entry of incoming) {
+        const index = entry.id
+            ? merged.findIndex((candidate) => candidate.id === entry.id)
+            : -1;
+        if (index >= 0)
+            merged[index] = { ...merged[index], ...entry };
+        else
+            merged.push({ ...entry });
+    }
+    return merged;
+}
+function applyPlanProjection(state, event) {
+    const id = planId(event);
+    if (!id)
+        return;
+    if (event.type === "plan.updated") {
+        const data = event.data;
+        if (data.representation !== "items"
+            && data.representation !== "markdown"
+            && data.representation !== "file") {
+            return;
+        }
+        const previous = state.plans[id];
+        const incoming = data.entries ?? [];
+        state.plans[id] = {
+            id,
+            representation: data.representation,
+            status: "active",
+            entries: data.update_mode === "merge" && previous
+                ? mergePlanEntries(previous.entries, incoming)
+                : incoming.map((entry) => ({ ...entry })),
+        };
+        if (!state.planOrder.includes(id))
+            state.planOrder.push(id);
+    }
+    else if (event.type === "plan.completed") {
+        const previous = state.plans[id];
+        if (previous)
+            state.plans[id] = { ...previous, status: "completed" };
+    }
+    else if (event.type === "plan.removed") {
+        delete state.plans[id];
+        state.planOrder = state.planOrder.filter((planId) => planId !== id);
+    }
+}
+function applyCallbackProjection(state, event) {
+    if (event.type !== "callback.requested"
+        && event.type !== "callback.completed"
+        && event.type !== "callback.failed"
+        && event.type !== "callback.notification") {
+        return;
+    }
+    if (!event.data || typeof event.data !== "object")
+        return;
+    const data = event.data;
+    const callbackId = (typeof data.callback_id === "string" && data.callback_id.length > 0) || typeof data.callback_id === "number"
+        ? String(data.callback_id)
+        : data.callback_id === null
+            ? "null"
+            : event.type === "callback.notification"
+                ? event.event_id
+                : undefined;
+    if (!callbackId) {
+        return;
+    }
+    if (typeof data.method !== "string" || data.method.length === 0)
+        return;
+    const id = callbackId;
+    const previous = state.callbacks[id];
+    const callback = {
+        id,
+        category: data.category,
+        method: data.method,
+        status: event.type === "callback.requested"
+            ? "pending"
+            : event.type === "callback.failed"
+                ? "failed"
+                : event.type === "callback.notification"
+                    ? "notification"
+                    : "completed",
+        ...(previous?.params !== undefined ? { params: previous.params } : {}),
+        ...(data.params !== undefined ? { params: data.params } : {}),
+        ...(data.result !== undefined ? { result: data.result } : {}),
+        ...(data.error !== undefined ? { error: data.error } : {}),
+    };
+    state.callbacks[id] = callback;
+    if (!state.callbackOrder.includes(id))
+        state.callbackOrder.push(id);
+}
+function applyOutcomeProjection(state, event) {
+    if (event.type !== "outcome.defined"
+        && event.type !== "outcome.evaluation_started"
+        && event.type !== "outcome.evaluation_progress"
+        && event.type !== "outcome.evaluation_completed") {
+        return;
+    }
+    if (!event.data || typeof event.data !== "object")
+        return;
+    const data = event.data;
+    const id = typeof data.outcome_id === "string" && data.outcome_id.length > 0
+        ? data.outcome_id
+        : event.event_id;
+    const previous = state.outcomes[id];
+    const status = event.type === "outcome.defined"
+        ? "defined"
+        : event.type === "outcome.evaluation_completed"
+            ? data.result ?? "completed"
+            : "evaluating";
+    state.outcomes[id] = {
+        id,
+        status,
+        ...(previous?.description !== undefined
+            ? { description: previous.description }
+            : {}),
+        ...(previous?.rubric !== undefined ? { rubric: previous.rubric } : {}),
+        ...(previous?.maxIterations !== undefined
+            ? { maxIterations: previous.maxIterations }
+            : {}),
+        ...(data.description !== undefined ? { description: data.description } : {}),
+        ...(data.rubric !== undefined ? { rubric: data.rubric } : {}),
+        ...(data.max_iterations !== undefined
+            ? { maxIterations: data.max_iterations }
+            : {}),
+        ...(data.iteration !== undefined ? { iteration: data.iteration } : {}),
+        ...(data.explanation !== undefined ? { explanation: data.explanation } : {}),
+        ...(data.usage !== undefined ? { usage: data.usage } : {}),
+        updatedAt: event.occurred_at,
+    };
+    if (!state.outcomeOrder.includes(id))
+        state.outcomeOrder.push(id);
+}
+function requireTurn(state, event) {
+    const turnId = event.turn_id;
+    if (!turnId)
+        return undefined;
+    const existing = state.turns[turnId];
+    if (existing)
+        return existing;
+    const created = {
+        id: turnId,
+        status: "unknown",
+        items: [],
+    };
+    state.turns[turnId] = created;
+    state.turnOrder.push(turnId);
+    return created;
+}
+function upsertMessage(turn, event, input) {
+    const data = eventData(event);
+    const id = typeof data.message_id === "string" && data.message_id.length > 0
+        ? data.message_id
+        : event.event_id;
+    const text = typeof data.text === "string" ? data.text : "";
+    const tail = turn.items.at(-1);
+    const existing = tail
+        && tail.kind !== "tool"
+        && tail.kind === input.kind
+        && tail.role === input.role
+        && (data.message_id === undefined
+            || (tail.messageId ?? tail.id) === data.message_id)
+        ? tail
+        : undefined;
+    if (existing) {
+        existing.kind = input.kind;
+        existing.role = input.role;
+        existing.text = input.streaming
+            ? mergeAgentUIStreamingText(existing.text, text)
+            : text;
+        existing.status = input.streaming ? "streaming" : "complete";
+        if (data.content !== undefined)
+            existing.content = data.content;
+        if (data.phase !== undefined)
+            existing.phase = data.phase;
+        return;
+    }
+    const segmentId = uniqueMessageItemId(turn, id);
+    turn.items.push({
+        id: segmentId,
+        ...(segmentId !== id ? { messageId: id } : {}),
+        kind: input.kind,
+        ...(input.kind === "thinking" ? { startedAt: event.occurred_at } : {}),
+        role: input.role,
+        text,
+        status: input.streaming ? "streaming" : "complete",
+        ...(data.content !== undefined ? { content: data.content } : {}),
+        ...(data.phase !== undefined ? { phase: data.phase } : {}),
+    });
+}
+function uniqueMessageItemId(turn, sourceId) {
+    if (!turn.items.some((item) => item.id === sourceId))
+        return sourceId;
+    let segment = 2;
+    while (turn.items.some((item) => item.id === `${sourceId}:segment:${segment}`)) {
+        segment += 1;
+    }
+    return `${sourceId}:segment:${segment}`;
+}
+function mergeAdapterMeta(current, incoming) {
+    const next = { ...(current ?? {}) };
+    for (const [key, value] of Object.entries(incoming)) {
+        const previous = next[key];
+        next[key] = value
+            && typeof value === "object"
+            && !Array.isArray(value)
+            && previous
+            && typeof previous === "object"
+            && !Array.isArray(previous)
+            ? mergeAdapterMeta(previous, value)
+            : value;
+    }
+    return next;
+}
+const MIN_STREAM_OVERLAP = 8;
+export function mergeAgentUIStreamingText(accumulated, incoming) {
+    if (!accumulated)
+        return incoming;
+    if (!incoming || incoming === accumulated)
+        return accumulated;
+    if (incoming.startsWith(accumulated))
+        return incoming;
+    if (incoming.length >= MIN_STREAM_OVERLAP && accumulated.endsWith(incoming)) {
+        return accumulated;
+    }
+    const maxOverlap = Math.min(accumulated.length, incoming.length);
+    for (let size = maxOverlap; size >= MIN_STREAM_OVERLAP; size -= 1) {
+        if (accumulated.endsWith(incoming.slice(0, size))) {
+            return accumulated + incoming.slice(size);
+        }
+    }
+    return accumulated + incoming;
+}
+function upsertTool(turn, event, status) {
+    if (!event.data || typeof event.data !== "object")
+        return;
+    const data = event.data;
+    if (typeof data.tool_call_id !== "string" || data.tool_call_id.length === 0) {
+        return;
+    }
+    let item = turn.items.find((candidate) => candidate.kind === "tool" && candidate.id === data.tool_call_id);
+    if (!item) {
+        item = {
+            id: data.tool_call_id,
+            kind: "tool",
+            status,
+            outputs: [],
+        };
+        turn.items.push(item);
+    }
+    item.status = status;
+    if (data.tool_name !== undefined)
+        item.name = data.tool_name;
+    if (data.title !== undefined)
+        item.title = data.title;
+    if (data.kind !== undefined)
+        item.toolKind = data.kind;
+    if (data.raw_input !== undefined)
+        item.rawInput = data.raw_input;
+    if (data.raw_output !== undefined)
+        item.rawOutput = data.raw_output;
+    if (data.content !== undefined)
+        item.content = [...data.content];
+    if (data.locations !== undefined) {
+        item.locations = data.locations.map((location) => ({ ...location }));
+    }
+    if (data.adapter_meta !== undefined) {
+        item.adapterMeta = mergeAdapterMeta(item.adapterMeta, data.adapter_meta);
+    }
+    if (data.error !== undefined)
+        item.error = data.error;
+    if (data.reason !== undefined)
+        item.reason = data.reason;
+    if (data.output !== undefined) {
+        if (data.output.append)
+            item.outputs.push(data.output);
+        else
+            item.outputs = [data.output];
+    }
+}
+function errorMessage(event) {
+    if (!event.data || typeof event.data !== "object")
+        return undefined;
+    const data = event.data;
+    for (const value of [data.error, data.message, data.reason]) {
+        if (typeof value === "string" && value.length > 0)
+            return value;
+    }
+    return undefined;
+}
+function stopReasonType(event) {
+    if (!event.data || typeof event.data !== "object")
+        return undefined;
+    const stopReason = event.data.stop_reason;
+    if (typeof stopReason === "string")
+        return stopReason;
+    if (!stopReason || typeof stopReason !== "object")
+        return undefined;
+    const type = stopReason.type;
+    return typeof type === "string" ? type : undefined;
+}
+/** Match ACP chat projection: first thought chunk through the following
+ * activity (or turn termination), rounded up to whole seconds. Metadata and
+ * unrelated turns must not stop a thought's clock. */
+function settleThoughtTiming(turn, event) {
+    if (!turn)
+        return;
+    const thought = [...turn.items].reverse().find((item) => item.kind === "thinking" && !item.endedAt);
+    if (!thought?.startedAt)
+        return;
+    const data = eventData(event);
+    const continuesThought = event.type === "agent.thinking"
+        && turn.items.at(-1) === thought
+        && (data.message_id === undefined || data.message_id === (thought.messageId ?? thought.id));
+    if (continuesThought)
+        return;
+    const boundary = event.type === "agent.thinking"
+        || event.type === "agent.message_chunk"
+        || event.type === "agent.message"
+        || event.type.startsWith("tool.")
+        || ["turn.completed", "turn.failed", "turn.cancelled", "session.error", "session.terminated"].includes(event.type)
+        || (event.type === "session.idle" && stopReasonType(event) === "end_turn");
+    if (!boundary)
+        return;
+    const elapsed = Date.parse(event.occurred_at) - Date.parse(thought.startedAt);
+    if (!Number.isFinite(elapsed))
+        return;
+    thought.endedAt = event.occurred_at;
+    thought.content = {
+        ...(thought.content && typeof thought.content === "object" && !Array.isArray(thought.content)
+            ? thought.content : {}),
+        durationSeconds: Math.max(1, Math.ceil(elapsed / 1_000)),
+    };
+}
+export function reduceAgentUIEvent(state, event) {
+    if (event.session_id !== state.sessionId || state.seenEventIds[event.event_id]) {
+        return state;
+    }
+    const next = cloneState(state);
+    applyWorkItemProjection(next, state, event);
+    applyPlanProjection(next, event);
+    applyCallbackProjection(next, event);
+    applyOutcomeProjection(next, event);
+    next.seenEventIds[event.event_id] = true;
+    const turn = requireTurn(next, event);
+    settleThoughtTiming(turn ?? (event.type.startsWith("session.") && next.activeTurnId ? next.turns[next.activeTurnId] : undefined), event);
+    switch (event.type) {
+        case "session.updated": {
+            const data = event.data;
+            next.sessionInfo = {
+                ...next.sessionInfo,
+                ...(Object.prototype.hasOwnProperty.call(data, "title")
+                    ? { title: data.title }
+                    : {}),
+                ...(Object.prototype.hasOwnProperty.call(data, "updated_at")
+                    ? { updatedAt: data.updated_at }
+                    : {}),
+            };
+            break;
+        }
+        case "command_catalog.updated": {
+            const data = event.data;
+            if (Array.isArray(data.commands))
+                next.commands = [...data.commands];
+            break;
+        }
+        case "capability.updated": {
+            const data = event.data;
+            if (typeof data.capability === "string" && data.capability.length > 0) {
+                next.capabilities[data.capability] = data.value;
+            }
+            break;
+        }
+        case "session.running":
+            next.status = "running";
+            if (turn) {
+                turn.status = "running";
+                turn.startedAt ??= event.occurred_at;
+                next.activeTurnId = turn.id;
+            }
+            break;
+        case "session.rescheduled":
+            next.status = "rescheduled";
+            break;
+        case "session.idle": {
+            next.status = "idle";
+            const idleTurn = turn
+                ?? (next.activeTurnId ? next.turns[next.activeTurnId] : undefined);
+            if (idleTurn && stopReasonType(event) === "end_turn") {
+                idleTurn.status = "completed";
+                idleTurn.endedAt = event.occurred_at;
+            }
+            next.activeTurnId = undefined;
+            break;
+        }
+        case "session.terminated":
+            next.status = "terminated";
+            next.activeTurnId = undefined;
+            break;
+        case "session.error": {
+            const message = errorMessage(event);
+            const failedTurn = turn
+                ?? (next.activeTurnId ? next.turns[next.activeTurnId] : undefined);
+            next.status = "error";
+            next.lastError = message;
+            if (failedTurn) {
+                failedTurn.status = "failed";
+                failedTurn.error = message;
+                failedTurn.endedAt = event.occurred_at;
+            }
+            next.activeTurnId = undefined;
+            break;
+        }
+        case "usage.updated": {
+            const data = event.data;
+            if (data.usage !== undefined) {
+                next.usage = {
+                    snapshot: data.usage,
+                    ...(data.budget !== undefined ? { budget: data.budget } : {}),
+                    updatedAt: event.occurred_at,
+                };
+            }
+            break;
+        }
+        case "turn.queued":
+            if (turn)
+                turn.status = "queued";
+            break;
+        case "turn.completed":
+            if (turn) {
+                turn.status = "completed";
+                turn.endedAt = event.occurred_at;
+            }
+            if (next.activeTurnId === event.turn_id)
+                next.activeTurnId = undefined;
+            break;
+        case "turn.failed":
+            if (turn) {
+                turn.status = "failed";
+                turn.error = errorMessage(event);
+                turn.endedAt = event.occurred_at;
+            }
+            if (next.activeTurnId === event.turn_id)
+                next.activeTurnId = undefined;
+            break;
+        case "turn.cancelled":
+            if (turn) {
+                turn.status = "cancelled";
+                turn.reason = errorMessage(event);
+                turn.endedAt = event.occurred_at;
+            }
+            if (next.activeTurnId === event.turn_id)
+                next.activeTurnId = undefined;
+            break;
+        case "user.message_chunk":
+            if (turn) {
+                if (turn.status === "unknown")
+                    turn.status = "queued";
+                upsertMessage(turn, event, {
+                    kind: "message",
+                    role: "user",
+                    streaming: true,
+                });
+            }
+            break;
+        case "user.message":
+            if (turn) {
+                if (turn.status === "unknown")
+                    turn.status = "queued";
+                upsertMessage(turn, event, {
+                    kind: "message",
+                    role: "user",
+                    streaming: false,
+                });
+            }
+            break;
+        case "agent.message_chunk":
+            if (turn) {
+                if (turn.status === "unknown" || turn.status === "queued") {
+                    turn.status = "running";
+                }
+                upsertMessage(turn, event, {
+                    kind: "message",
+                    role: "assistant",
+                    streaming: true,
+                });
+            }
+            break;
+        case "agent.message":
+            if (turn) {
+                upsertMessage(turn, event, {
+                    kind: "message",
+                    role: "assistant",
+                    streaming: false,
+                });
+            }
+            break;
+        case "agent.thinking":
+            if (turn) {
+                const data = eventData(event);
+                upsertMessage(turn, event, {
+                    kind: "thinking",
+                    role: "assistant",
+                    streaming: typeof data.text === "string" && data.text.length > 0,
+                });
+            }
+            break;
+        case "system.message":
+            if (turn) {
+                upsertMessage(turn, event, {
+                    kind: "message",
+                    role: "system",
+                    streaming: false,
+                });
+            }
+            break;
+        case "system.notice":
+            if (turn) {
+                upsertMessage(turn, event, {
+                    kind: "notice",
+                    role: "system",
+                    streaming: false,
+                });
+            }
+            break;
+        case "tool.started":
+            if (turn)
+                upsertTool(turn, event, "in_progress");
+            break;
+        case "tool.progress":
+            if (turn)
+                upsertTool(turn, event, "in_progress");
+            break;
+        case "tool.completed":
+            if (turn)
+                upsertTool(turn, event, "completed");
+            break;
+        case "tool.failed":
+            if (turn)
+                upsertTool(turn, event, "failed");
+            break;
+        case "tool.cancelled":
+            if (turn)
+                upsertTool(turn, event, "cancelled");
+            break;
+        case "raw.event":
+        case "vendor.event":
+            if (turn)
+                turn.items.push({ id: event.event_id, kind: "raw", event });
+            break;
+    }
+    return next;
+}
+export function replayAgentUIEvents(sessionId, events) {
+    const ordered = events
+        .map((event, index) => ({ event, index }))
+        .sort((left, right) => {
+        if (left.event.seq === undefined && right.event.seq === undefined) {
+            return left.index - right.index;
+        }
+        if (left.event.seq === undefined)
+            return 1;
+        if (right.event.seq === undefined)
+            return -1;
+        return left.event.seq - right.event.seq || left.index - right.index;
+    });
+    return ordered.reduce((state, entry) => reduceAgentUIEvent(state, entry.event), createAgentUIState(sessionId));
+}
+/** Rebuild a live Agent UI store from persisted host events. Stable seq
+ * ordering and the store's event-id dedupe make repeated hydration safe. */
+export function replayAgentUIEventLog(store, rows, decode) {
+    const ordered = rows
+        .map((row, index) => ({ row, index }))
+        .sort((left, right) => left.row.seq - right.row.seq || left.index - right.index);
+    for (const { row } of ordered) {
+        const decoded = decode(row);
+        if (!decoded)
+            continue;
+        for (const event of Array.isArray(decoded) ? decoded : [decoded]) {
+            store.dispatch(event);
+        }
+    }
+    return store.getState();
+}
+export function createAgentUISessionRegistry() {
+    const stores = new Map();
+    return {
+        get(sessionId) {
+            const existing = stores.get(sessionId);
+            if (existing)
+                return existing;
+            const created = createAgentUIStore(sessionId);
+            stores.set(sessionId, created);
+            return created;
+        },
+        has(sessionId) {
+            return stores.has(sessionId);
+        },
+        remove(sessionId) {
+            return stores.delete(sessionId);
+        },
+        clear() {
+            stores.clear();
+        },
+    };
+}
+export function createAgentUIStore(sessionId) {
+    let state = createAgentUIState(sessionId);
+    const listeners = new Set();
+    const streamSubscribers = new Map();
+    const getState = () => state;
+    return {
+        getState,
+        getSnapshot: getState,
+        dispatch: (event) => {
+            const stream = streamProjection(event);
+            const beforeText = stream
+                ? accumulatedTurnStream(state, event.turn_id, stream.kind)
+                : "";
+            const opensSegment = stream ? opensStreamSegment(state, event, stream.kind) : false;
+            const next = reduceAgentUIEvent(state, event);
+            if (next !== state) {
+                state = next;
+                if (stream) {
+                    const afterText = accumulatedTurnStream(state, event.turn_id, stream.kind);
+                    const delta = afterText.startsWith(beforeText)
+                        ? afterText.slice(beforeText.length)
+                        : "";
+                    if (delta && event.turn_id) {
+                        for (const subscriber of streamSubscribers.get(event.turn_id) ?? []) {
+                            subscriber({ kind: stream.kind, text: delta });
+                        }
+                    }
+                }
+                if (!stream || opensSegment) {
+                    for (const listener of listeners)
+                        listener(state);
+                }
+            }
+            return state;
+        },
+        subscribe: (listener) => {
+            listeners.add(listener);
+            return () => listeners.delete(listener);
+        },
+        subscribeTurnStream: (turnId, listener) => {
+            let subscribers = streamSubscribers.get(turnId);
+            if (!subscribers) {
+                subscribers = new Set();
+                streamSubscribers.set(turnId, subscribers);
+            }
+            subscribers.add(listener);
+            const thought = accumulatedTurnStream(state, turnId, "thought");
+            if (thought)
+                listener({ kind: "thought", text: thought });
+            const assistant = accumulatedTurnStream(state, turnId, "assistant");
+            if (assistant)
+                listener({ kind: "assistant", text: assistant });
+            return () => {
+                const current = streamSubscribers.get(turnId);
+                if (!current)
+                    return;
+                current.delete(listener);
+                if (current.size === 0)
+                    streamSubscribers.delete(turnId);
+            };
+        },
+    };
+}
+function streamProjection(event) {
+    if (event.type === "agent.message_chunk")
+        return { kind: "assistant" };
+    if (event.type === "agent.thinking") {
+        const data = eventData(event);
+        return typeof data.text === "string" && data.text.length > 0
+            ? { kind: "thought" }
+            : undefined;
+    }
+    return undefined;
+}
+function accumulatedTurnStream(state, turnId, kind) {
+    if (!turnId)
+        return "";
+    const turn = state.turns[turnId];
+    if (!turn)
+        return "";
+    return turn.items.flatMap((item) => {
+        if (item.kind === "tool")
+            return [];
+        if (kind === "thought")
+            return item.kind === "thinking" ? [item.text] : [];
+        return item.kind === "message" && item.role === "assistant" ? [item.text] : [];
+    }).join("");
+}
+function opensStreamSegment(state, event, kind) {
+    if (!event.turn_id)
+        return false;
+    const tail = state.turns[event.turn_id]?.items.at(-1);
+    if (!tail || tail.kind === "tool")
+        return true;
+    if (kind === "thought") {
+        if (tail.kind !== "thinking")
+            return true;
+    }
+    else if (tail.kind !== "message" || tail.role !== "assistant") {
+        return true;
+    }
+    const data = eventData(event);
+    if (typeof data.message_id !== "string" || data.message_id.length === 0) {
+        return false;
+    }
+    return (tail.messageId ?? tail.id) !== data.message_id;
+}
+export * from "./presentation.js";
+//# sourceMappingURL=index.js.map