import { Content, GoogleGenAI } from '@google/genai';
import { BaseMessage, BaseModel, BaseTool, GenerateTextOptions, generateTextReturn } from './AIBase';
export type GoogleGenerativeAIModelId = 'gemini-pro-latest' | 'gemini-flash-latest' | 'gemini-3.1-pro-preview' | 'gemini-3-pro-preview' | 'gemini-3-flash-preview' | 'gemini-3.1-flash-lite' | 'gemini-2.5-pro' | 'gemini-2.5-flash' | 'gemini-2.5-flash-lite-preview-06-17' | 'gemini-3-pro-image-preview' | 'gemini-3.1-flash-image-preview' | (string & {});
export type GoogleSafetySettings = Array<{
    category: string;
    threshold: string;
}>;
type GoogleMessages = {
    systemMessage?: string;
    messages: Content[];
};
export declare class GoogleModel extends BaseModel {
    provider: GoogleGenAI;
    modelID: GoogleGenerativeAIModelId;
    safetySettings?: GoogleSafetySettings;
    constructor(provider: GoogleGenAI, model: GoogleGenerativeAIModelId, safetySettings?: GoogleSafetySettings);
    convertMessagesForGoogle(messages: BaseMessage[]): GoogleMessages;
    convertMessagesFromGoogle(toolCallIDs: Map<any, string>, messages: Content[]): BaseMessage[];
    parseZodObject(zodObject: any): any;
    zodTypeToSchemaType(zodType: string): string;
    convertToolCallForGoogle(name: string, zodTool: BaseTool): any;
    validateParameterTypes(schema: any): void;
    generateToolCallID(): string;
    private accumulateUsage;
    isRetryableError(error: unknown): boolean;
    generateText(options: GenerateTextOptions): Promise<generateTextReturn>;
}
export declare function createGoogleProvider(apiKey: string): (model: GoogleGenerativeAIModelId, safetySettings?: GoogleSafetySettings) => GoogleModel;
export {};
//# sourceMappingURL=googleAI.d.ts.map