cleanup of stt

Author: tschellenbach

plugins/deepgram/vision_agents/plugins/deepgram/stt.py

@@ -3,7 +3,7 @@
 import logging
 import os
 import time
-from typing import TYPE_CHECKING, Any, Dict, Optional
+from typing import Any, Dict, Optional
 
 import numpy as np
 import websockets

plugins/fish/vision_agents/plugins/fish/stt.py

@@ -2,7 +2,7 @@
 import logging
 import os
 import wave
-from typing import TYPE_CHECKING, Any, Dict, Optional
+from typing import Optional
 
 import numpy as np
 from fish_audio_sdk import Session, ASRRequest

plugins/moonshine/tests/test_moonshine_stt.py

@@ -0,0 +1,50 @@
+import pytest
+
+from vision_agents.plugins import moonshine
+from conftest import STTSession
+
+
+class TestMoonshineSTT:
+    """Integration tests for Moonshine STT"""
+
+    @pytest.fixture
+    async def stt(self):
+        """Create and manage Moonshine STT lifecycle"""
+        stt = moonshine.STT()
+        try:
+            yield stt
+        finally:
+            await stt.close()
+
+    @pytest.mark.integration
+    async def test_transcribe_mia_audio(self, stt, mia_audio_16khz):
+        # Create session to collect transcripts and errors
+        session = STTSession(stt)
+        
+        # Process the audio
+        await stt.process_audio(mia_audio_16khz)
+        
+        # Wait for result
+        await session.wait_for_result(timeout=30.0)
+        assert not session.errors
+        
+        # Verify transcript
+        full_transcript = session.get_full_transcript()
+        assert "forgotten treasures" in full_transcript.lower()
+
+    @pytest.mark.integration
+    async def test_transcribe_mia_audio_48khz(self, stt, mia_audio_48khz):
+        # Create session to collect transcripts and errors
+        session = STTSession(stt)
+        
+        # Process the audio
+        await stt.process_audio(mia_audio_48khz)
+        
+        # Wait for result
+        await session.wait_for_result(timeout=30.0)
+        assert not session.errors
+        
+        # Verify transcript
+        full_transcript = session.get_full_transcript()
+        assert "forgotten treasures" in full_transcript.lower()
+