laantungir/routstr-core
1
0
Fork 0
mirror of https://github.com/Routstr/routstr-core.git synced 2026-08-09 02:54:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

ruff fmt

Browse Source
This commit is contained in:
Shroominic
2025-10-18 13:29:37 +08:00
parent eda1f8dadf
commit 229702a983
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
routstr/payment/models.py
+9 -2
View File
@@ -199,12 +199,19 @@ def _model_to_row_payload(model: Model) -> dict[str, str | int | None]:
# Create adjusted pricing with fees applied
adjusted_pricing = model.pricing.dict()
for key in ['prompt', 'completion', 'request', 'image', 'web_search', 'internal_reasoning']:
for key in [
"prompt",
"completion",
"request",
"image",
"web_search",
"internal_reasoning",
]:
if key in adjusted_pricing:
adjusted_pricing[key] = adjusted_pricing[key] * total_fee_multiplier
# Also adjust max costs if present
for key in ['max_prompt_cost', 'max_completion_cost', 'max_cost']:
for key in ["max_prompt_cost", "max_completion_cost", "max_cost"]:
if key in adjusted_pricing:
adjusted_pricing[key] = adjusted_pricing[key] * total_fee_multiplier