From 17e6285cca1d7aec69139a4c0ea25edaaf7d1f16 Mon Sep 17 00:00:00 2001 From: Kerry Lu Date: Thu, 15 Oct 2026 09:27:46 -0700 Subject: [PATCH] test(cost): let CostMapEntry hold flex, priority and ultrafast cache and tier prices (#46629) Co-authored-by: devin-ai-integration[bot] <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../cost_calculation/cost_tracking_case.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tests/integration/cost_calculation/cost_tracking_case.py b/tests/integration/cost_calculation/cost_tracking_case.py index 8f28b875d89..fc730524f8e 100644 --- a/tests/integration/cost_calculation/cost_tracking_case.py +++ b/tests/integration/cost_calculation/cost_tracking_case.py @@ -106,6 +106,25 @@ class CostMapEntry(BaseModel): output_cost_per_token_flex: float | None = None input_cost_per_token_priority: float | None = None output_cost_per_token_priority: float | None = None + input_cost_per_token_ultrafast: float | None = None + output_cost_per_token_ultrafast: float | None = None + input_cost_per_token_above_272k_tokens_ultrafast: float | None = None + output_cost_per_token_above_272k_tokens_ultrafast: float | None = None + input_cost_per_token_above_200k_tokens_priority: float | None = None + output_cost_per_token_above_200k_tokens_priority: float | None = None + cache_read_input_token_cost_flex: float | None = None + cache_read_input_token_cost_priority: float | None = None + cache_read_input_token_cost_ultrafast: float | None = None + cache_read_input_token_cost_above_272k_tokens_flex: float | None = None + cache_read_input_token_cost_above_272k_tokens_priority: float | None = None + cache_read_input_token_cost_above_272k_tokens_ultrafast: float | None = None + cache_read_input_token_cost_above_200k_tokens_priority: float | None = None + cache_creation_input_token_cost_flex: float | None = None + cache_creation_input_token_cost_priority: float | None = None + cache_creation_input_token_cost_ultrafast: float | None = None + cache_creation_input_token_cost_above_272k_tokens_flex: float | None = None + cache_creation_input_token_cost_above_272k_tokens_priority: float | None = None + cache_creation_input_token_cost_above_272k_tokens_ultrafast: float | None = None search_context_cost_per_query: SearchContextCostPerQuery | None = None web_search_billing_unit: str | None = None google_maps_grounding_cost_per_query: float | None = None