messages

1import type { UIMessage } from "ai";
2import z from "zod";
3import type { Tools } from "@/registry/ai/tools/tools-registry";
4 
5const metadataSchema = z.object({
6	timestamp: z.string().optional(),
7	outputTokens: z.number().optional(),
8	agent: z
9		.object({
10			id: z.string(),
11			name: z.string(),
12			image: z.string().optional(),
13		})
14		.optional(),
15	mentions: z
16		.array(
17			z.object({
18				id: z.string(),
19				name: z.string(),
20			}),
21		)
22		.optional(),
23});
24 
25type AIUIMetadata = z.infer<typeof metadataSchema>;
26 
27const dataPartSchema = z.object({
28	someDataPart: z.object({}),
29});
30 
31type AIUIDataPart = z.infer<typeof dataPartSchema>;
32 
33type AIUITools = Tools;
34 
35export type AIUIMessage = UIMessage<AIUIMetadata, AIUIDataPart, AIUITools>;