Function filterMessagesWithFeedback

packages/server/src/utils/getChatMessage.ts:298–345

(
    messages: Array<ChatMessage & { feedback: ChatMessageFeedback }>,
    feedbackTypes: ChatMessageRatingType[]
)

}

function filterMessagesWithFeedback(
    messages: Array<ChatMessage & { feedback: ChatMessageFeedback }>,
    feedbackTypes: ChatMessageRatingType[]
): ChatMessage[] {
    // Group messages by session for proper filtering
    const sessionGroups = new Map<string, Array<ChatMessage & { feedback: ChatMessageFeedback }>>()

    messages.forEach((message) => {
        const sessionId = message.sessionId
        if (!sessionId) return // Skip messages without sessionId

        if (!sessionGroups.has(sessionId)) {
            sessionGroups.set(sessionId, [])
        }
        sessionGroups.get(sessionId)!.push(message)
    })

    const result: ChatMessage[] = []

    // Process each session group
    sessionGroups.forEach((sessionMessages) => {
        // Sort by creation date to ensure proper order
        sessionMessages.sort((a, b) => new Date(a.createdDate).getTime() - new Date(b.createdDate).getTime())

        const toInclude = new Set<number>()

        sessionMessages.forEach((message, index) => {
            if (message.role === 'apiMessage' && message.feedback && feedbackTypes.includes(message.feedback.rating)) {
                // Include the feedback message
                toInclude.add(index)
                // Include the previous message (user message) if it exists
                if (index > 0) {
                    toInclude.add(index - 1)
                }
            }
        })

        // Add filtered messages to result
        sessionMessages.forEach((message, index) => {
            if (toInclude.has(index)) {
                result.push(message)
            }
        })
    })

    // Sort final result by creation date
    return result.sort((a, b) => new Date(a.createdDate).getTime() - new Date(b.createdDate).getTime())
}