Function groupMessageIntoSteps

packages/web/src/features/chat/utils.ts:306–332

(parts: SBChatMessagePart[])

// Groups message parts into groups based on step-start delimiters.
export const groupMessageIntoSteps = (parts: SBChatMessagePart[]) => {
    if (!parts || parts.length === 0) {
        return [];
    }

    const steps: SBChatMessagePart[][] = [];
    let currentStep: SBChatMessagePart[] = [];

    for (let i = 0; i < parts.length; i++) {
        const part = parts[i];

        if (part.type === 'step-start') {
            if (currentStep.length > 0) {
                steps.push([...currentStep]);
            }
            currentStep = [part];
        } else {
            currentStep.push(part);
        }
    }

    if (currentStep.length > 0) {
        steps.push(currentStep);
    }

    return steps;
}

export const getLastStepParts = (parts: SBChatMessagePart[]): SBChatMessagePart[] => {
    return groupMessageIntoSteps(parts).at(-1) ?? parts;