Function generateSubcategories

scripts/generate.ts:140–185

()

};

const generateSubcategories = async () => {
  const data = [] as any;
  const subcollections = (await getSubcollections()).map(
    (c) => c.subcollections,
  );

  const promises = subcollections.map(async (subcol) => {
    const { object } = await generateObject({
      model: client.languageModel("gpt-4o-mini", { structuredOutputs: true }),
      schema: z.object({
        subcategories: z.array(z.string()),
      }),
      system: `You are given the name of a subcollection of products in an art supply store.
                Your task is to generate 10 unique subcategories that belong to this subcollection.
                Make sure the subcategory names are broad.

                YOU MUST OUTPUT IN ONLY JSON.

                EXAMPLE:

                INPUT:
                Subcollection Name: Art Hitory

                OUTPUT:
                { subcategories: ["Art History Books", "Art History CDs", ...] }

                Remember, ONLY RETURN THE JSON of 10 unique subcategories and nothing else.

                MAKE SURE THERE ARE 10 SUBCATEGORIES IN THE OUTPUT.`,
      prompt: `Subcollection Name: ${subcol}`,
    });

    const { subcategories: sc } = object;
    console.log(`Subcategories generated: ${sc.length}`);

    const subcategoriesToAdd = sc.map((subcat: string) => ({
      name: subcat,
      slug: slugify(subcat, { lower: true }),
      subcollection_id: subcol.id,
    }));
    data.push(...subcategoriesToAdd);
  });

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

// getSubcollections();
// generateSubcategories();