more organization
This commit is contained in:
+1
-1
@@ -57,7 +57,7 @@ def estimate_cost(model: str, input_tokens: int, output_tokens: int) -> float:
|
||||
|
||||
for pattern, rates in MODEL_PRICING:
|
||||
if re.search(pattern, model, re.IGNORECASE):
|
||||
input_cost = (input_tokens / 1_000_000) * rates["input_per_mtok"]
|
||||
input_cost = (input_tokens / 1_000_000) * rates["input_per_mtok"]
|
||||
output_cost = (output_tokens / 1_000_000) * rates["output_per_mtok"]
|
||||
return input_cost + output_cost
|
||||
return 0.0
|
||||
|
||||
Reference in New Issue
Block a user