scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

style(spend): satisfy ruff format on key metadata recovery

Browse source 
Lint CI failed because ruff format splits the Set alias import and collapses a couple of long lines.

Co-authored-by: Mateo Wang <mateo-berri@users.noreply.github.com>
This commit is contained in:
Cursor Agent 2026-09-03 14:34:44 +00:00
parent 6f0f2fcc8d
commit fd520e53ab
No known key found for this signature in database
1 changed files with 4 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/spend_tracking/key_metadata_recovery.py
View file

@ -1,4 +1,5 @@
from collections.abc import Mapping, Sequence, Set as AbstractSet
from collections.abc import Mapping, Sequence
from collections.abc import Set as AbstractSet
from typing import Final, Protocol
from typing_extensions import TypedDict
@ -149,9 +150,7 @@ async def _spend_logs_key_metadata(
"user_email": row.get("user_email"),
}
for row in spend_log_rows
if isinstance(row, dict)
and isinstance(row.get("api_key"), str)
and row["api_key"] in wanted
if isinstance(row, dict) and isinstance(row.get("api_key"), str) and row["api_key"] in wanted
}
@ -162,9 +161,7 @@ async def _emails_for_user_ids(
if not user_ids:
return {}
try:
users: Final = await UserRepository(prisma_client).table.find_many(
where={"user_id": {"in": list(user_ids)}}
)
users: Final = await UserRepository(prisma_client).table.find_many(where={"user_id": {"in": list(user_ids)}})
except Exception as e:
verbose_proxy_logger.warning(
"Failed user_email recovery for %d user ids: %s",