@mastra/voice-deepgram 0.0.0-storage-20250225005900 → 0.0.0-stream-vnext-usage-20250908171242

package/src/index.test.ts

@@ -1,134 +0,0 @@
-import { writeFileSync, mkdirSync, createReadStream } from 'fs';
-import path from 'path';
-import { PassThrough } from 'stream';
-import { describe, expect, it, beforeAll } from 'vitest';
-
-import { DeepgramVoice } from './index.js';
-
-describe('DeepgramVoice Integration Tests', () => {
-  let voice: DeepgramVoice;
-  const outputDir = path.join(process.cwd(), 'test-outputs');
-
-  beforeAll(() => {
-    try {
-      mkdirSync(outputDir, { recursive: true });
-    } catch (err) {
-      console.log('Directory already exists: ', err);
-    }
-
-    voice = new DeepgramVoice({
-      speechModel: {
-        name: 'aura',
-      },
-      listeningModel: {
-        name: 'whisper',
-      },
-      speaker: 'asteria-en',
-    });
-  });
-
-  describe('getSpeakers', () => {
-    it('should list available voices', async () => {
-      const speakers = await voice.getSpeakers();
-      const expectedVoiceIds = ['asteria-en', 'stella-en', 'luna-en'];
-      expectedVoiceIds.forEach(voiceId => {
-        expect(speakers.some(s => s.voiceId === voiceId)).toBe(true);
-      });
-    });
-  });
-
-  describe('speak', () => {
-    it('should generate audio and save to file', async () => {
-      const audioResult = await voice.speak('Hello World', {
-        text: 'Hello World',
-      });
-
-      const chunks: Buffer[] = [];
-      for await (const chunk of audioResult) {
-        chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
-      }
-      const audioBuffer = Buffer.concat(chunks);
-      const outputPath = path.join(outputDir, 'deepgram-speech-test.mp3');
-      writeFileSync(outputPath, audioBuffer);
-      expect(audioBuffer.length).toBeGreaterThan(0);
-    }, 10000);
-
-    it('should work with different parameters', async () => {
-      const audioResult = await voice.speak('Hello World', {
-        text: 'Test with parameters',
-        speaker: 'luna-en',
-      });
-
-      const chunks: Buffer[] = [];
-      for await (const chunk of audioResult) {
-        chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
-      }
-      const audioBuffer = Buffer.concat(chunks);
-      const outputPath = path.join(outputDir, 'deepgram-speech-test-params.mp3');
-      writeFileSync(outputPath, audioBuffer);
-      expect(audioBuffer.length).toBeGreaterThan(0);
-    }, 10000);
-  });
-
-  // Error cases
-  describe('error handling', () => {
-    it('should handle invalid voice names', async () => {
-      await expect(voice.speak('Test', { speaker: 'invalid_voice' })).rejects.toThrow();
-    });
-
-    it('should handle empty text', async () => {
-      await expect(voice.speak('', { speaker: 'asteria-en' })).rejects.toThrow('Input text is empty');
-    });
-
-    it('should handle whitespace-only text', async () => {
-      await expect(voice.speak('   \n\t  ', { speaker: 'asteria-en' })).rejects.toThrow('Input text is empty');
-    });
-  });
-
-  describe('listen', () => {
-    it('should transcribe audio buffer', async () => {
-      // First generate some audio to transcribe
-      const audioResult = await voice.speak('This is a test of transcription');
-
-      // Collect audio chunks
-      const chunks: Buffer[] = [];
-      for await (const chunk of audioResult) {
-        chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
-      }
-      const audioBuffer = Buffer.concat(chunks);
-
-      // Create stream from the buffer
-      const stream = new PassThrough();
-      stream.end(audioBuffer);
-      const result = await voice.listen(stream);
-
-      expect(typeof result).toBe('string');
-      expect(result.toLowerCase()).toContain('test');
-      expect(result.toLowerCase()).toContain('transcription');
-    }, 15000);
-
-    it('should transcribe audio from fixture file', async () => {
-      const fixturePath = path.join(process.cwd(), '__fixtures__', 'voice-test.m4a');
-      const audioStream = createReadStream(fixturePath);
-
-      console.log('listening to audio stream');
-      const text = await voice.listen(audioStream, {
-        filetype: 'm4a',
-      });
-      console.log('text', text);
-
-      expect(text).toBeTruthy();
-      console.log(text);
-      expect(typeof text).toBe('string');
-      expect(text.length).toBeGreaterThan(0);
-    }, 15000);
-
-    it('should handle invalid audio', async () => {
-      const invalidAudio = Buffer.from('not valid audio');
-      const stream = new PassThrough();
-      stream.end(invalidAudio);
-
-      await expect(voice.listen(stream)).rejects.toThrow();
-    });
-  });
-});

package/src/index.ts

