From 668aadcdc935aa3953097277bc8687a43e7590d6 Mon Sep 17 00:00:00 2001 From: xds Date: Mon, 9 Feb 2026 09:49:49 +0300 Subject: [PATCH] fix --- models/Generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Generation.py b/models/Generation.py index ea4500b..784d513 100644 --- a/models/Generation.py +++ b/models/Generation.py @@ -41,7 +41,7 @@ class Generation(BaseModel): @computed_field def cost(self) -> float: - if self.status == GenerationStatus.DONE: + if self.status == GenerationStatus.DONE and self.input_token_usage and self.output_token_usage: cost_input = self.input_token_usage * 0.000002 cost_output = self.output_token_usage * 0.00012 return round(cost_input + cost_output, 3)