@builder.io/ai-utils 0.59.0 → 0.60.0

package/src/mapping.d.ts

@@ -1,3 +1,4 @@
+import { z } from "zod";
 import type { ESMImport } from "./codegen";
 export interface RawFigmaJson {
     documents: any[];
@@ -7,18 +8,18 @@ export interface RawFigmaJson {
     componentSets?: Record<string, any>;
     schemaVersion?: number;
 }
-export interface UserContext {
-    client: string;
-    clientVersion: string;
-    nodeVersion: string;
-    systemPlatform: string;
-    frameworks: string[];
-    systemEOL: string;
-    systemArch: string;
-    systemShell?: string;
-    inGitRepo?: boolean;
-    [key: string]: string | string[] | boolean | undefined;
-}
+export declare const UserContextSchema: z.ZodObject<{
+    client: z.ZodString;
+    clientVersion: z.ZodString;
+    nodeVersion: z.ZodString;
+    systemPlatform: z.ZodString;
+    frameworks: z.ZodArray<z.ZodString>;
+    systemEOL: z.ZodString;
+    systemArch: z.ZodString;
+    systemShell: z.ZodOptional<z.ZodString>;
+    inGitRepo: z.ZodOptional<z.ZodBoolean>;
+}, z.core.$catchall<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>, z.ZodBoolean, z.ZodUndefined]>>>;
+export type UserContext = z.infer<typeof UserContextSchema>;
 export type ExportType = "default" | "named";
 /**
  * Gets the latest component mappings for a space

package/src/mapping.js

@@ -1 +1,15 @@
-export {};
+import { z } from "zod";
+export const UserContextSchema = z
+    .object({
+    client: z.string(),
+    clientVersion: z.string(),
+    nodeVersion: z.string(),
+    systemPlatform: z.string(),
+    frameworks: z.array(z.string()),
+    systemEOL: z.string(),
+    systemArch: z.string(),
+    systemShell: z.string().optional(),
+    inGitRepo: z.boolean().optional(),
+})
+    .catchall(z.union([z.string(), z.array(z.string()), z.boolean(), z.undefined()]))
+    .meta({ title: "UserContext" });

package/src/messages.d.ts

@@ -1,24 +1,10 @@
+import { z } from "zod";
 import type { ContentUpdatePatch } from "./events.js";
 /**
  * Message param does not know the id of the message.
  * This is an input message.
  */
 export type MessageParam = SystemMessageParam | UserMessageParam | AssistantMessageParam;
