cleanup test

Author: tschellenbach

conftest.py

@@ -16,9 +16,9 @@
 from vision_agents.core.edge.types import PcmData
 from vision_agents.core.stt.events import STTTranscriptEvent, STTErrorEvent
 
-
 load_dotenv()
 
+
 class STTSession:
     """Helper class for testing STT implementations.
     
@@ -65,6 +65,14 @@ async def wait_for_result(self, timeout: float = 30.0):
 
         # Wait for an event
         await asyncio.wait_for(self._event.wait(), timeout=timeout)
+    
+    def get_full_transcript(self) -> str:
+        """Get full transcription text from all transcript events.
+        
+        Returns:
+            Combined text from all transcripts
+        """
+        return " ".join(t.text for t in self.transcripts)
 
 
 def get_assets_dir():

plugins/deepgram/tests/test_deepgram_stt.py

@@ -0,0 +1,34 @@
+import pytest
+
+from vision_agents.plugins import deepgram
+from conftest import STTSession
+
+
+class TestDeepgramSTT:
+    """Integration tests for Deepgram STT"""
+
+    @pytest.fixture
+    async def stt(self):
+        """Create and manage Deepgram STT lifecycle"""
+        stt = deepgram.STT()
+        try:
+            yield stt
+        finally:
+            await stt.close()
+
+    @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()
+