@otto-assistant/otto 0.1.1 → 0.7.15

package/dist/session-handler/event-stream-state.js

@@ -0,0 +1,475 @@
+// Pure event-stream derivation functions for session lifecycle state.
+// These functions derive lifecycle decisions from an event buffer array.
+// Zero imports from thread-session-runtime.ts, store.ts, or state.ts.
+// Only types from @opencode-ai/sdk/v2 and the getOpencodeEventSessionId helper.
+import { getOpencodeEventSessionId } from './opencode-session-event-log.js';
+export function getEventBufferSessionId(event) {
+    if (event.type === 'queue.question-handoff-started') {
+        return event.properties.sessionID;
+    }
+    return getOpencodeEventSessionId(event);
+}
+function getTaskChildSessionId({ part, }) {
+    // Event-shape reference:
+    // - cli/src/session-handler/event-stream-fixtures/real-session-task-three-parallel-sleeps.jsonl
+    // - In real task events, state.metadata.sessionId appears on running/completed
+    //   tool updates and is the canonical child-session identifier.
+    // We intentionally do not parse state.output because it is user-facing text
+    // and can change format across providers/versions.
+    const metadataValue = part.state.metadata;
+    const metadataSessionId = metadataValue && typeof metadataValue === 'object'
+        ? metadataValue.sessionId
+        : undefined;
+    if (typeof metadataSessionId === 'string' && metadataSessionId.length > 0) {
+        return metadataSessionId;
+    }
+    return undefined;
+}
+function getTaskCandidateFromEvent({ event, mainSessionId, }) {
+    if (event.type !== 'message.part.updated') {
+        return undefined;
+    }
+    const part = event.properties.part;
+    if (part.sessionID !== mainSessionId) {
+        return undefined;
+    }
+    if (part.type !== 'tool' || part.tool !== 'task' || part.state.status === 'pending') {
+        return undefined;
+    }
+    const childSessionId = getTaskChildSessionId({ part });
+    if (!childSessionId) {
+        return undefined;
+    }
+    const subagentType = part.state.input?.subagent_type;
+    const description = part.state.input?.description;
+    return {
+        assistantMessageId: part.messageID,
+        childSessionId,
+        subagentType: typeof subagentType === 'string' ? subagentType : undefined,
+        description: typeof description === 'string' ? description : undefined,
+    };
+}
+// Scans backward for most recent session-scoped lifecycle event.
+// Returns true if the latest lifecycle event for sessionId is session.status busy.
+export function isSessionBusy({ events, sessionId, upToIndex, }) {
+    const end = upToIndex ?? events.length - 1;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const e = entry.event;
+        const eid = getEventBufferSessionId(e);
+        if (eid !== sessionId) {
+            continue;
+        }
+        if (e.type === 'session.idle') {
+            return false;
+        }
+        if (e.type === 'session.status') {
+            return e.properties.status.type === 'busy';
+        }
+    }
+    return false;
+}
+export function didQuestionQueueHandoffSinceLatestQuestionAsked({ events, sessionId, upToIndex, }) {
+    const end = upToIndex ?? events.length - 1;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const event = entry.event;
+        const eventSessionId = getEventBufferSessionId(event);
+        if (eventSessionId !== sessionId) {
+            continue;
+        }
+        if (event.type === 'queue.question-handoff-started') {
+            return true;
+        }
+        if (event.type === 'question.asked') {
+            return false;
+        }
+    }
+    return false;
+}
+export function isAssistantMessageNaturalCompletion({ message, }) {
+    if (typeof message.time.completed !== 'number') {
+        return false;
+    }
+    if (message.error) {
+        return false;
+    }
+    // finish="tool-calls" means the model's last step was tool execution.
+    // Mid-turn tool-call steps don't get footers — the footer comes from the
+    // final text response (finish="stop") that follows. If the turn ends with
+    // only tool-calls and no text follow-up, no footer is emitted. This is
+    // acceptable since models almost always follow up with text after tools.
+    return message.finish !== 'tool-calls';
+}
+export function hasAssistantMessageCompletedBefore({ events, sessionId, messageId, upToIndex, }) {
+    const end = upToIndex ?? events.length - 1;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const event = entry.event;
+        if (event.type !== 'message.updated') {
+            continue;
+        }
+        const info = event.properties.info;
+        if (info.sessionID !== sessionId || info.role !== 'assistant' || info.id !== messageId) {
+            continue;
+        }
+        return typeof info.time.completed === 'number';
+    }
+    return false;
+}
+export function getLatestUserMessage({ events, sessionId, upToIndex, }) {
+    const end = upToIndex ?? events.length - 1;
+    let latestUserMessage;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const event = entry.event;
+        if (event.type !== 'message.updated') {
+            continue;
+        }
+        const info = event.properties.info;
+        if (info.sessionID !== sessionId || info.role !== 'user') {
+            continue;
+        }
+        if (!latestUserMessage) {
+            latestUserMessage = info;
+            continue;
+        }
+        if (info.time.created > latestUserMessage.time.created) {
+            latestUserMessage = info;
+        }
+    }
+    return latestUserMessage;
+}
+export function getCurrentTurnStartTime({ events, sessionId, upToIndex, }) {
+    const latestUserMessage = getLatestUserMessage({
+        events,
+        sessionId,
+        upToIndex,
+    });
+    return latestUserMessage?.time.created;
+}
+// Token total helper — sum of input + output + reasoning + cache.read + cache.write
+function getTokenTotal(tokens) {
+    return tokens.input + tokens.output + tokens.reasoning + tokens.cache.read + tokens.cache.write;
+}
+// Scans backward for most recent message.updated with role=assistant for sessionId.
+// Extracts model, providerID, agent, tokensUsed.
+export function getLatestRunInfo({ events, sessionId, upToIndex, }) {
+    const result = {
+        model: undefined,
+        providerID: undefined,
+        agent: undefined,
+        tokensUsed: 0,
+    };
+    const end = upToIndex ?? events.length - 1;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const e = entry.event;
+        if (e.type !== 'message.updated') {
+            continue;
+        }
+        const msg = e.properties.info;
+        if (msg.sessionID !== sessionId || msg.role !== 'assistant') {
+            continue;
+        }
+        return {
+            model: 'modelID' in msg ? msg.modelID : undefined,
+            providerID: 'providerID' in msg ? msg.providerID : undefined,
+            agent: 'mode' in msg ? msg.mode : undefined,
+            tokensUsed: 'tokens' in msg && msg.tokens
+                ? getTokenTotal(msg.tokens)
+                : 0,
+        };
+    }
+    return result;
+}
+export function getAssistantMessageIdsForLatestUserTurn({ events, sessionId, upToIndex, }) {
+    const latestUserMessage = getLatestUserMessage({
+        events,
+        sessionId,
+        upToIndex,
+    });
+    if (!latestUserMessage) {
+        return new Set();
+    }
+    const end = upToIndex === undefined ? events.length : upToIndex + 1;
+    const assistantMessageIds = new Set();
+    for (let i = 0; i < end; i++) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const e = entry.event;
+        if (e.type !== 'message.updated') {
+            continue;
+        }
+        const msg = e.properties.info;
+        if (msg.sessionID !== sessionId || msg.role !== 'assistant') {
+            continue;
+        }
+        if (msg.parentID === latestUserMessage.id) {
+            assistantMessageIds.add(msg.id);
+        }
+    }
+    return assistantMessageIds;
+}
+export function getLatestAssistantMessageIdForLatestUserTurn({ events, sessionId, upToIndex, }) {
+    const latestUserMessage = getLatestUserMessage({
+        events,
+        sessionId,
+        upToIndex,
+    });
+    if (!latestUserMessage) {
+        return undefined;
+    }
+    const end = upToIndex ?? events.length - 1;
+    let latestAssistantMessage;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const event = entry.event;
+        if (event.type !== 'message.updated') {
+            continue;
+        }
+        const info = event.properties.info;
+        if (info.sessionID !== sessionId || info.role !== 'assistant') {
+            continue;
+        }
+        if (info.parentID !== latestUserMessage.id) {
+            continue;
+        }
+        if (!latestAssistantMessage) {
+            latestAssistantMessage = info;
+            continue;
+        }
+        if (info.time.created > latestAssistantMessage.time.created) {
+            latestAssistantMessage = info;
+        }
+    }
+    return latestAssistantMessage?.id;
+}
+function hasRenderablePartSummary(message) {
+    if (!('partsSummary' in message) || !Array.isArray(message.partsSummary)) {
+        return false;
+    }
+    return message.partsSummary.some((part) => {
+        return part.type === 'text' || part.type === 'tool';
+    });
+}
+function hasAssistantPartEvidence({ events, sessionId, messageId, upToIndex, }) {
+    const end = upToIndex ?? events.length - 1;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const event = entry.event;
+        if (event.type === 'message.updated') {
+            const info = event.properties.info;
+            if (info.sessionID !== sessionId || info.role !== 'assistant' || info.id !== messageId) {
+                continue;
+            }
+            if (hasRenderablePartSummary(info)) {
+                return true;
+            }
+            continue;
+        }
+        if (event.type !== 'message.part.updated') {
+            continue;
+        }
+        const { part } = event.properties;
+        if (part.messageID !== messageId) {
+            continue;
+        }
+        if (part.type === 'text' || part.type === 'tool') {
+            return true;
+        }
+    }
+    return false;
+}
+function hasAssistantStepFinished({ events, messageId, upToIndex, }) {
+    const end = upToIndex ?? events.length - 1;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry || entry.event.type !== 'message.part.updated') {
+            continue;
+        }
+        const { part } = entry.event.properties;
+        if (part.messageID !== messageId) {
+            continue;
+        }
+        if (part.type === 'step-finish') {
+            return true;
+        }
+    }
+    return false;
+}
+export function doesLatestUserTurnHaveNaturalCompletion({ events, sessionId, upToIndex, }) {
+    const latestAssistantMessageId = getLatestAssistantMessageIdForLatestUserTurn({
+        events,
+        sessionId,
+        upToIndex,
+    });
+    if (!latestAssistantMessageId) {
+        return false;
+    }
+    const end = upToIndex ?? events.length - 1;
+    let latestAssistantMessage;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const event = entry.event;
+        if (event.type !== 'message.updated') {
+            continue;
+        }
+        const info = event.properties.info;
+        if (info.sessionID !== sessionId || info.role !== 'assistant') {
+            continue;
+        }
+        if (info.id !== latestAssistantMessageId) {
+            continue;
+        }
+        latestAssistantMessage = info;
+        if (isAssistantMessageNaturalCompletion({ message: info })) {
+            return true;
+        }
+        break;
+    }
+    if (!latestAssistantMessage) {
+        return false;
+    }
+    if (latestAssistantMessage.error) {
+        return false;
+    }
+    if (latestAssistantMessage.finish === 'tool-calls') {
+        return false;
+    }
+    return hasAssistantStepFinished({
+        events,
+        messageId: latestAssistantMessageId,
+        upToIndex,
+    }) && hasAssistantPartEvidence({
+        events,
+        sessionId,
+        messageId: latestAssistantMessageId,
+        upToIndex,
+    });
+}
+export function isAssistantMessageInLatestUserTurn({ events, sessionId, messageId, upToIndex, }) {
+    const assistantMessageIds = getAssistantMessageIdsForLatestUserTurn({
+        events,
+        sessionId,
+        upToIndex,
+    });
+    return assistantMessageIds.has(messageId);
+}
+// Returns a stable 1-based subtask index for candidateSessionId.
+// Indexing scope is the parent assistant message that spawned the task tool calls,
+// so numbering restarts at 1 for each assistant message.
+export function getDerivedSubtaskIndex({ events, mainSessionId, candidateSessionId, upToIndex, }) {
+    const end = upToIndex ?? events.length - 1;
+    let parentAssistantMessageId;
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const candidate = getTaskCandidateFromEvent({
+            event: entry.event,
+            mainSessionId,
+        });
+        if (!candidate) {
+            continue;
+        }
+        if (candidate.childSessionId !== candidateSessionId) {
+            continue;
+        }
+        parentAssistantMessageId = candidate.assistantMessageId;
+        break;
+    }
+    if (!parentAssistantMessageId) {
+        return undefined;
+    }
+    const indexByChildSessionId = new Map();
+    for (let i = 0; i <= end; i++) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const candidate = getTaskCandidateFromEvent({
+            event: entry.event,
+            mainSessionId,
+        });
+        if (!candidate || candidate.assistantMessageId !== parentAssistantMessageId) {
+            continue;
+        }
+        if (!indexByChildSessionId.has(candidate.childSessionId)) {
+            indexByChildSessionId.set(candidate.childSessionId, indexByChildSessionId.size + 1);
+        }
+    }
+    return indexByChildSessionId.get(candidateSessionId);
+}
+// Returns the subagent_type (e.g. "explore", "general") for a given child session.
+// Used to build labels like "explore-1" instead of generic "task-1".
+export function getDerivedSubtaskAgentType({ events, mainSessionId, candidateSessionId, }) {
+    for (let i = events.length - 1; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const candidate = getTaskCandidateFromEvent({
+            event: entry.event,
+            mainSessionId,
+        });
+        if (!candidate || candidate.childSessionId !== candidateSessionId) {
+            continue;
+        }
+        return candidate.subagentType;
+    }
+    return undefined;
+}
+export function getDerivedSubagentSessions({ events, mainSessionId, upToIndex, }) {
+    const end = upToIndex ?? events.length - 1;
+    const seenChildSessionIds = new Set();
+    const sessions = [];
+    for (let i = end; i >= 0; i--) {
+        const entry = events[i];
+        if (!entry) {
+            continue;
+        }
+        const candidate = getTaskCandidateFromEvent({
+            event: entry.event,
+            mainSessionId,
+        });
+        if (!candidate || seenChildSessionIds.has(candidate.childSessionId)) {
+            continue;
+        }
+        seenChildSessionIds.add(candidate.childSessionId);
+        sessions.push({
+            childSessionId: candidate.childSessionId,
+            subagentType: candidate.subagentType,
+            description: candidate.description,
+            timestamp: entry.timestamp,
+        });
+    }
+    return sessions;
+}