@@ -1,190 +0,0 @@
-import { createClient } from '@deepgram/sdk';
-import { MastraVoice } from '@mastra/core/voice';
-import { PassThrough } from 'stream';
-
-import { DEEPGRAM_VOICES } from './voices';
-import type { DeepgramVoiceId, DeepgramModel } from './voices';
-
-interface DeepgramVoiceConfig {
-  name?: DeepgramModel;
-  apiKey?: string;
-  properties?: Record<string, any>;
-  language?: string;
-}
-
-export class DeepgramVoice extends MastraVoice {
-  private speechClient?: ReturnType<typeof createClient>;
-  private listeningClient?: ReturnType<typeof createClient>;
-
-  constructor({
-    speechModel,
-    listeningModel,
-    speaker,
-  }: { speechModel?: DeepgramVoiceConfig; listeningModel?: DeepgramVoiceConfig; speaker?: DeepgramVoiceId } = {}) {
-    const defaultApiKey = process.env.DEEPGRAM_API_KEY;
-
-    const defaultSpeechModel = {
-      name: 'aura',
-      apiKey: defaultApiKey,
-    };
-
-    const defaultListeningModel = {
-      name: 'nova',
-      apiKey: defaultApiKey,
-    };
-
-    super({
-      speechModel: {
-        name: speechModel?.name ?? defaultSpeechModel.name,
-        apiKey: speechModel?.apiKey ?? defaultSpeechModel.apiKey,
-      },
-      listeningModel: {
-        name: listeningModel?.name ?? defaultListeningModel.name,
-        apiKey: listeningModel?.apiKey ?? defaultListeningModel.apiKey,
-      },
-      speaker,
-    });
-
-    const speechApiKey = speechModel?.apiKey || defaultApiKey;
-    const listeningApiKey = listeningModel?.apiKey || defaultApiKey;
-    console.log('speechApiKey', speechApiKey);
-    console.log('listeningApiKey', listeningApiKey);
-    if (!speechApiKey && !listeningApiKey) {
-      throw new Error('At least one of DEEPGRAM_API_KEY, speechModel.apiKey, or listeningModel.apiKey must be set');
-    }
-
-    if (speechApiKey) {
-      this.speechClient = createClient(speechApiKey);
-    }
-    if (listeningApiKey) {
-      this.listeningClient = createClient(listeningApiKey);
-    }
-
-    this.speaker = speaker || 'asteria-en';
-  }
-
-  async getSpeakers() {
-    return this.traced(async () => {
-      return DEEPGRAM_VOICES.map(voice => ({
-        voiceId: voice,
-      }));
-    }, 'voice.deepgram.getSpeakers')();
-  }
-
-  async speak(
-    input: string | NodeJS.ReadableStream,
-    options?: {
-      speaker?: string;
-      [key: string]: any;
-    },
-  ): Promise<NodeJS.ReadableStream> {
-    if (!this.speechClient) {
-      throw new Error('Deepgram speech client not configured');
-    }
-
-    let text: string;
-    if (typeof input !== 'string') {
-      const chunks: Buffer[] = [];
-      for await (const chunk of input) {
-        chunks.push(Buffer.from(chunk));
-      }
-      text = Buffer.concat(chunks).toString('utf-8');
-    } else {
-      text = input;
-    }
-
-    if (text.trim().length === 0) {
-      throw new Error('Input text is empty');
-    }
-
-    return this.traced(async () => {
-      if (!this.speechClient) {
-        throw new Error('No speech client configured');
-      }
-
-      let model;
-      if (options?.speaker) {
-        model = this.speechModel?.name + '-' + options.speaker;
-      } else if (this.speaker) {
-        model = this.speechModel?.name + '-' + this.speaker;
-      }
-
-      const speakClient = this.speechClient.speak;
-      const response = await speakClient.request(
-        { text },
-        {
-          model,
-          ...options,
-        },
-      );
-
-      const webStream = await response.getStream();
-      if (!webStream) {
-        throw new Error('No stream returned from Deepgram');
-      }
-
-      const reader = webStream.getReader();
-      const nodeStream = new PassThrough();
-
-      // Add error handling for the stream processing
-      (async () => {
-        try {
-          while (true) {
-            const { done, value } = await reader.read();
-            if (done) {
-              nodeStream.end();
-              break;
-            }
-            nodeStream.write(value);
-          }
-        } catch (error) {
-          nodeStream.destroy(error as Error);
-        }
-      })().catch(error => {
-        nodeStream.destroy(error as Error);
-      });
-
-      return nodeStream;
-    }, 'voice.deepgram.speak')();
-  }
-
-  async listen(
-    audioStream: NodeJS.ReadableStream,
-    options?: {
-      [key: string]: any;
-    },
-  ): Promise<string> {
-    if (!this.listeningClient) {
-      throw new Error('Deepgram listening client not configured');
-    }
-
-    const chunks: Buffer[] = [];
-    for await (const chunk of audioStream) {
-      chunks.push(Buffer.from(chunk));
-    }
-    const buffer = Buffer.concat(chunks);
-
-    return this.traced(async () => {
-      if (!this.listeningClient) {
-        throw new Error('No listening client configured');
-      }
-      const { result, error } = await this.listeningClient.listen.prerecorded.transcribeFile(buffer, {
-        model: this.listeningModel?.name,
-        ...options,
-      });
-
-      if (error) {
-        throw error;
-      }
-
-      const transcript = result.results?.channels?.[0]?.alternatives?.[0]?.transcript;
-      if (!transcript) {
-        throw new Error('No transcript found in Deepgram response');
-      }
-
-      return transcript;
-    }, 'voice.deepgram.listen')();
-  }
-}
-
-export type { DeepgramVoiceConfig, DeepgramVoiceId, DeepgramModel };

package/tsconfig.json

@@ -1,5 +0,0 @@
-{
-  "extends": "../../tsconfig.node.json",
-  "include": ["src/**/*"],
-  "exclude": ["node_modules", "**/*.test.ts"]
-}

package/vitest.config.ts

@@ -1,9 +0,0 @@
-import { defineConfig } from 'vitest/config';
-
-export default defineConfig({
-  test: {
-    globals: true,
-    environment: 'node',
-    include: ['src/**/*.test.ts'],
-  },
-});