diff --git a/litellm/tests/test_alangfuse.py b/litellm/tests/test_alangfuse.py index 8f91ffa67..097974e2f 100644 --- a/litellm/tests/test_alangfuse.py +++ b/litellm/tests/test_alangfuse.py @@ -245,6 +245,48 @@ async def test_langfuse_logging_without_request_response(stream, langfuse_client pytest.fail(f"An exception occurred - {e}") +# Get the current directory of the file being run +pwd = os.path.dirname(os.path.realpath(__file__)) +print(pwd) + +file_path = os.path.join(pwd, "gettysburg.wav") + +audio_file = open(file_path, "rb") + + +@pytest.mark.asyncio +async def test_langfuse_logging_audio_transcriptions(langfuse_client): + """ + Test that creates a trace with masked input and output + """ + import uuid + + _unique_trace_name = f"litellm-test-{str(uuid.uuid4())}" + litellm.set_verbose = True + litellm.success_callback = ["langfuse"] + await litellm.atranscription( + model="whisper-1", + file=audio_file, + metadata={ + "trace_id": _unique_trace_name, + }, + ) + + langfuse_client.flush() + await asyncio.sleep(2) + + # get trace with _unique_trace_name + trace = langfuse_client.get_trace(id=_unique_trace_name) + generations = list( + reversed(langfuse_client.get_generations(trace_id=_unique_trace_name).data) + ) + + print("generations for given trace=", generations) + + assert len(generations) == 1 + assert generations[0].name == "litellm-atranscription" + + @pytest.mark.asyncio async def test_langfuse_masked_input_output(langfuse_client): """