Generate Object
Learn how to generate object using the AI TOOLKIT and Next.js
Earlier functions like generateText and streamText gave us the ability to generate unstructured text. However, if you want to generate structured data like JSON, you can provide a schema that describes the structure of your desired object to the generateObject function. The function requires you to provide a schema using zod, a library for defining schemas for JavaScript objects. By using zod, you can also use it to validate the generated object and ensure that it conforms to the specified structure. object={{ notifications: [ { name: 'Jamie Roberts', message: "Hey! How's the study grind going? Need a coffee boost?", minutesAgo: 15, }, { name: 'Prof. Morgan', message: 'Reminder: Your term paper is due promptly at 8 AM tomorrow. Please ensure it meets the submission guidelines outlined.', minutesAgo: 46, }, { name: 'Alex Chen', message: "Dude, urgent! Borrow your notes for tomorrow's exam? I swear mine got eaten by my dog!", minutesAgo: 30, }, ], }} />
Run it locally
$ npm install aiClient
Let's create a simple React component that will make a POST request to the /api/completion endpoint when a button is clicked. The endpoint will return the generated object based on the input prompt and we'll display it.
'use client';
import { useState } from 'react';
export default function Page() {
const [generation, setGeneration] = useState();
const [isLoading, setIsLoading] = useState(false);
return (
<div>
<div
onClick={async () => {
setIsLoading(true);
await fetch('/api/completion', {
method: 'POST',
body: JSON.stringify({
prompt: 'Messages during finals week.',
}),
}).then(response => {
response.json().then(json => {
setGeneration(json.notifications);
setIsLoading(false);
});
});
}}
>
Generate
</div>
{isLoading ? (
'Loading...'
) : (
<pre>{JSON.stringify(generation, null, 2)}</pre>
)}
</div>
);
}
- 1'use client';
- 2import { useState } from 'react';
- 3export default function Page() {
- 4 const [generation, setGeneration] = useState();
- 5 const [isLoading, setIsLoading] = useState(false);
- 6 return (
- 7 <div>
- 8 <div
- 9 onClick={async () => {
- 10 setIsLoading(true);
- 11 await fetch('/api/completion', {
- 12 method: 'POST',
- 13 body: JSON.stringify({
- 14 prompt: 'Messages during finals week.',
- 15 }),
- 16 }).then(response => {
- 17 response.json().then(json => {
- 18 setGeneration(json.notifications);
- 19 setIsLoading(false);
- 20 });
- 21 });
- 22 }}
- 23 >
- 24 Generate
- 25 </div>
- 26 {isLoading ? (
- 27 'Loading...'
- 28 ) : (
- 29 <pre>{JSON.stringify(generation, null, 2)}</pre>
- 30 )}
- 31 </div>
- 32 );
- 33}
Server
Let's create a route handler for /api/completion that will generate an object based on the input prompt. The route will call the generateObject function from the ai module, which will then generate an object based on the input prompt and return it.
import { generateObject } from 'ai-toolkit';
import { z } from 'zod';
export async function POST(req: Request) {
const { prompt }: { prompt: string } = await req.json();
const result = await generateObject({
model: 'openai/gpt-4o',
system: 'You generate three notifications for a messages app.',
prompt,
schema: z.object({
notifications: z.array(
z.object({
name: z.string().describe('Name of a fictional person.'),
message: z.string().describe('Do not use emojis or links.'),
minutesAgo: z.number(),
}),
),
}),
});
return result.toJsonResponse();
}
---
<GithubLink link="https://github.com/khulnasoft/ai-toolkit/blob/main/examples/next-openai-pages/pages/basics/generate-object/index.tsx" />
- 1import { generateObject } from 'ai-toolkit';
- 2import { z } from 'zod';
- 3export async function POST(req: Request) {
- 4 const { prompt }: { prompt: string } = await req.json();
- 5 const result = await generateObject({
- 6 model: 'openai/gpt-4o',
- 7 system: 'You generate three notifications for a messages app.',
- 8 prompt,
- 9 schema: z.object({
- 10 notifications: z.array(
- 11 z.object({
- 12 name: z.string().describe('Name of a fictional person.'),
- 13 message: z.string().describe('Do not use emojis or links.'),
- 14 minutesAgo: z.number(),
- 15 }),
- 16 ),
- 17 }),
- 18 });
- 19 return result.toJsonResponse();
- 20}