diff --git a/src/web-ui/src/flow_chat/utils/flowChatTurnOrdinal.ts b/src/web-ui/src/flow_chat/utils/flowChatTurnOrdinal.ts index 7d8087c28..117fa4f22 100644 --- a/src/web-ui/src/flow_chat/utils/flowChatTurnOrdinal.ts +++ b/src/web-ui/src/flow_chat/utils/flowChatTurnOrdinal.ts @@ -28,7 +28,9 @@ export function absoluteSessionTurnIndexForId( turnId: string, ): number | undefined { const ordinal = resolveTurnOrdinal(session, turnId); - return ordinal === undefined ? undefined : ordinal + 1; + if (ordinal !== undefined) return ordinal + 1; + const localIndex = session.dialogTurns.findIndex(turn => turn.id === turnId); + return localIndex < 0 ? undefined : absoluteSessionTurnIndexForLocalIndex(session, localIndex); } export function loadedSessionTurnIdForAbsoluteIndex(