import { z, ActionMetadata } from 'genkit';
import { ModelReference, ModelAction } from 'genkit/model';
import { GoogleAIPluginOptions, Model } from './types.js';
import '../common/interaction-types.js';
import '../common/types.js';

/**
 * Copyright 2026 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

declare const AntigravityConfigSchema: z.ZodObject<{
    apiKey: z.ZodOptional<z.ZodString>;
    baseUrl: z.ZodOptional<z.ZodString>;
    apiVersion: z.ZodOptional<z.ZodString>;
    previousInteractionId: z.ZodOptional<z.ZodString>;
    store: z.ZodOptional<z.ZodBoolean>;
    environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
    responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
    model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
}, "passthrough", z.ZodTypeAny, z.objectOutputType<{
    apiKey: z.ZodOptional<z.ZodString>;
    baseUrl: z.ZodOptional<z.ZodString>;
    apiVersion: z.ZodOptional<z.ZodString>;
    previousInteractionId: z.ZodOptional<z.ZodString>;
    store: z.ZodOptional<z.ZodBoolean>;
    environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
    responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
    model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
}, z.ZodTypeAny, "passthrough">, z.objectInputType<{
    apiKey: z.ZodOptional<z.ZodString>;
    baseUrl: z.ZodOptional<z.ZodString>;
    apiVersion: z.ZodOptional<z.ZodString>;
    previousInteractionId: z.ZodOptional<z.ZodString>;
    store: z.ZodOptional<z.ZodBoolean>;
    environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
    responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
    model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
}, z.ZodTypeAny, "passthrough">>;
type AntigravityConfigSchemaType = typeof AntigravityConfigSchema;
type AntigravityConfig = z.infer<AntigravityConfigSchemaType>;
type ConfigSchemaType = AntigravityConfigSchemaType;
declare const KNOWN_MODELS: {
    readonly 'antigravity-preview-09-2026': ModelReference<z.ZodObject<{
        apiKey: z.ZodOptional<z.ZodString>;
        baseUrl: z.ZodOptional<z.ZodString>;
        apiVersion: z.ZodOptional<z.ZodString>;
        previousInteractionId: z.ZodOptional<z.ZodString>;
        store: z.ZodOptional<z.ZodBoolean>;
        environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
        responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
        model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
    }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
        apiKey: z.ZodOptional<z.ZodString>;
        baseUrl: z.ZodOptional<z.ZodString>;
        apiVersion: z.ZodOptional<z.ZodString>;
        previousInteractionId: z.ZodOptional<z.ZodString>;
        store: z.ZodOptional<z.ZodBoolean>;
        environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
        responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
        model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
    }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
        apiKey: z.ZodOptional<z.ZodString>;
        baseUrl: z.ZodOptional<z.ZodString>;
        apiVersion: z.ZodOptional<z.ZodString>;
        previousInteractionId: z.ZodOptional<z.ZodString>;
        store: z.ZodOptional<z.ZodBoolean>;
        environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
        responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
        model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
    }, z.ZodTypeAny, "passthrough">>>;
};
type KnownModels = keyof typeof KNOWN_MODELS;
type AntigravityModelName = `antigravity-${string}`;
declare function isAntigravityModelName(value?: string): value is AntigravityModelName;
declare function model(version: string, config?: AntigravityConfig): ModelReference<ConfigSchemaType>;
declare function listActions(models: Model[]): ActionMetadata[];
declare function listKnownModels(options?: GoogleAIPluginOptions): ModelAction<z.ZodObject<{
    apiKey: z.ZodOptional<z.ZodString>;
    baseUrl: z.ZodOptional<z.ZodString>;
    apiVersion: z.ZodOptional<z.ZodString>;
    previousInteractionId: z.ZodOptional<z.ZodString>;
    store: z.ZodOptional<z.ZodBoolean>;
    environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
    responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
    model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
}, "passthrough", z.ZodTypeAny, z.objectOutputType<{
    apiKey: z.ZodOptional<z.ZodString>;
    baseUrl: z.ZodOptional<z.ZodString>;
    apiVersion: z.ZodOptional<z.ZodString>;
    previousInteractionId: z.ZodOptional<z.ZodString>;
    store: z.ZodOptional<z.ZodBoolean>;
    environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
    responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
    model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
}, z.ZodTypeAny, "passthrough">, z.objectInputType<{
    apiKey: z.ZodOptional<z.ZodString>;
    baseUrl: z.ZodOptional<z.ZodString>;
    apiVersion: z.ZodOptional<z.ZodString>;
    previousInteractionId: z.ZodOptional<z.ZodString>;
    store: z.ZodOptional<z.ZodBoolean>;
    environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
    responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
    model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
}, z.ZodTypeAny, "passthrough">>>[];
declare function defineModel(name: string, pluginOptions?: GoogleAIPluginOptions): ModelAction<AntigravityConfigSchemaType>;
declare const TEST_ONLY: {
    KNOWN_MODELS: {
        readonly 'antigravity-preview-09-2026': ModelReference<z.ZodObject<{
            apiKey: z.ZodOptional<z.ZodString>;
            baseUrl: z.ZodOptional<z.ZodString>;
            apiVersion: z.ZodOptional<z.ZodString>;
            previousInteractionId: z.ZodOptional<z.ZodString>;
            store: z.ZodOptional<z.ZodBoolean>;
            environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
            responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
            model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
        }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
            apiKey: z.ZodOptional<z.ZodString>;
            baseUrl: z.ZodOptional<z.ZodString>;
            apiVersion: z.ZodOptional<z.ZodString>;
            previousInteractionId: z.ZodOptional<z.ZodString>;
            store: z.ZodOptional<z.ZodBoolean>;
            environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
            responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
            model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
        }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
            apiKey: z.ZodOptional<z.ZodString>;
            baseUrl: z.ZodOptional<z.ZodString>;
            apiVersion: z.ZodOptional<z.ZodString>;
            previousInteractionId: z.ZodOptional<z.ZodString>;
            store: z.ZodOptional<z.ZodBoolean>;
            environment: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
            responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE"]>, "many">>;
            model: z.ZodOptional<z.ZodUnion<[z.ZodEnum<["gemini-4-argon", "gemini-3.8-flash", "gemini-3.6-flash", "gemini-3.5-flash-lite"]>, z.ZodType<string & {}, z.ZodTypeDef, string & {}>]>>;
        }, z.ZodTypeAny, "passthrough">>>;
    };
};

export { type AntigravityConfig, AntigravityConfigSchema, type AntigravityConfigSchemaType, type AntigravityModelName, type KnownModels, TEST_ONLY, defineModel, isAntigravityModelName, listActions, listKnownModels, model };