-export interface ContentMessageItemText {
-    type: "text";
-    text: string;
-    cache?: boolean;
-    citations?: TextCitationParam[] | null;
-    ephemeral?: boolean;
-    thoughtSignature?: string;
-    tag?: string;
-}
-export interface ContentMessageItemImage {
-    type: "image";
-    source: ImageBase64Source | ImageUrlSource;
-    cache?: boolean;
-    ephemeral?: boolean;
-}
 export interface ContentMessageItemDocument {
     type: "document";
     source: DocumentBase64Source | DocumentUrlSource | DocumentTextSource;
@@ -32,15 +18,6 @@ export interface ContentMessageItemVideo {
     cache?: boolean;
     ephemeral?: boolean;
 }
-export interface ContentMessageItemResource {
-    type: "resource";
-    resource: {
-        uri: string;
-        mimeType?: string;
-        text?: string;
-        blob?: string;
-    };
-}
 export interface DocumentBase64Source {
     type: "base64";
     media_type: "application/pdf";
@@ -56,28 +33,6 @@ export interface DocumentTextSource {
     media_type: "text/plain";
     data: string;
 }
-export interface ContentMessageItemToolResult {
-    type: "tool_result";
-    tool_use_id: string;
-    tool_name?: string;
-    tool_input?: string;
-    title?: string;
-    content: string | (ContentMessageItemText | ContentMessageItemImage | ContentMessageItemResource)[];
-    is_error?: boolean;
-    cache?: boolean;
-    ephemeral?: boolean;
-    structured_result?: Record<string, any>;
-}
-export interface ImageBase64Source {
-    type: "base64";
-    media_type: "image/webp" | "image/png" | "image/jpeg" | "image/gif";
-    data: string;
-    original_url?: string;
-}
-export interface ImageUrlSource {
-    type: "url";
-    url: string;
-}
 export interface VideoUrlSource {
     type: "url";
     url: string;
@@ -105,38 +60,6 @@ export interface ContentMessageItemToolUse {
     thoughtSignature?: string;
     name: string;
 }
-export interface CitationCharLocationParam {
-    cited_text: string;
-    document_index: number;
-    document_title: string | null;
-    end_char_index: number;
-    start_char_index: number;
-    type: "char_location";
-}
-export interface CitationContentBlockLocationParam {
-    cited_text: string;
-    document_index: number;
-    document_title: string | null;
-    end_block_index: number;
-    start_block_index: number;
-    type: "content_block_location";
-}
-export interface CitationPageLocationParam {
-    cited_text: string;
-    document_index: number;
-    document_title: string | null;
-    end_page_number: number;
-    start_page_number: number;
-    type: "page_location";
-}
-export interface CitationWebSearchResultLocationParam {
-    cited_text: string;
-    encrypted_index: string;
-    title: string | null;
-    type: "web_search_result_location";
-    url: string;
-}
-export type TextCitationParam = CitationCharLocationParam | CitationPageLocationParam | CitationContentBlockLocationParam | CitationWebSearchResultLocationParam;
 export interface ServerToolUseContent {
     type: "web_search_result";
     title: string;
@@ -300,33 +223,308 @@ export interface AssistantActionMessage {
 export type Message = SystemMessage | UserMessage | AssistantMessage;
 export type GeneratingMessage = null | Partial<AssistantActionMessage | AssistantMessage>;
 export declare function getContentText(message: string | ContentMessage): string;
-export declare function getContentAttachments(message: string | ContentMessage): (ImageBase64Source | ImageUrlSource)[];
-export type Attachment = FileUpload | Template | URL | BuilderContentAttachment | FigmaContentAttachment;
-export interface URL {
+export declare function getContentAttachments(message: string | ContentMessage): ({
+    type: "base64";
+    media_type: "image/webp" | "image/png" | "image/jpeg" | "image/gif";
+    data: string;
+    original_url?: string | undefined;
+} | {
     type: "url";
-    value: string;
-}
-export interface FileUpload {
-    type: "upload";
-    contentType: "image/webp" | "image/png" | "image/jpeg" | "image/gif" | "application/pdf" | "application/json" | "text/plain";
-    name: string;
-    dataUrl: string;
-    text?: string;
-    size: number;
-    id: string;
-    originalUrl?: string;
-    ephemeral?: boolean;
-}
-export interface Template {
-    type: "template";
-    name: string;
-    id: number;
-}
-export interface BuilderContentAttachment {
-    type: "builder";
-    html: string;
-}
-export interface FigmaContentAttachment {
-    type: "figma";
-    html: string;
-}
+    url: string;
+})[];
+export declare const URLSchema: z.ZodObject<{
+    type: z.ZodLiteral<"url">;
+    value: z.ZodString;
+}, z.core.$strip>;
+export type URL = z.infer<typeof URLSchema>;
+export declare const FileUploadSchema: z.ZodObject<{
+    type: z.ZodLiteral<"upload">;
+    contentType: z.ZodEnum<{
+        "image/webp": "image/webp";
+        "image/png": "image/png";
+        "image/jpeg": "image/jpeg";
+        "image/gif": "image/gif";
+        "application/pdf": "application/pdf";
+        "application/json": "application/json";
+        "text/plain": "text/plain";
+    }>;
+    name: z.ZodString;
+    dataUrl: z.ZodString;
+    text: z.ZodOptional<z.ZodString>;
+    size: z.ZodNumber;
+    id: z.ZodString;
+    originalUrl: z.ZodOptional<z.ZodString>;
+    ephemeral: z.ZodOptional<z.ZodBoolean>;
+}, z.core.$strip>;
+export type FileUpload = z.infer<typeof FileUploadSchema>;
+export declare const TemplateSchema: z.ZodObject<{
+    type: z.ZodLiteral<"template">;
+    name: z.ZodString;
+    id: z.ZodNumber;
+}, z.core.$strip>;
+export type Template = z.infer<typeof TemplateSchema>;
+export declare const BuilderContentAttachmentSchema: z.ZodObject<{
+    type: z.ZodLiteral<"builder">;
+    html: z.ZodString;
+}, z.core.$strip>;
+export type BuilderContentAttachment = z.infer<typeof BuilderContentAttachmentSchema>;
+export declare const FigmaContentAttachmentSchema: z.ZodObject<{
+    type: z.ZodLiteral<"figma">;
+    html: z.ZodString;
+}, z.core.$strip>;
+export type FigmaContentAttachment = z.infer<typeof FigmaContentAttachmentSchema>;
+export declare const AttachmentSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
+    type: z.ZodLiteral<"upload">;
+    contentType: z.ZodEnum<{
+        "image/webp": "image/webp";
+        "image/png": "image/png";
+        "image/jpeg": "image/jpeg";
+        "image/gif": "image/gif";
+        "application/pdf": "application/pdf";
+        "application/json": "application/json";
+        "text/plain": "text/plain";
+    }>;
+    name: z.ZodString;
+    dataUrl: z.ZodString;
+    text: z.ZodOptional<z.ZodString>;
+    size: z.ZodNumber;
+    id: z.ZodString;
+    originalUrl: z.ZodOptional<z.ZodString>;
+    ephemeral: z.ZodOptional<z.ZodBoolean>;
+}, z.core.$strip>, z.ZodObject<{
+    type: z.ZodLiteral<"template">;
+    name: z.ZodString;
+    id: z.ZodNumber;
+}, z.core.$strip>, z.ZodObject<{
+    type: z.ZodLiteral<"url">;
+    value: z.ZodString;
+}, z.core.$strip>, z.ZodObject<{
+    type: z.ZodLiteral<"builder">;
+    html: z.ZodString;
+}, z.core.$strip>, z.ZodObject<{
+    type: z.ZodLiteral<"figma">;
+    html: z.ZodString;
+}, z.core.$strip>], "type">;
+export type Attachment = z.infer<typeof AttachmentSchema>;
+export declare const CitationCharLocationParamSchema: z.ZodObject<{
+    cited_text: z.ZodString;
+    document_index: z.ZodNumber;
+    document_title: z.ZodNullable<z.ZodString>;
+    end_char_index: z.ZodNumber;
+    start_char_index: z.ZodNumber;
+    type: z.ZodLiteral<"char_location">;
+}, z.core.$strip>;
+export type CitationCharLocationParam = z.infer<typeof CitationCharLocationParamSchema>;
+export declare const CitationContentBlockLocationParamSchema: z.ZodObject<{
+    cited_text: z.ZodString;
+    document_index: z.ZodNumber;
+    document_title: z.ZodNullable<z.ZodString>;
+    end_block_index: z.ZodNumber;
+    start_block_index: z.ZodNumber;
+    type: z.ZodLiteral<"content_block_location">;
+}, z.core.$strip>;
+export type CitationContentBlockLocationParam = z.infer<typeof CitationContentBlockLocationParamSchema>;
+export declare const CitationPageLocationParamSchema: z.ZodObject<{
+    cited_text: z.ZodString;
+    document_index: z.ZodNumber;
+    document_title: z.ZodNullable<z.ZodString>;
+    end_page_number: z.ZodNumber;
+    start_page_number: z.ZodNumber;
+    type: z.ZodLiteral<"page_location">;
+}, z.core.$strip>;
+export type CitationPageLocationParam = z.infer<typeof CitationPageLocationParamSchema>;
+export declare const CitationWebSearchResultLocationParamSchema: z.ZodObject<{
+    cited_text: z.ZodString;
+    encrypted_index: z.ZodString;
+    title: z.ZodNullable<z.ZodString>;
+    type: z.ZodLiteral<"web_search_result_location">;
+    url: z.ZodString;
+}, z.core.$strip>;
+export type CitationWebSearchResultLocationParam = z.infer<typeof CitationWebSearchResultLocationParamSchema>;
+export declare const TextCitationParamSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
+    cited_text: z.ZodString;
+    document_index: z.ZodNumber;
+    document_title: z.ZodNullable<z.ZodString>;
+    end_char_index: z.ZodNumber;
+    start_char_index: z.ZodNumber;
+    type: z.ZodLiteral<"char_location">;
+}, z.core.$strip>, z.ZodObject<{
+    cited_text: z.ZodString;
+    document_index: z.ZodNumber;
+    document_title: z.ZodNullable<z.ZodString>;
+    end_page_number: z.ZodNumber;
+    start_page_number: z.ZodNumber;
+    type: z.ZodLiteral<"page_location">;
+}, z.core.$strip>, z.ZodObject<{
+    cited_text: z.ZodString;
+    document_index: z.ZodNumber;
+    document_title: z.ZodNullable<z.ZodString>;
+    end_block_index: z.ZodNumber;
+    start_block_index: z.ZodNumber;
+    type: z.ZodLiteral<"content_block_location">;
+}, z.core.$strip>, z.ZodObject<{
+    cited_text: z.ZodString;
+    encrypted_index: z.ZodString;
+    title: z.ZodNullable<z.ZodString>;
+    type: z.ZodLiteral<"web_search_result_location">;
+    url: z.ZodString;
+}, z.core.$strip>], "type">;
+export type TextCitationParam = z.infer<typeof TextCitationParamSchema>;
+export declare const ImageBase64SourceSchema: z.ZodObject<{
+    type: z.ZodLiteral<"base64">;
+    media_type: z.ZodEnum<{
+        "image/webp": "image/webp";
+        "image/png": "image/png";
+        "image/jpeg": "image/jpeg";
+        "image/gif": "image/gif";
+    }>;
+    data: z.ZodString;
+    original_url: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export type ImageBase64Source = z.infer<typeof ImageBase64SourceSchema>;
+export declare const ImageUrlSourceSchema: z.ZodObject<{
+    type: z.ZodLiteral<"url">;
+    url: z.ZodString;
+}, z.core.$strip>;
+export type ImageUrlSource = z.infer<typeof ImageUrlSourceSchema>;
+export declare const ContentMessageItemTextSchema: z.ZodObject<{
+    type: z.ZodLiteral<"text">;
+    text: z.ZodString;
+    cache: z.ZodOptional<z.ZodBoolean>;
+    citations: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
+        cited_text: z.ZodString;
+        document_index: z.ZodNumber;
+        document_title: z.ZodNullable<z.ZodString>;
+        end_char_index: z.ZodNumber;
+        start_char_index: z.ZodNumber;
+        type: z.ZodLiteral<"char_location">;
+    }, z.core.$strip>, z.ZodObject<{
+        cited_text: z.ZodString;
+        document_index: z.ZodNumber;
+        document_title: z.ZodNullable<z.ZodString>;
+        end_page_number: z.ZodNumber;
+        start_page_number: z.ZodNumber;
+        type: z.ZodLiteral<"page_location">;
+    }, z.core.$strip>, z.ZodObject<{
+        cited_text: z.ZodString;
+        document_index: z.ZodNumber;
+        document_title: z.ZodNullable<z.ZodString>;
+        end_block_index: z.ZodNumber;
+        start_block_index: z.ZodNumber;
+        type: z.ZodLiteral<"content_block_location">;
+    }, z.core.$strip>, z.ZodObject<{
+        cited_text: z.ZodString;
+        encrypted_index: z.ZodString;
+        title: z.ZodNullable<z.ZodString>;
+        type: z.ZodLiteral<"web_search_result_location">;
+        url: z.ZodString;
+    }, z.core.$strip>], "type">>>>;
+    ephemeral: z.ZodOptional<z.ZodBoolean>;
+    thoughtSignature: z.ZodOptional<z.ZodString>;
+    tag: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export type ContentMessageItemText = z.infer<typeof ContentMessageItemTextSchema>;
+export declare const ContentMessageItemImageSchema: z.ZodObject<{
+    type: z.ZodLiteral<"image">;
+    source: z.ZodDiscriminatedUnion<[z.ZodObject<{
+        type: z.ZodLiteral<"base64">;
+        media_type: z.ZodEnum<{
+            "image/webp": "image/webp";
+            "image/png": "image/png";
+            "image/jpeg": "image/jpeg";
+            "image/gif": "image/gif";
+        }>;
+        data: z.ZodString;
+        original_url: z.ZodOptional<z.ZodString>;
+    }, z.core.$strip>, z.ZodObject<{
+        type: z.ZodLiteral<"url">;
+        url: z.ZodString;
+    }, z.core.$strip>], "type">;
+    cache: z.ZodOptional<z.ZodBoolean>;
+    ephemeral: z.ZodOptional<z.ZodBoolean>;
+}, z.core.$strip>;
+export type ContentMessageItemImage = z.infer<typeof ContentMessageItemImageSchema>;
+export declare const ContentMessageItemResourceSchema: z.ZodObject<{
+    type: z.ZodLiteral<"resource">;
+    resource: z.ZodObject<{
+        uri: z.ZodString;
+        mimeType: z.ZodOptional<z.ZodString>;
+        text: z.ZodOptional<z.ZodString>;
+        blob: z.ZodOptional<z.ZodString>;
+    }, z.core.$strip>;
+}, z.core.$strip>;
+export type ContentMessageItemResource = z.infer<typeof ContentMessageItemResourceSchema>;
+export declare const ContentMessageItemToolResultSchema: z.ZodObject<{
+    type: z.ZodLiteral<"tool_result">;
+    tool_use_id: z.ZodString;
+    tool_name: z.ZodOptional<z.ZodString>;
+    tool_input: z.ZodOptional<z.ZodString>;
+    title: z.ZodOptional<z.ZodString>;
+    content: z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
+        type: z.ZodLiteral<"text">;
+        text: z.ZodString;
+        cache: z.ZodOptional<z.ZodBoolean>;
+        citations: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
+            cited_text: z.ZodString;
+            document_index: z.ZodNumber;
+            document_title: z.ZodNullable<z.ZodString>;
+            end_char_index: z.ZodNumber;
+            start_char_index: z.ZodNumber;
+            type: z.ZodLiteral<"char_location">;
+        }, z.core.$strip>, z.ZodObject<{
+            cited_text: z.ZodString;
+            document_index: z.ZodNumber;
+            document_title: z.ZodNullable<z.ZodString>;
+            end_page_number: z.ZodNumber;
+            start_page_number: z.ZodNumber;
+            type: z.ZodLiteral<"page_location">;
+        }, z.core.$strip>, z.ZodObject<{
+            cited_text: z.ZodString;
+            document_index: z.ZodNumber;
+            document_title: z.ZodNullable<z.ZodString>;
+            end_block_index: z.ZodNumber;
+            start_block_index: z.ZodNumber;
+            type: z.ZodLiteral<"content_block_location">;
+        }, z.core.$strip>, z.ZodObject<{
+            cited_text: z.ZodString;
+            encrypted_index: z.ZodString;
+            title: z.ZodNullable<z.ZodString>;
+            type: z.ZodLiteral<"web_search_result_location">;
+            url: z.ZodString;
+        }, z.core.$strip>], "type">>>>;
+        ephemeral: z.ZodOptional<z.ZodBoolean>;
+        thoughtSignature: z.ZodOptional<z.ZodString>;
+        tag: z.ZodOptional<z.ZodString>;
+    }, z.core.$strip>, z.ZodObject<{
+        type: z.ZodLiteral<"image">;
+        source: z.ZodDiscriminatedUnion<[z.ZodObject<{
+            type: z.ZodLiteral<"base64">;
+            media_type: z.ZodEnum<{
+                "image/webp": "image/webp";
+                "image/png": "image/png";
+                "image/jpeg": "image/jpeg";
+                "image/gif": "image/gif";
+            }>;
+            data: z.ZodString;
+            original_url: z.ZodOptional<z.ZodString>;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"url">;
+            url: z.ZodString;
+        }, z.core.$strip>], "type">;
+        cache: z.ZodOptional<z.ZodBoolean>;
+        ephemeral: z.ZodOptional<z.ZodBoolean>;
+    }, z.core.$strip>, z.ZodObject<{
+        type: z.ZodLiteral<"resource">;
+        resource: z.ZodObject<{
+            uri: z.ZodString;
+            mimeType: z.ZodOptional<z.ZodString>;
+            text: z.ZodOptional<z.ZodString>;
+            blob: z.ZodOptional<z.ZodString>;
+        }, z.core.$strip>;
+    }, z.core.$strip>], "type">>]>;
+    is_error: z.ZodOptional<z.ZodBoolean>;
+    cache: z.ZodOptional<z.ZodBoolean>;
+    ephemeral: z.ZodOptional<z.ZodBoolean>;
+    structured_result: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+}, z.core.$strip>;
+export type ContentMessageItemToolResult = z.infer<typeof ContentMessageItemToolResultSchema>;

