[test] **kwargs supports #676

tests/recognizers/test_openai.py

@@ -7,9 +7,9 @@
 from speech_recognition import AudioData, Recognizer
 from speech_recognition.recognizers import openai
 
+
 @pytest.fixture
 def setenv_openai_api_key(monkeypatch):
-def test_transcribe_with_openai_whisper(respx_mock, monkeypatch):
     monkeypatch.setenv("OPENAI_API_KEY", "sk_openai_api_key")
 
 
@@ -36,7 +36,7 @@ def test_transcribe_with_openai_whisper(respx_mock, setenv_openai_api_key):
 
 
 @respx.mock(assert_all_called=True, assert_all_mocked=True)
-def test_transcribe_with_specified_language(respx_mock, monkeypatch):
+def test_transcribe_with_specified_language(respx_mock, setenv_openai_api_key):
     # https://github.com/Uberi/speech_recognition/issues/681
     respx_mock.post(
         "https://api.openai.com/v1/audio/transcriptions",
@@ -54,3 +54,27 @@ def test_transcribe_with_specified_language(respx_mock, monkeypatch):
     )
 
     assert actual == "English transcription"
+
+
+@respx.mock(assert_all_called=True, assert_all_mocked=True)
+def test_transcribe_with_specified_prompt(respx_mock, setenv_openai_api_key):
+    # https://github.com/Uberi/speech_recognition/pull/676
+    respx_mock.post(
+        "https://api.openai.com/v1/audio/transcriptions",
+        # ref: https://cookbook.openai.com/examples/whisper_prompting_guide
+        data__contains={"prompt": "Glossary: Aimee, Shawn, BBQ"},
+    ).respond(
+        200,
+        json={"text": "Prompted transcription"},
+    )
+
+    audio_data = MagicMock(spec=AudioData)
+    audio_data.get_wav_data.return_value = b"audio_data"
+
+    actual = openai.recognize(
+        MagicMock(spec=Recognizer),
+        audio_data,
+        prompt="Glossary: Aimee, Shawn, BBQ",
+    )
+
+    assert actual == "Prompted transcription"