test(prometheus): seed startup snapshot in mocked logger test

This commit is contained in:
Yucheng He 2026-09-07 11:15:26 -07:00
parent b99ce6e1bf
commit 8196a38dc0

View file

@ -84,6 +84,7 @@ async def test_async_post_call_success_hook_includes_client_ip_user_agent():
"litellm.integrations.prometheus.PrometheusLogger.__init__", return_value=None
):
logger = PrometheusLogger()
logger._emit_input_sequence_length_label = False
logger.litellm_proxy_total_requests_metric = MagicMock()
logger.get_labels_for_metric = MagicMock(
return_value=["client_ip", "user_agent"]