import { type Model } from "@oh-my-pi/pi-ai";
import * as z from "zod/v4";
import { type ModelRegistry } from "../config/model-registry";
import type { CustomTool } from "../extensibility/custom-tools/types";
type ImageProvider = "antigravity" | "gemini" | "openai" | "openai-codex" | "openrouter";
declare const responseModalitySchema: z.ZodEnum<{
    IMAGE: "IMAGE";
    TEXT: "TEXT";
}>;
export declare const imageGenSchema: z.ZodObject<{
    subject: z.ZodString;
    action: z.ZodOptional<z.ZodString>;
    scene: z.ZodOptional<z.ZodString>;
    composition: z.ZodOptional<z.ZodString>;
    lighting: z.ZodOptional<z.ZodString>;
    style: z.ZodOptional<z.ZodString>;
    text: z.ZodOptional<z.ZodString>;
    changes: z.ZodOptional<z.ZodArray<z.ZodString>>;
    aspect_ratio: z.ZodOptional<z.ZodEnum<{
        "16:9": "16:9";
        "1:1": "1:1";
        "3:4": "3:4";
        "4:3": "4:3";
        "9:16": "9:16";
    }>>;
    image_size: z.ZodOptional<z.ZodEnum<{
        "1024x1024": "1024x1024";
        "1024x1536": "1024x1536";
        "1536x1024": "1536x1024";
    }>>;
    input: z.ZodOptional<z.ZodArray<z.ZodObject<{
        path: z.ZodOptional<z.ZodString>;
        data: z.ZodOptional<z.ZodString>;
        mime_type: z.ZodOptional<z.ZodString>;
    }, z.core.$strict>>>;
}, z.core.$strict>;
export type ImageGenParams = z.infer<typeof imageGenSchema>;
export type GeminiResponseModality = z.infer<typeof responseModalitySchema>;
interface GeminiSafetyRating {
    category?: string;
    probability?: string;
}
interface GeminiPromptFeedback {
    blockReason?: string;
    safetyRatings?: GeminiSafetyRating[];
}
interface GeminiUsageMetadata {
    promptTokenCount?: number;
    candidatesTokenCount?: number;
    totalTokenCount?: number;
}
interface OpenAIResponsesUsage {
    input_tokens?: number;
    output_tokens?: number;
    total_tokens?: number;
}
type ImageUsageMetadata = GeminiUsageMetadata | OpenAIResponsesUsage;
interface ImageGenToolDetails {
    provider: ImageProvider;
    model: string;
    imageCount: number;
    imagePaths: string[];
    images: InlineImageData[];
    responseText?: string;
    promptFeedback?: GeminiPromptFeedback;
    revisedPrompt?: string;
    usage?: ImageUsageMetadata;
}
interface InlineImageData {
    data: string;
    mimeType: string;
}
/** Set the preferred image provider from settings */
export declare function setPreferredImageProvider(provider: ImageProvider | "auto"): void;
export declare const imageGenTool: CustomTool<typeof imageGenSchema, ImageGenToolDetails>;
export declare function getImageGenTools(modelRegistry?: ModelRegistry, activeModel?: Model): Promise<Array<CustomTool<typeof imageGenSchema, ImageGenToolDetails>>>;
export declare function getImageGenToolsWithRegistry(modelRegistry: ModelRegistry, activeModel?: Model): Promise<Array<CustomTool<typeof imageGenSchema, ImageGenToolDetails>>>;
export {};