scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

style: apply ruff format to hosted_vllm rerank transformation

Browse source
This commit is contained in:
spions 2026-08-07 16:17:17 +03:00
parent 5ae21a66d1
commit 63cd9e1467
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/hosted_vllm/rerank/transformation.py
View file

@ -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)