scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-26 01:12:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 99

test(integration): only accept omitted breakdown components when the case expects zero, fix xai output cost

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-09-19 23:12:43 +00:00
parent c7113f043a
commit 789f0c61bf
2 changed files with 8 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/integration/cost_calculation/cost_tracking_cases.json
View file

@ -28262,7 +28262,7 @@
"expected": {
"spend": 0.0135964,
"input_cost": 0.002484,
"output_cost": 0.0011224,
"output_cost": 0.0011124,
"tool_usage_cost": 0.01,
"prompt_tokens": 1840,
"completion_tokens": 412

15
tests/integration/cost_calculation/test_cost_tracking.py
View file

@ -148,16 +148,15 @@ def _assert_breakdown(
):
if expected_component is None:
continue
actual_value: Final = actual_component or 0.0
assert approx_equal(actual_value, expected_component), (
f"{case.name}: {field} {actual_component} != expected {expected_component}"
)
omitted_component_allowed: Final = expected_component == 0.0
assert (actual_component is None and omitted_component_allowed) or (
actual_component is not None and approx_equal(actual_component, expected_component)
), f"{case.name}: {field} {actual_component} != expected {expected_component}"
if expected.cost_header and case.response.content_type == "application/json":
header: Final = response.headers.get(header_name)
header_value: Final = float(header) if header is not None else 0.0
assert approx_equal(header_value, expected_component), (
f"{case.name}: {header_name} {header} != expected {expected_component}"
)
assert (header is None and omitted_component_allowed) or (
header is not None and approx_equal(float(header), expected_component)
), f"{case.name}: {header_name} {header} != expected {expected_component}"
if expected.cost_header and case.response.content_type == "application/json" and any(
component is not None
for component in (