package/src/messages.js

@@ -1,3 +1,4 @@
+import { z } from "zod";
 export function getContentText(message) {
     if (typeof message === "string") {
         return message;
@@ -14,3 +15,160 @@ export function getContentAttachments(message) {
         .filter((item) => item.type === "image")
         .map((item) => item.source);
 }
+export const URLSchema = z
+    .object({ type: z.literal("url"), value: z.string() })
+    .meta({ title: "URL" });
+export const FileUploadSchema = z
+    .object({
+    type: z.literal("upload"),
+    contentType: z.enum([
+        "image/webp",
+        "image/png",
+        "image/jpeg",
+        "image/gif",
+        "application/pdf",
+        "application/json",
+        "text/plain",
+    ]),
+    name: z.string(),
+    dataUrl: z.string(),
+    text: z.string().optional(),
+    size: z.number(),
+    id: z.string(),
+    originalUrl: z.string().optional(),
+    ephemeral: z.boolean().optional(),
+})
+    .meta({ title: "FileUpload" });
+export const TemplateSchema = z
+    .object({ type: z.literal("template"), name: z.string(), id: z.number() })
+    .meta({ title: "Template" });
+export const BuilderContentAttachmentSchema = z
+    .object({ type: z.literal("builder"), html: z.string() })
+    .meta({ title: "BuilderContentAttachment" });
+export const FigmaContentAttachmentSchema = z
+    .object({ type: z.literal("figma"), html: z.string() })
+    .meta({ title: "FigmaContentAttachment" });
+export const AttachmentSchema = z
+    .discriminatedUnion("type", [
+    FileUploadSchema,
+    TemplateSchema,
+    URLSchema,
+    BuilderContentAttachmentSchema,
+    FigmaContentAttachmentSchema,
+])
+    .meta({ title: "Attachment" });
+export const CitationCharLocationParamSchema = z
+    .object({
+    cited_text: z.string(),
+    document_index: z.number(),
+    document_title: z.string().nullable(),
+    end_char_index: z.number(),
+    start_char_index: z.number(),
+    type: z.literal("char_location"),
+})
+    .meta({ title: "CitationCharLocationParam" });
+export const CitationContentBlockLocationParamSchema = z
+    .object({
+    cited_text: z.string(),
+    document_index: z.number(),
+    document_title: z.string().nullable(),
+    end_block_index: z.number(),
+    start_block_index: z.number(),
+    type: z.literal("content_block_location"),
+})
+    .meta({ title: "CitationContentBlockLocationParam" });
+export const CitationPageLocationParamSchema = z
+    .object({
+    cited_text: z.string(),
+    document_index: z.number(),
+    document_title: z.string().nullable(),
+    end_page_number: z.number(),
+    start_page_number: z.number(),
+    type: z.literal("page_location"),
+})
+    .meta({ title: "CitationPageLocationParam" });
+export const CitationWebSearchResultLocationParamSchema = z
+    .object({
+    cited_text: z.string(),
+    encrypted_index: z.string(),
+    title: z.string().nullable(),
+    type: z.literal("web_search_result_location"),
+    url: z.string(),
+})
+    .meta({ title: "CitationWebSearchResultLocationParam" });
+export const TextCitationParamSchema = z
+    .discriminatedUnion("type", [
+    CitationCharLocationParamSchema,
+    CitationPageLocationParamSchema,
+    CitationContentBlockLocationParamSchema,
+    CitationWebSearchResultLocationParamSchema,
+])
+    .meta({ title: "TextCitationParam" });
+export const ImageBase64SourceSchema = z
+    .object({
+    type: z.literal("base64"),
+    media_type: z.enum(["image/webp", "image/png", "image/jpeg", "image/gif"]),
+    data: z.string(),
+    original_url: z.string().optional(),
+})
+    .meta({ title: "ImageBase64Source" });
+export const ImageUrlSourceSchema = z
+    .object({
+    type: z.literal("url"),
+    url: z.string(),
+})
+    .meta({ title: "ImageUrlSource" });
+export const ContentMessageItemTextSchema = z
+    .object({
+    type: z.literal("text"),
+    text: z.string(),
+    cache: z.boolean().optional(),
+    citations: z.array(TextCitationParamSchema).nullish(),
+    ephemeral: z.boolean().optional(),
+    thoughtSignature: z.string().optional(),
+    tag: z.string().optional(),
+})
+    .meta({ title: "ContentMessageItemText" });
+export const ContentMessageItemImageSchema = z
+    .object({
+    type: z.literal("image"),
+    source: z.discriminatedUnion("type", [
+        ImageBase64SourceSchema,
+        ImageUrlSourceSchema,
+    ]),
+    cache: z.boolean().optional(),
+    ephemeral: z.boolean().optional(),
+})
+    .meta({ title: "ContentMessageItemImage" });
+export const ContentMessageItemResourceSchema = z
+    .object({
+    type: z.literal("resource"),
+    resource: z.object({
+        uri: z.string(),
+        mimeType: z.string().optional(),
+        text: z.string().optional(),
+        blob: z.string().optional(),
+    }),
+})
+    .meta({ title: "ContentMessageItemResource" });
+export const ContentMessageItemToolResultSchema = z
+    .object({
+    type: z.literal("tool_result"),
+    tool_use_id: z.string(),
+    tool_name: z.string().optional(),
+    tool_input: z.string().optional(),
+    title: z.string().optional(),
+    content: z.union([
+        z.string(),
+        z.array(z.discriminatedUnion("type", [
+            ContentMessageItemTextSchema,
+            ContentMessageItemImageSchema,
+            ContentMessageItemResourceSchema,
+        ])),
+    ]),
+    is_error: z.boolean().optional(),
+    cache: z.boolean().optional(),
+    ephemeral: z.boolean().optional(),
+    structured_result: z.record(z.string(), z.any()).optional(),
+})
+    .meta({ title: "ContentMessageItemToolResult" });