import {
  lazySchema,
  zodSchema,
  type InferSchema,
} from '@ai-sdk/provider-utils';
import { z } from 'zod/v4';

export type GoogleTranscriptionModelId =
  | 'gemini-3.5-transcribe'
  | 'gemini-3.5-transcribe-live'
  | (string & {});

/**
 * Speech recognition options shared by unary (`gemini-3.5-transcribe`) and
 * live (`gemini-3.5-transcribe-live`) transcription. Maps onto Google's
 * `AudioTranscriptionConfig`.
 */
export const googleTranscriptionModelOptions = lazySchema(() =>
  zodSchema(
    z.object({
      /**
       * BCP-47 language codes providing hints about the languages present in the
       * audio. If omitted or empty, defaults to automatic language detection.
       */
      languageCodes: z.array(z.string()).optional(),

      /**
       * Custom vocabulary phrases, which bias the speech recognition model
       * toward recognizing specific terms.
       */
      customVocabulary: z.array(z.string()).optional(),

      /**
       * Enables word-level timestamp generation.
       */
      wordTimestamp: z.boolean().optional(),

      /**
       * Enables speaker diarization.
       */
      diarization: z.boolean().optional(),

      /**
       * Transcription output formatting mode.
       *
       * - `VERBATIM` (default): exact literal transcript preserving filler
       *   words, repetitions, and false starts.
       * - `SMART`: cleans up and structures the transcript in real time —
       *   disfluency removal, inline self-corrections, structured formatting
       *   (lists, numbers, dates, paragraph breaks), and grammar/casing polish.
       */
      mode: z.enum(['SMART', 'VERBATIM']).optional(),
    }),
  ),
);

export type GoogleTranscriptionModelOptions = InferSchema<
  typeof googleTranscriptionModelOptions
>;