test(cost): pin model resolution in the provider-prefixed cost test

This commit is contained in:
Alex Schapiro 2026-10-14 21:30:33 +00:00
parent b6f38bc90c
commit 3232409d1c

View file

@ -136,6 +136,7 @@ def test_cost_callback_estimates_cost_with_provider_prefixed_model() -> None:
with (
patch("strix.report.state.get_global_report_state", return_value=report_state),
patch("strix.report.state.resolve_litellm_model", side_effect=lambda model: model),
patch("litellm.completion_cost", side_effect=fake_completion_cost),
):
litellm_cost_callback(kwargs, response)