Refactor assertions for cost calculation

This commit is contained in:
Praveen11558 2026-03-22 22:36:09 +05:30 • committed by GitHub
parent 47a0b8a887
commit 6ecb6033d3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -99,10 +99,8 @@ def _generic_cost_per_character(
assert (
"input_cost_per_character" in model_info
and model_info["input_cost_per_character"] is not None
), (
"model info for model={} does not have 'input_cost_per_character'-pricing\nmodel_info={}".format(
model, model_info
)
), "model info for model={} does not have 'input_cost_per_character'-pricing\nmodel_info={}".format(
model, model_info
)
custom_prompt_cost = model_info["input_cost_per_character"]
@ -122,10 +120,8 @@ def _generic_cost_per_character(
assert (
"output_cost_per_character" in model_info
and model_info["output_cost_per_character"] is not None
), (
"model info for model={} does not have 'output_cost_per_character'-pricing\nmodel_info={}".format(
model, model_info
)
), "model info for model={} does not have 'output_cost_per_character'-pricing\nmodel_info={}".format(
model, model_info
)
custom_completion_cost = model_info["output_cost_per_character"]
completion_cost = completion_characters * custom_completion_cost