From 229702a9833735069ebebcab4196473e15f55cea Mon Sep 17 00:00:00 2001 From: Shroominic Date: Sat, 18 Oct 2025 13:29:37 +0800 Subject: [PATCH] ruff fmt --- routstr/payment/models.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/routstr/payment/models.py b/routstr/payment/models.py index 7ce3c245..44d23c0f 100644 --- a/routstr/payment/models.py +++ b/routstr/payment/models.py @@ -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