clearLocalBtwRunIds,

localMode,

} = context;

const finalizedRuns = new Map<string, number>();

const sessionRuns = new Map<string, number>();

const finalizedRuns = new Map<string, number>();

const finalizedRunsWithDisplay = new Map<string, number>();

const postFinalizingRuns = new Map<string, number>();

let streamAssembler = new TuiStreamAssembler();

let lastSessionKey = state.currentSessionKey;

}

lastSessionKey = state.currentSessionKey;

finalizedRuns.clear();

finalizedRunsWithDisplay.clear();

sessionRuns.clear();

postFinalizingRuns.clear();

streamAssembler = new TuiStreamAssembler();

pruneRunMap(sessionRuns);

};

const noteFinalizedRun = (runId: string) => {

const noteFinalizedRun = (runId: string, opts?: { displayedFinal?: boolean }) => {

finalizedRuns.set(runId, Date.now());

if (opts?.displayedFinal === true) {

finalizedRunsWithDisplay.set(runId, Date.now());

}

sessionRuns.delete(runId);

streamAssembler.drop(runId);

pruneRunMap(finalizedRuns);

pruneRunMap(finalizedRunsWithDisplay);

};

const notePostFinalizingRun = (runId: string) => {

runId: string;

wasActiveRun: boolean;

status: "idle" | "error";

displayedFinal?: boolean;

}) => {

noteFinalizedRun(params.runId);

noteFinalizedRun(params.runId, { displayedFinal: params.displayedFinal });

clearActiveRunIfMatch(params.runId);

flushPendingHistoryRefreshIfIdle();

if (params.wasActiveRun) {

return;

}

if (evt.state === "final") {

clearStaleStreamingIfNoTrackedRunRemains();

return;

const hasLateDisplayableFinal =

Boolean(evt.message) && !finalizedRunsWithDisplay.has(evt.runId);

if (!hasLateDisplayableFinal) {

clearStaleStreamingIfNoTrackedRunRemains();

return;

}

}

}

if (reconnectPendingRunId === evt.runId) {

if (text) {

chatLog.addSystem(text);

}

finalizeRun({ runId: evt.runId, wasActiveRun, status: "idle" });

finalizeRun({ runId: evt.runId, wasActiveRun, status: "idle", displayedFinal: true });

tui.requestRender();

return;

}

runId: evt.runId,

wasActiveRun,

status: stopReason === "error" ? "error" : "idle",

displayedFinal: !suppressEmptyExternalPlaceholder,

});

}

if (evt.state === "aborted") {