litlyx/dashboard/server/api/ai/[chat_id]/get_messages.ts

import { getUserProjectFromId } from "~/server/LIVE_DEMO_DATA";
import { AiChatModel } from "@schema/ai/AiChatSchema";
import type OpenAI from "openai";
import { getChartsInMessage } from "~/server/services/AiService";

export default defineEventHandler(async event => {
    const data = await getRequestData(event);
    if (!data) return;

    const { project_id } = data;

    if (!event.context.params) return;
    const chat_id = event.context.params['chat_id'];

    const chat = await AiChatModel.findOne({ _id: chat_id, project_id });
    if (!chat) return;

    return (chat.messages as OpenAI.Chat.Completions.ChatCompletionMessageParam[])
        .filter(e => e.role === 'assistant' || e.role === 'user')
        .map(e => {
            const charts = getChartsInMessage(e);
            const content = e.content;
            return { role: e.role, content, charts }
        })
        .filter(e=>{
            return e.charts.length > 0 || e.content
        })
});