From aea0e4f2d42a3471e5ef1f8329e2d381e246f579 Mon Sep 17 00:00:00 2001 From: Kerry Lu Date: Sat, 31 Oct 2026 11:03:10 -0700 Subject: [PATCH] test(cost): expect cohere rerank-v4 prompt_tokens from billed_units.total_tokens (#49455) Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/integration/cost_calculation/cases/misc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration/cost_calculation/cases/misc.py b/tests/integration/cost_calculation/cases/misc.py index 7743af61e8c..9153a565555 100644 --- a/tests/integration/cost_calculation/cases/misc.py +++ b/tests/integration/cost_calculation/cases/misc.py @@ -271,7 +271,7 @@ COHERE_RERANK_V4_TOTAL_TOKENS_FALLBACK: Final = CostTrackingTestCase( "meta": {"billed_units": {"total_tokens": 99}}, }, ), - expected=ExactExpected(spend=0.0, input_cost=0.0, output_cost=0.0, prompt_tokens=0, completion_tokens=0), + expected=ExactExpected(spend=0.0, input_cost=0.0, output_cost=0.0, prompt_tokens=99, completion_tokens=0), )