import { type ZodType, z } from "zod";
import type { PromiseOrValue } from "../utils/type-utils.js";
import type { AgentInvokeOptions, AgentOptions, AgentProcessResult, AgentResponse, AgentResponseStream, GetterSchema, Message } from "./agent.js";
import { type ChatModelOutputUsage } from "./chat-model.js";
import { type FileType, type FileUnionContent, Model } from "./model.js";
export interface ImageModelOptions<I extends ImageModelInput = ImageModelInput, O extends ImageModelOutput = ImageModelOutput> extends Omit<AgentOptions<I, O>, "model"> {
    model?: string;
    modelOptions?: ImageModelInputOptionsWithGetter;
}
export declare abstract class ImageModel<I extends ImageModelInput = ImageModelInput, O extends ImageModelOutput = ImageModelOutput> extends Model<I, O> {
    options?: ImageModelOptions<I, O> | undefined;
    tag: string;
    constructor(options?: ImageModelOptions<I, O> | undefined);
    get credential(): PromiseOrValue<{
        url?: string;
        apiKey?: string;
        model?: string;
    }>;
    protected preprocess(input: I, options: AgentInvokeOptions): Promise<void>;
    protected postprocess(input: I, output: O, options: AgentInvokeOptions): Promise<void>;
    abstract process(input: I, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<O>>;
    protected processAgentOutput(input: I, output: Exclude<AgentResponse<O>, AgentResponseStream<O>>, options: AgentInvokeOptions): Promise<O>;
}
export type ImageModelInputImage = FileUnionContent[];
export interface ImageModelInput extends Message {
    prompt: string;
    image?: ImageModelInputImage;
    n?: number;
    outputFileType?: FileType;
    modelOptions?: ImageModelInputOptions;
}
export interface ImageModelInputOptions extends Record<string, unknown> {
    model?: string;
    preferInputFileType?: "file" | "url";
}
export type ImageModelInputOptionsWithGetter = GetterSchema<ImageModelInputOptions>;
export declare const imageModelInputSchema: z.ZodObject<{
    prompt: z.ZodString;
    image: z.ZodOptional<z.ZodArray<z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
    } & {
        type: z.ZodLiteral<"local">;
        path: z.ZodString;
    }, "strip", z.ZodTypeAny, {
        path: string;
        type: "local";
        filename?: string | undefined;
        mimeType?: string | undefined;
    }, {
        path: string;
        type: "local";
        filename?: string | undefined;
        mimeType?: string | undefined;
    }>, z.ZodObject<{
        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
    } & {
        type: z.ZodLiteral<"url">;
        url: z.ZodString;
    }, "strip", z.ZodTypeAny, {
        type: "url";
        url: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    }, {
        type: "url";
        url: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    }>, z.ZodObject<{
        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
    } & {
        type: z.ZodLiteral<"file">;
        data: z.ZodString;
    }, "strip", z.ZodTypeAny, {
        type: "file";
        data: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    }, {
        type: "file";
        data: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    }>]>, "many">>;
    n: z.ZodOptional<z.ZodNumber>;
    outputFileType: z.ZodOptional<z.ZodEnum<["local", "file", "url"]>>;
    modelOptions: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
}, "strip", z.ZodTypeAny, {
    prompt: string;
    modelOptions?: Record<string, unknown> | undefined;
    image?: ({
        type: "url";
        url: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    } | {
        type: "file";
        data: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    } | {
        path: string;
        type: "local";
        filename?: string | undefined;
        mimeType?: string | undefined;
    })[] | undefined;
    n?: number | undefined;
    outputFileType?: "local" | "file" | "url" | undefined;
}, {
    prompt: string;
    modelOptions?: Record<string, unknown> | undefined;
    image?: ({
        type: "url";
        url: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    } | {
        type: "file";
        data: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    } | {
        path: string;
        type: "local";
        filename?: string | undefined;
        mimeType?: string | undefined;
    })[] | undefined;
    n?: number | undefined;
    outputFileType?: "local" | "file" | "url" | undefined;
}>;
export interface ImageModelOutput extends Message {
    images: FileUnionContent[];
    /**
     * Token usage statistics
     */
    usage?: ChatModelOutputUsage;
    /**
     * Model name or version used
     */
    model?: string;
}
export declare const imageModelOutputSchema: z.ZodObject<{
    images: z.ZodArray<z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
    } & {
        type: z.ZodLiteral<"local">;
        path: z.ZodString;
    }, "strip", z.ZodTypeAny, {
        path: string;
        type: "local";
        filename?: string | undefined;
        mimeType?: string | undefined;
    }, {
        path: string;
        type: "local";
        filename?: string | undefined;
        mimeType?: string | undefined;
    }>, z.ZodObject<{
        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
    } & {
        type: z.ZodLiteral<"url">;
        url: z.ZodString;
    }, "strip", z.ZodTypeAny, {
        type: "url";
        url: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    }, {
        type: "url";
        url: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    }>, z.ZodObject<{
        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
    } & {
        type: z.ZodLiteral<"file">;
        data: z.ZodString;
    }, "strip", z.ZodTypeAny, {
        type: "file";
        data: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    }, {
        type: "file";
        data: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    }>]>, "many">;
    usage: z.ZodOptional<z.ZodObject<{
        inputTokens: z.ZodNumber;
        outputTokens: z.ZodNumber;
        aigneHubCredits: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
        cacheCreationInputTokens: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
        cacheReadInputTokens: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
        creditPrefix: ZodType<"$" | "€" | "¥" | undefined, z.ZodTypeDef, "$" | "€" | "¥" | undefined>;
    }, "strip", z.ZodTypeAny, {
        inputTokens: number;
        outputTokens: number;
        aigneHubCredits?: number | undefined;
        cacheCreationInputTokens?: number | undefined;
        cacheReadInputTokens?: number | undefined;
        creditPrefix?: "$" | "€" | "¥" | undefined;
    }, {
        inputTokens: number;
        outputTokens: number;
        aigneHubCredits?: number | undefined;
        cacheCreationInputTokens?: number | undefined;
        cacheReadInputTokens?: number | undefined;
        creditPrefix?: "$" | "€" | "¥" | undefined;
    }>>;
    model: z.ZodOptional<z.ZodString>;
}, "strip", z.ZodTypeAny, {
    images: ({
        type: "url";
        url: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    } | {
        type: "file";
        data: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    } | {
        path: string;
        type: "local";
        filename?: string | undefined;
        mimeType?: string | undefined;
    })[];
    model?: string | undefined;
    usage?: {
        inputTokens: number;
        outputTokens: number;
        aigneHubCredits?: number | undefined;
        cacheCreationInputTokens?: number | undefined;
        cacheReadInputTokens?: number | undefined;
        creditPrefix?: "$" | "€" | "¥" | undefined;
    } | undefined;
}, {
    images: ({
        type: "url";
        url: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    } | {
        type: "file";
        data: string;
        filename?: string | undefined;
        mimeType?: string | undefined;
    } | {
        path: string;
        type: "local";
        filename?: string | undefined;
        mimeType?: string | undefined;
    })[];
    model?: string | undefined;
    usage?: {
        inputTokens: number;
        outputTokens: number;
        aigneHubCredits?: number | undefined;
        cacheCreationInputTokens?: number | undefined;
        cacheReadInputTokens?: number | undefined;
        creditPrefix?: "$" | "€" | "¥" | undefined;
    } | undefined;
}>;