From 63cd9e14677c98191911e9bf39a208f7c1876ec2 Mon Sep 17 00:00:00 2001 From: spions Date: Fri, 7 Aug 2026 16:17:17 +0300 Subject: [PATCH] style: apply ruff format to hosted_vllm rerank transformation --- litellm/llms/hosted_vllm/rerank/transformation.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/litellm/llms/hosted_vllm/rerank/transformation.py b/litellm/llms/hosted_vllm/rerank/transformation.py index b467bcc5317..55605d82c8d 100644 --- a/litellm/llms/hosted_vllm/rerank/transformation.py +++ b/litellm/llms/hosted_vllm/rerank/transformation.py @@ -177,9 +177,7 @@ class HostedVLLMRerankConfig(BaseRerankConfig): # Check both so either shape is picked up. raw_meta: Final = response.get("meta") or {} usage_data: Final = response.get("usage", {}) - total_tokens: Final = raw_meta.get("billed_units", {}).get("total_tokens") or usage_data.get( - "total_tokens", 0 - ) + total_tokens: Final = raw_meta.get("billed_units", {}).get("total_tokens") or usage_data.get("total_tokens", 0) input_tokens: Final = raw_meta.get("tokens", {}).get("input_tokens") or usage_data.get("total_tokens", 0) _billed_units: Final = RerankBilledUnits(total_tokens=total_tokens) _tokens: Final = RerankTokens(input_tokens=input_tokens)