import { ModelReference, z, ActionMetadata } from 'genkit';
import { ModelAction } from 'genkit/model';
import { ClientOptions, VertexPluginOptions, Model } from './types.js';
import 'google-auth-library';
import '../common/types.js';

/**
 * Copyright 2025 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 LyriaConfigSchema: z.ZodObject<{
    negativePrompt: z.ZodOptional<z.ZodString>;
    seed: z.ZodOptional<z.ZodNumber>;
    sampleCount: z.ZodOptional<z.ZodNumber>;
    location: z.ZodOptional<z.ZodString>;
}, "passthrough", z.ZodTypeAny, z.objectOutputType<{
    negativePrompt: z.ZodOptional<z.ZodString>;
    seed: z.ZodOptional<z.ZodNumber>;
    sampleCount: z.ZodOptional<z.ZodNumber>;
    location: z.ZodOptional<z.ZodString>;
}, z.ZodTypeAny, "passthrough">, z.objectInputType<{
    negativePrompt: z.ZodOptional<z.ZodString>;
    seed: z.ZodOptional<z.ZodNumber>;
    sampleCount: z.ZodOptional<z.ZodNumber>;
    location: z.ZodOptional<z.ZodString>;
}, z.ZodTypeAny, "passthrough">>;
type LyriaConfigSchemaType = typeof LyriaConfigSchema;
type LyriaConfig = z.infer<LyriaConfigSchemaType>;
declare const Lyria3ConfigSchema: z.ZodObject<{
    location: z.ZodOptional<z.ZodString>;
    responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE", "AUDIO"]>, "many">>;
}, "passthrough", z.ZodTypeAny, z.objectOutputType<{
    location: z.ZodOptional<z.ZodString>;
    responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE", "AUDIO"]>, "many">>;
}, z.ZodTypeAny, "passthrough">, z.objectInputType<{
    location: z.ZodOptional<z.ZodString>;
    responseModalities: z.ZodOptional<z.ZodArray<z.ZodEnum<["TEXT", "IMAGE", "AUDIO"]>, "many">>;
}, z.ZodTypeAny, "passthrough">>;
type Lyria3ConfigSchemaType = typeof Lyria3ConfigSchema;
type Lyria3Config = z.infer<Lyria3ConfigSchemaType>;
type ConfigSchemaType = LyriaConfigSchemaType | Lyria3ConfigSchemaType;
type ConfigSchema = z.infer<ConfigSchemaType>;
declare const KNOWN_LYRIA_LEGACY_MODELS: {
    readonly 'lyria-002': ModelReference<ConfigSchemaType>;
};
type KnownLyriaLegacyModels = keyof typeof KNOWN_LYRIA_LEGACY_MODELS;
type LyriaLegacyModelName = `lyria-002`;
declare const KNOWN_LYRIA_INTERACTIONS_MODELS: {
    readonly 'lyria-3-clip-preview': ModelReference<ConfigSchemaType>;
    readonly 'lyria-3-pro-preview': ModelReference<ConfigSchemaType>;
};
type KnownLyriaInteractionsModels = keyof typeof KNOWN_LYRIA_INTERACTIONS_MODELS;
declare const KNOWN_MODELS: {
    'lyria-3-clip-preview': ModelReference<ConfigSchemaType>;
    'lyria-3-pro-preview': ModelReference<ConfigSchemaType>;
    'lyria-002': ModelReference<ConfigSchemaType>;
};
type KnownModels = keyof typeof KNOWN_MODELS;
type LyriaModelName = `lyria-${string}`;
declare function isLyriaModelName(value?: string): value is LyriaModelName;
declare function model(version: string, config?: ConfigSchema): ModelReference<ConfigSchemaType>;
declare function listActions(models: Model[]): ActionMetadata[];
declare function listKnownModels(clientOptions: ClientOptions, pluginOptions?: VertexPluginOptions): ModelAction<ConfigSchemaType>[];
declare function defineModel(name: string, clientOptions: ClientOptions, pluginOptions?: VertexPluginOptions): ModelAction<ConfigSchemaType>;
declare const TEST_ONLY: {
    GENERIC_MODEL: ModelReference<ConfigSchemaType>;
    KNOWN_MODELS: {
        'lyria-3-clip-preview': ModelReference<ConfigSchemaType>;
        'lyria-3-pro-preview': ModelReference<ConfigSchemaType>;
        'lyria-002': ModelReference<ConfigSchemaType>;
    };
    KNOWN_LYRIA_LEGACY_MODELS: {
        readonly 'lyria-002': ModelReference<ConfigSchemaType>;
    };
    KNOWN_LYRIA_INTERACTIONS_MODELS: {
        readonly 'lyria-3-clip-preview': ModelReference<ConfigSchemaType>;
        readonly 'lyria-3-pro-preview': ModelReference<ConfigSchemaType>;
    };
    HIDDEN_MODELS: {};
    ALL_MODELS: {
        'lyria-3-clip-preview': ModelReference<ConfigSchemaType>;
        'lyria-3-pro-preview': ModelReference<ConfigSchemaType>;
        'lyria-002': ModelReference<ConfigSchemaType>;
    };
};

export { type KnownLyriaInteractionsModels, type KnownLyriaLegacyModels, type KnownModels, type Lyria3Config, Lyria3ConfigSchema, type Lyria3ConfigSchemaType, type LyriaConfig, LyriaConfigSchema, type LyriaConfigSchemaType, type LyriaLegacyModelName, type LyriaModelName, TEST_ONLY, defineModel, isLyriaModelName, listActions, listKnownModels, model };