(
model: str,
input_tokens: int,
output_tokens: int,
pricing: dict[str, ModelPricing],
*,
input_cost_multiplier: float = 1.0,
)
| 87 | |
| 88 | |
| 89 | def _calc_cost( |
| 90 | model: str, |
| 91 | input_tokens: int, |
| 92 | output_tokens: int, |
| 93 | pricing: dict[str, ModelPricing], |
| 94 | *, |
| 95 | input_cost_multiplier: float = 1.0, |
| 96 | ) -> float: |
| 97 | mp = _pricing_for_model(model, pricing) |
| 98 | effective_multiplier = max(0.1, min(2.0, input_cost_multiplier)) |
| 99 | return ( |
| 100 | (input_tokens / 1_000_000) * mp.input_price * effective_multiplier |
| 101 | + (output_tokens / 1_000_000) * mp.output_price |
| 102 | ) |
| 103 | |
| 104 | |
| 105 | def _supports_requirements( |
no test coverage detected