import { GoogleGenAI } from '@google/genai';
import { ModelCapabilities, ChatModel, EmbeddingModel, ImageModel } from '@core-ai/core-ai';
import { z } from 'zod';

type GoogleModelCapabilities = Omit<ModelCapabilities, 'reasoning'> & {
    reasoning: ModelCapabilities['reasoning'] & {
        thinkingParam: 'thinkingLevel' | 'thinkingBudget';
    };
};
declare function getGoogleModelCapabilities(modelId: string): GoogleModelCapabilities;

/**
 * Thought signature Google attaches under the `google` provider metadata key.
 * Used on reasoning parts today, and on tool-call parts so Gemini 3 function
 * calls can be replayed without a missing-`thought_signature` 400.
 */
type GoogleReasoningMetadata = {
    thoughtSignature?: string;
};

type GoogleGenAIClient = {
    models: GoogleGenAI['models'];
};
type GoogleGenAIProviderBaseOptions = {
    apiKey?: string;
    apiVersion?: string;
    baseUrl?: string;
    client?: GoogleGenAIClient;
};
type GoogleGenAIProviderFactoryOptions = {
    providerId?: string;
};
type GoogleGenAIProviderOptions = GoogleGenAIProviderBaseOptions;
type GoogleGenAIProvider = {
    chatModel(modelId: string): ChatModel;
    embeddingModel(modelId: string): EmbeddingModel;
    imageModel(modelId: string): ImageModel;
};
declare function createGoogleGenAIProvider(options?: GoogleGenAIProviderBaseOptions, factoryOptions?: GoogleGenAIProviderFactoryOptions): GoogleGenAIProvider;
declare function createGoogleGenAI(options?: GoogleGenAIProviderOptions): GoogleGenAIProvider;

declare const googleGenerateProviderOptionsSchema: z.ZodObject<{
    stopSequences: z.ZodOptional<z.ZodArray<z.ZodString>>;
    frequencyPenalty: z.ZodOptional<z.ZodNumber>;
    presencePenalty: z.ZodOptional<z.ZodNumber>;
    seed: z.ZodOptional<z.ZodNumber>;
    topK: z.ZodOptional<z.ZodNumber>;
}, z.core.$strict>;
type GoogleGenerateProviderOptions = z.infer<typeof googleGenerateProviderOptionsSchema>;
declare const googleEmbedProviderOptionsSchema: z.ZodObject<{
    taskType: z.ZodOptional<z.ZodString>;
    title: z.ZodOptional<z.ZodString>;
    mimeType: z.ZodOptional<z.ZodString>;
    autoTruncate: z.ZodOptional<z.ZodBoolean>;
}, z.core.$strict>;
type GoogleEmbedProviderOptions = z.infer<typeof googleEmbedProviderOptionsSchema>;
declare const googleImageProviderOptionsSchema: z.ZodObject<{
    outputGcsUri: z.ZodOptional<z.ZodString>;
    negativePrompt: z.ZodOptional<z.ZodString>;
    aspectRatio: z.ZodOptional<z.ZodString>;
    guidanceScale: z.ZodOptional<z.ZodNumber>;
    seed: z.ZodOptional<z.ZodNumber>;
    safetyFilterLevel: z.ZodOptional<z.ZodEnum<{
        BLOCK_LOW_AND_ABOVE: "BLOCK_LOW_AND_ABOVE";
        BLOCK_MEDIUM_AND_ABOVE: "BLOCK_MEDIUM_AND_ABOVE";
        BLOCK_ONLY_HIGH: "BLOCK_ONLY_HIGH";
        BLOCK_NONE: "BLOCK_NONE";
    }>>;
    personGeneration: z.ZodOptional<z.ZodEnum<{
        DONT_ALLOW: "DONT_ALLOW";
        ALLOW_ADULT: "ALLOW_ADULT";
        ALLOW_ALL: "ALLOW_ALL";
    }>>;
    includeSafetyAttributes: z.ZodOptional<z.ZodBoolean>;
    includeRaiReason: z.ZodOptional<z.ZodBoolean>;
    language: z.ZodOptional<z.ZodString>;
    outputMimeType: z.ZodOptional<z.ZodString>;
    outputCompressionQuality: z.ZodOptional<z.ZodNumber>;
    addWatermark: z.ZodOptional<z.ZodBoolean>;
    labels: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
    imageSize: z.ZodOptional<z.ZodString>;
    enhancePrompt: z.ZodOptional<z.ZodBoolean>;
}, z.core.$strict>;
type GoogleImageProviderOptions = z.infer<typeof googleImageProviderOptionsSchema>;
declare module '@core-ai/core-ai' {
    interface GenerateProviderOptions {
        google?: GoogleGenerateProviderOptions;
    }
    interface EmbedProviderOptions {
        google?: GoogleEmbedProviderOptions;
    }
    interface ImageProviderOptions {
        google?: GoogleImageProviderOptions;
    }
}
declare const googleProviderOptionsSchema: z.ZodObject<{
    stopSequences: z.ZodOptional<z.ZodArray<z.ZodString>>;
    frequencyPenalty: z.ZodOptional<z.ZodNumber>;
    presencePenalty: z.ZodOptional<z.ZodNumber>;
    seed: z.ZodOptional<z.ZodNumber>;
    topK: z.ZodOptional<z.ZodNumber>;
}, z.core.$strict>;
type GoogleProviderOptions = GoogleGenerateProviderOptions;

export { type GoogleEmbedProviderOptions, type GoogleGenAIClient, type GoogleGenAIProvider, type GoogleGenAIProviderBaseOptions, type GoogleGenAIProviderFactoryOptions, type GoogleGenAIProviderOptions, type GoogleGenerateProviderOptions, type GoogleImageProviderOptions, type GoogleModelCapabilities, type GoogleProviderOptions as GoogleModelProviderOptions, type GoogleReasoningMetadata, createGoogleGenAI, createGoogleGenAIProvider, getGoogleModelCapabilities, googleEmbedProviderOptionsSchema, googleGenerateProviderOptionsSchema, googleImageProviderOptionsSchema, googleProviderOptionsSchema };