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 4cbdcd09620..aa1d5bb4ed6 100644 --- a/tests/unit/proxy/spend_tracking/test_spend_tracking_utils.py +++ b/tests/unit/proxy/spend_tracking/test_spend_tracking_utils.py @@ -6724,6 +6724,10 @@ def _stored_request_body(body: Mapping[str, object]) -> dict[str, str]: ), ({"input": [{"role": "user", "content": "q"}, {"type": "function_call_output", "output": "out"}]}, {}), ({"messages": [{"role": "user", "content": "q"}, {"role": "assistant", "content": "prefill"}]}, {}), + ( + {"messages": [{"role": "user", "content": ["plain part", {"type": "text", "text": "block part"}]}]}, + {"latest_user_message": "plain part\nblock part"}, + ), ( {"messages": [{"role": "user", "content": [{"type": []}, {"type": {}}, {"type": "text", "text": "latest"}]}]}, {"latest_user_message": "latest"},