Function getTurnProgressState

packages/web/src/features/chat/utils.ts:338–377

({
    messages,
    status,
}: {
    messages: SBChatMessage[];
    status: ChatStatus;
})

}

export const getTurnProgressState = ({
    messages,
    status,
}: {
    messages: SBChatMessage[];
    status: ChatStatus;
}) => {
    const isNetworkActive = status === 'submitted' || status === 'streaming';
    const latestMessage = messages.at(-1);
    const latestAssistantMessage = latestMessage?.role === 'assistant' ? latestMessage : undefined;
    const latestStepToolParts = getLastStepParts(latestAssistantMessage?.parts ?? [])
        .filter(isSBChatToolPart);

    const hasPendingToolApproval = latestStepToolParts.some(
        (part) => part.state === 'approval-requested'
    );
    const hasApprovalContinuationReady =
        latestStepToolParts.some((part) => part.state === 'approval-responded') &&
        latestStepToolParts.every((part) =>
            part.state === 'output-available' ||
            part.state === 'output-error' ||
            part.state === 'approval-responded'
        );

    const isReady = status === 'ready';
    const isTurnInProgress =
        isNetworkActive ||
        (isReady && (hasPendingToolApproval || hasApprovalContinuationReady));
    const isAwaitingToolApproval = isReady && hasPendingToolApproval;
    const shouldGuardNavigation = isNetworkActive || (isReady && hasApprovalContinuationReady);

    return {
        isNetworkActive,
        hasPendingToolApproval,
        hasApprovalContinuationReady,
        isAwaitingToolApproval,
        isTurnInProgress,
        shouldGuardNavigation,
    };
}

// LLMs like to not follow instructions... this takes care of some common mistakes they tend to make.
export const repairReferences = (text: string): string => {