diff --git a/tests/unit/proxy/spend_tracking/test_spend_tracking_utils.py b/tests/unit/proxy/spend_tracking/test_spend_tracking_utils.py index d558c41bc53..7a4d06082ac 100644 --- a/tests/unit/proxy/spend_tracking/test_spend_tracking_utils.py +++ b/tests/unit/proxy/spend_tracking/test_spend_tracking_utils.py @@ -237,7 +237,6 @@ def test_batch_session_outranks_the_per_request_trace_id(): assert session_id == "batch-uid-1" - def test_caller_litellm_session_id_wins_over_the_per_request_trace_id(): session_id: Final = _get_session_id_for_spend_log( kwargs={"litellm_session_id": "sess-1", "litellm_trace_id": "trace-abc"}, @@ -258,6 +257,7 @@ def test_batch_session_outranks_a_caller_litellm_session_id(): ) assert session_id == "batch-uid-1" + def test_omit_policy_still_suppresses_batch_sessions(): session_id: Final = _get_session_id_for_spend_log( kwargs={},