import * as _ai_sdk_provider from '@ai-sdk/provider';
import { TranscriptionModelV4, ProviderV4 } from '@ai-sdk/provider';
import { FetchFunction, WORKFLOW_SERIALIZE, WORKFLOW_DESERIALIZE } from '@ai-sdk/provider-utils';
import { z } from 'zod/v4';

type AssemblyAIConfig = {
    provider: string;
    url: (options: {
        modelId: string;
        path: string;
    }) => string;
    headers?: () => Record<string, string | undefined>;
    fetch?: FetchFunction;
    generateId?: () => string;
};

/**
 * Legacy AssemblyAI speech model, sent via the deprecated singular
 * `speech_model` request parameter.
 *
 * @deprecated Use `universal-3-5-pro` instead.
 * @see https://www.assemblyai.com/docs/pre-recorded-audio/select-the-speech-model
 */
type AssemblyAIDeprecatedTranscriptionModelId = 'best';
type AssemblyAITranscriptionModelId = 'universal-2' | 'universal-3-pro' | 'universal-3-5-pro' | AssemblyAIDeprecatedTranscriptionModelId | (string & {});

interface AssemblyAITranscriptionModelConfig extends AssemblyAIConfig {
    _internal?: {
        currentDate?: () => Date;
    };
    /**
     * The polling interval for checking transcript status in milliseconds.
     */
    pollingInterval?: number;
}
declare class AssemblyAITranscriptionModel implements TranscriptionModelV4 {
    readonly modelId: AssemblyAITranscriptionModelId;
    private readonly config;
    readonly specificationVersion = "v4";
    private readonly POLLING_INTERVAL_MS;
    get provider(): string;
    static [WORKFLOW_SERIALIZE](model: AssemblyAITranscriptionModel): {
        modelId: string;
        config: _ai_sdk_provider.JSONObject;
    };
    static [WORKFLOW_DESERIALIZE](options: {
        modelId: AssemblyAITranscriptionModelId;
        config: AssemblyAITranscriptionModelConfig;
    }): AssemblyAITranscriptionModel;
    constructor(modelId: AssemblyAITranscriptionModelId, config: AssemblyAITranscriptionModelConfig);
    private getArgs;
    /**
     * Polls the given transcript until we have a status other than `processing` or `queued`.
     *
     * @see https://www.assemblyai.com/docs/getting-started/transcribe-an-audio-file#step-33
     */
    private waitForCompletion;
    doGenerate(options: Parameters<TranscriptionModelV4['doGenerate']>[0]): Promise<Awaited<ReturnType<TranscriptionModelV4['doGenerate']>>>;
}

interface AssemblyAIProvider extends ProviderV4 {
    (modelId: AssemblyAITranscriptionModelId, settings?: {}): {
        transcription: AssemblyAITranscriptionModel;
    };
    /**
     * Creates a model for transcription.
     */
    transcription(modelId: AssemblyAITranscriptionModelId): TranscriptionModelV4;
    /**
     * @deprecated Use `embeddingModel` instead.
     */
    textEmbeddingModel(modelId: string): never;
}
interface AssemblyAIProviderSettings {
    /**
     * API key for authenticating requests.
     */
    apiKey?: string;
    /**
     * Custom headers to include in the requests.
     */
    headers?: Record<string, string>;
    /**
     * Custom fetch implementation. You can use it as a middleware to intercept requests,
     * or to provide a custom fetch implementation for e.g. testing.
     */
    fetch?: FetchFunction;
}
/**
 * Create an AssemblyAI provider instance.
 */
declare function createAssemblyAI(options?: AssemblyAIProviderSettings): AssemblyAIProvider;
/**
 * Default AssemblyAI provider instance.
 */
declare const assemblyai: AssemblyAIProvider;

declare const assemblyaiTranscriptionModelOptionsSchema: z.ZodObject<{
    audioEndAt: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
    audioStartFrom: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
    autoChapters: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    autoHighlights: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    boostParam: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    contentSafety: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    contentSafetyConfidence: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
    customSpelling: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodObject<{
        from: z.ZodArray<z.ZodString>;
        to: z.ZodString;
    }, z.core.$strip>>>>;
    disfluencies: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    domain: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    entityDetection: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    filterProfanity: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    formatText: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    iabCategories: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    keytermsPrompt: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString>>>;
    languageCode: z.ZodOptional<z.ZodNullable<z.ZodUnion<readonly [z.ZodLiteral<"en">, z.ZodString]>>>;
    languageConfidenceThreshold: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
    languageDetection: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    languageDetectionOptions: z.ZodOptional<z.ZodNullable<z.ZodObject<{
        expectedLanguages: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString>>>;
        fallbackLanguage: z.ZodOptional<z.ZodNullable<z.ZodString>>;
        codeSwitching: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
        codeSwitchingConfidenceThreshold: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
    }, z.core.$strip>>>;
    multichannel: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    prompt: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    punctuate: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    redactPii: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    redactPiiAudio: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    redactPiiAudioOptions: z.ZodOptional<z.ZodNullable<z.ZodObject<{
        returnRedactedNoSpeechAudio: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
        overrideAudioRedactionMethod: z.ZodOptional<z.ZodNullable<z.ZodEnum<{
            silence: "silence";
        }>>>;
    }, z.core.$strip>>>;
    redactPiiAudioQuality: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    redactPiiPolicies: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString>>>;
    redactPiiReturnUnredacted: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    redactPiiSub: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    redactStaticEntities: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodArray<z.ZodString>>>>;
    removeAudioTags: z.ZodOptional<z.ZodNullable<z.ZodEnum<{
        all: "all";
        speaker: "speaker";
    }>>>;
    sentimentAnalysis: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    speakerLabels: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    speakerOptions: z.ZodOptional<z.ZodNullable<z.ZodObject<{
        minSpeakersExpected: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
        maxSpeakersExpected: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
    }, z.core.$strip>>>;
    speakersExpected: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
    speechThreshold: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
    summarization: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
    summaryModel: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    summaryType: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    temperature: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
    webhookAuthHeaderName: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    webhookAuthHeaderValue: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    webhookUrl: z.ZodOptional<z.ZodNullable<z.ZodString>>;
    wordBoost: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString>>>;
}, z.core.$strip>;
type AssemblyAITranscriptionModelOptions = z.infer<typeof assemblyaiTranscriptionModelOptionsSchema>;

declare const VERSION: string;

export { type AssemblyAIProvider, type AssemblyAIProviderSettings, type AssemblyAITranscriptionModelOptions, VERSION, assemblyai, createAssemblyAI };