mirror of
https://github.com/usestrix/strix.git
synced 2026-10-15 04:18:10 +00:00
test(cost): pin model resolution in the provider-prefixed cost test
This commit is contained in:
parent
b6f38bc90c
commit
3232409d1c
1 changed files with 1 additions and 0 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue