ai/src/test/mock-speech-model-v2.ts

AI SDK by Vercel - The AI Toolkit for TypeScript and JavaScript

import { SpeechModelV2 } from '@ai-sdk/provider';
import { notImplemented } from './not-implemented';

export class MockSpeechModelV2 implements SpeechModelV2 {
  readonly specificationVersion = 'v2';
  readonly provider: SpeechModelV2['provider'];
  readonly modelId: SpeechModelV2['modelId'];

  doGenerate: SpeechModelV2['doGenerate'];

  constructor({
    provider = 'mock-provider',
    modelId = 'mock-model-id',
    doGenerate = notImplemented,
  }: {
    provider?: SpeechModelV2['provider'];
    modelId?: SpeechModelV2['modelId'];
    doGenerate?: SpeechModelV2['doGenerate'];
  } = {}) {
    this.provider = provider;
    this.modelId = modelId;
    this.doGenerate = doGenerate;
  }
}