diff --git a/tests/local_testing/test_custom_callback_input.py b/tests/local_testing/test_custom_callback_input.py index 3ec114dd32..9cc41869fe 100644 --- a/tests/local_testing/test_custom_callback_input.py +++ b/tests/local_testing/test_custom_callback_input.py @@ -1599,7 +1599,7 @@ def test_logging_key_masking_gemini(): assert "PART" == trimmed_key -@pytest.mark.parametrize("sync_mode", [True, False]) +@pytest.mark.parametrize("sync_mode", [True]) @pytest.mark.asyncio async def test_standard_logging_payload_stream_usage(sync_mode): """ @@ -1652,9 +1652,10 @@ async def test_standard_logging_payload_stream_usage(sync_mode): ) built_response = stream_chunk_builder(chunks=chunks) + print(f"built_response: {built_response}") assert ( built_response.usage.total_tokens - != standard_logging_object["total_tokens"] + == standard_logging_object["total_tokens"] ) print(f"standard_logging_object usage: {built_response.usage}") except litellm.InternalServerError: