Function generateCategories

scripts/generate.ts:43–70

()

// generate 20 categories per each collection
const generateCategories = async () => {
  const data = [] as any;
  const c = await getCollections();

  const promises = c.map(async (col) => {
    const { object } = await generateObject({
      model: client.languageModel("gpt-4o-mini", { structuredOutputs: true }),
      schema: z.object({
        categories: z.array(z.string()),
      }),
      system,
      prompt: `Collection Name: ${col.name}`,
    });

    const { categories: cats } = object;
    console.log(`Categories generated: ${cats.length}`);

    const categoriesToAdd = cats.map((category: string) => ({
      name: category,
      collection_id: col.id,
      slug: slugify(category, { lower: true }),
    }));
    data.push(...categoriesToAdd);
  });

  await Promise.all(promises);
  await db.insert(categories).values(data).onConflictDoNothing();
};

// generateCategories();