vibecode-party-starter/starter/app/_hooks/useGenerateObject.ts

A Next.js starter project for vibecoding Saas apps with auth, payments, email, and more

import { experimental_useObject as useObject } from "@ai-sdk/react"
import { z } from "zod"
import { useState } from "react"
import { zodToJsonSchema } from "zod-to-json-schema"

export function useGenerateObject<T extends z.ZodTypeAny>(schema: T) {
  const [error, setError] = useState<string>("")
  
  type InferredType = z.infer<T>
  
  const { object, isLoading, submit } = useObject<InferredType>({
    api: "/api/ai/generate/object",
    schema,
  })

  const generate = async (prompt: string) => {
    setError("")
    try {
      // Convert Zod schema to JSON schema
      const jsonSchema = zodToJsonSchema(schema)
      
      await submit({
        schema: jsonSchema,
        prompt,
      })
    } catch (err) {
      console.error("Error generating object:", err)
      setError("Failed to generate structured data")
    }
  }

  return {
    object,
    isLoading,
    error,
    generate,
  }
}

// Example usage:
// const personSchema = z.object({
//   name: z.string().describe("The person's full name"),
//   age: z.number().describe("The person's age"),
//   occupation: z.string().describe("The person's job or profession"),
//   interests: z
//     .array(z.string())
//     .describe("List of the person's hobbies and interests"),
//   contact: z
//     .object({
//       email: z.string().email().describe("The person's email address"),
//       phone: z.string().describe("The person's phone number"),
//     })
//     .describe("Contact information"),
// })

// Usage in a component:
// const { object, isLoading, error, generate } = useGenerateObject(personSchema)