mirror of
https://github.com/deepset-ai/haystack.git
synced 2026-01-08 21:28:00 +00:00
bf6d306d68
* ci: Simplify Python code with ruff rules SIM * Revert #5828 * ruff --select=I --fix haystack/modeling/infer.py --------- Co-authored-by: Massimiliano Pippi <mpippi@gmail.com>
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
from haystack.preview.components.audio.whisper_remote import RemoteWhisperTranscriber
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
not os.environ.get("OPENAI_API_KEY", None),
|
|
reason="Export an env var called OPENAI_API_KEY containing the OpenAI API key to run this test.",
|
|
)
|
|
def test_whisper_remote_transcriber(preview_samples_path):
|
|
comp = RemoteWhisperTranscriber(api_key=os.environ.get("OPENAI_API_KEY"))
|
|
|
|
output = comp.run(
|
|
audio_files=[
|
|
preview_samples_path / "audio" / "this is the content of the document.wav",
|
|
str((preview_samples_path / "audio" / "the context for this answer is here.wav").absolute()),
|
|
open(preview_samples_path / "audio" / "answer.wav", "rb"),
|
|
]
|
|
)
|
|
docs = output["documents"]
|
|
assert len(docs) == 3
|
|
|
|
assert docs[0].text.strip().lower() == "this is the content of the document."
|
|
assert preview_samples_path / "audio" / "this is the content of the document.wav" == docs[0].metadata["audio_file"]
|
|
|
|
assert docs[1].text.strip().lower() == "the context for this answer is here."
|
|
assert (
|
|
str((preview_samples_path / "audio" / "the context for this answer is here.wav").absolute())
|
|
== docs[1].metadata["audio_file"]
|
|
)
|
|
|
|
assert docs[2].text.strip().lower() == "answer."
|
|
assert docs[2].metadata["audio_file"] == "<<binary stream>>"
|