From c126207d18eab40c357a08b1ad6e5840d2bbc78e Mon Sep 17 00:00:00 2001 From: mrinal Date: Tue, 6 Oct 2026 21:53:57 +0000 Subject: [PATCH] test(spend_logs): poll guardrail daily metrics instead of reading once Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../spend/test_spend_logs_metadata_fields.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/integration/spend/test_spend_logs_metadata_fields.py b/tests/integration/spend/test_spend_logs_metadata_fields.py index 800887a354e..b34cd39b330 100644 --- a/tests/integration/spend/test_spend_logs_metadata_fields.py +++ b/tests/integration/spend/test_spend_logs_metadata_fields.py @@ -172,9 +172,13 @@ def test_guardrail_usage_is_tracked_when_guardrail_information_is_not_stored(gat lambda values: len(values) == 1, seconds=70, ) - metrics: Final = read_rows( - 'SELECT requests_evaluated, passed_count FROM "LiteLLM_DailyGuardrailMetrics" WHERE guardrail_id=%s', - (string_value(indexed[0]["guardrail_id"]),), + metrics: Final = eventually( + lambda: read_rows( + 'SELECT requests_evaluated, passed_count FROM "LiteLLM_DailyGuardrailMetrics" WHERE guardrail_id=%s', + (string_value(indexed[0]["guardrail_id"]),), + ), + lambda values: values == [{"requests_evaluated": 1, "passed_count": 1}], + seconds=70, ) assert metrics == [{"requests_evaluated": 1, "passed_count": 1}], metrics