diff --git a/packages/session-ui/src/timeline/projection.ts b/packages/session-ui/src/timeline/projection.ts index 47c33900881..d8e98cc25c6 100644 --- a/packages/session-ui/src/timeline/projection.ts +++ b/packages/session-ui/src/timeline/projection.ts @@ -271,7 +271,7 @@ export namespace Timeline { detail, new Set( messages - .filter((message) => message.type === "compaction" && timelineNoticeRequired(message)) + .filter((message) => message.type === "compaction" || message.type === "model-switched") .map((message) => message.id), ), ) @@ -415,13 +415,13 @@ function shellFailed(message: SessionMessageShell) { ) } -function groupMessages(rows: TimelineRow.TimelineRow[], detail: TimelineDetail, required: ReadonlySet) { +function groupMessages(rows: TimelineRow.TimelineRow[], detail: TimelineDetail, separate: ReadonlySet) { return rows.reduce((result, row) => { const previous = result.at(-1) const current = ((row._tag === "Notice" && detail.notices.placement === "grouped") || (row._tag === "Shell" && detail.shell.placement === "grouped")) && - !required.has(row.messageID) + !separate.has(row.messageID) ? new TimelineRow.AssistantPart({ userMessageID: row.userMessageID, previousAssistantPart: previous?._tag === "AssistantPart",