Function generateSubCollections

scripts/generate.ts:79–121

()

// generate 10 subcollections per each category
const generateSubCollections = async () => {
  const data = [] as any;
  const c = await getCategories();

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

                YOU MUST OUTPUT IN ONLY JSON.
                
                EXAMPLE:

                INPUT:
                Category Name: Art Hitory

                OUTPUT:
                { subcollections: ["Art History Books", "Art History CDs", ...] }
                
                Remember, ONLY RETURN THE JSON of 10 unique subcollections and nothing else.
                  
                MAKE SURE THERE ARE 10 SUBCOLLECTIONS IN THE OUTPUT.`,
      prompt: `Category Name: ${cat.name}`,
    });

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

    const categoriesToAdd = sc.map((subcol: string) => ({
      name: subcol,
      category_slug: cat.slug,
    }));
    data.push(...categoriesToAdd);
  });

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

// generateSubCollections();