ai-char/ai-char-bot
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
xds
2026-02-09 09:49:49 +03:00
parent 4461964791
commit 668aadcdc9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/Generation.py
View File

@@ -41,7 +41,7 @@ class Generation(BaseModel):
@computed_field @computed_field
def cost(self) -> float: 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_input = self.input_token_usage * 0.000002
cost_output = self.output_token_usage * 0.00012 cost_output = self.output_token_usage * 0.00012
return round(cost_input + cost_output, 3) return round(cost_input + cost_output, 3)