/**
 * Google/Gemini chat model implementation
 */
import { z } from 'zod';
import { HarmCategory, HarmBlockThreshold } from '@google/generative-ai';
import { AbstractChatModel } from '../base';
import { BaseMessage } from '../messages';
import { ChatInvokeCompletion } from '../views';
export declare const GoogleChatModelSchema: z.ZodObject<{
    model: z.ZodDefault<z.ZodString>;
    temperature: z.ZodOptional<z.ZodNumber>;
    top_p: z.ZodOptional<z.ZodNumber>;
    top_k: z.ZodOptional<z.ZodNumber>;
    max_output_tokens: z.ZodOptional<z.ZodNumber>;
    candidate_count: z.ZodDefault<z.ZodNumber>;
    stop_sequences: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
    api_key: z.ZodOptional<z.ZodString>;
    safety_settings: z.ZodOptional<z.ZodArray<z.ZodObject<{
        category: z.ZodNativeEnum<typeof HarmCategory>;
        threshold: z.ZodNativeEnum<typeof HarmBlockThreshold>;
    }, "strip", z.ZodTypeAny, {
        category: HarmCategory;
        threshold: HarmBlockThreshold;
    }, {
        category: HarmCategory;
        threshold: HarmBlockThreshold;
    }>, "many">>;
    generation_config: z.ZodOptional<z.ZodAny>;
}, "strip", z.ZodTypeAny, {
    model: string;
    candidate_count: number;
    api_key?: string | undefined;
    temperature?: number | undefined;
    max_output_tokens?: number | undefined;
    top_p?: number | undefined;
    top_k?: number | undefined;
    stop_sequences?: string[] | undefined;
    safety_settings?: {
        category: HarmCategory;
        threshold: HarmBlockThreshold;
    }[] | undefined;
    generation_config?: any;
}, {
    api_key?: string | undefined;
    model?: string | undefined;
    temperature?: number | undefined;
    max_output_tokens?: number | undefined;
    top_p?: number | undefined;
    top_k?: number | undefined;
    candidate_count?: number | undefined;
    stop_sequences?: string[] | undefined;
    safety_settings?: {
        category: HarmCategory;
        threshold: HarmBlockThreshold;
    }[] | undefined;
    generation_config?: any;
}>;
type GoogleChatModelConfig = z.infer<typeof GoogleChatModelSchema>;
export declare class ChatGoogle extends AbstractChatModel {
    private config;
    private client;
    private genModel;
    constructor(config: Partial<GoogleChatModelConfig> & {
        model?: string;
        api_key?: string;
    });
    get provider(): string;
    private serializeMessages;
    private serializeSystemMessage;
    private serializeUserMessage;
    private serializeAssistantMessage;
    private getUsage;
    private fixGeminiSchema;
    private isRetryableError;
    ainvoke<T = string>(messages: BaseMessage[], outputFormat?: z.ZodSchema<T>): Promise<ChatInvokeCompletion<T>>;
}
export {};
//# sourceMappingURL=google.d.ts.map