import { PrismaClient, Prisma } from "@prisma/client";
import { v4 as uuidv4 } from "uuid";

const prisma = new PrismaClient();

const chatData: Prisma.ChatCreateInput[] = [
  {
    id: uuidv4(),
    createdAt: new Date(),
    model: {},
    ctx: {},
    messages: {
      create: [
        {
          id: uuidv4(),
          createdAt: new Date(),
          role: "system",
          content: "You are a bot",
          upvote: true,
          downvote: false,
        },
        {
          id: uuidv4(),
          createdAt: new Date(),
          role: "user",
          content: "What can I help you with today?",
          upvote: true,
          downvote: false,
        },
        {
          id: uuidv4(),
          createdAt: new Date(),
          role: "assistant",
          content: "Hello",
          upvote: true,
          downvote: false,
        },
      ],
    },
  },
];

async function main() {
  console.log(`Start seeding ...`);
  for (const c of chatData) {
    const chat = await prisma.chat.create({
      data: c,
    });
    console.log(`Created chat with id: ${chat.id}`);
  }
  console.log(`Seeding finished.`);
}

main()
  .then(async () => {
    await prisma.$disconnect();
  })
  .catch(async (e) => {
    console.error(e);
    await prisma.$disconnect();
    process.exit(1);
  });