diff --git a/tests/basic/test_voice.py b/tests/basic/test_voice.py index dbfcb4987..e218d168a 100644 --- a/tests/basic/test_voice.py +++ b/tests/basic/test_voice.py @@ -75,11 +75,11 @@ def test_record_and_transcribe(mock_litellm, mock_soundfile): # Set up the mock stream to simulate some audio data mock_data = np.zeros((1000, 1)) mock_stream.return_value.__enter__.return_value = mock_data - + # Mock prompt_toolkit's prompt function more completely with patch("prompt_toolkit.shortcuts.prompt") as mock_prompt: mock_prompt.return_value = "" # Simulate pressing Enter - + # Mock the transcription response mock_litellm.transcription.return_value = Mock(text="Hello, world!")