phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fix linting error

Browse source
This commit is contained in:
Krrish Dholakia 2025-04-19 08:04:56 -07:00
parent a62805f98f
commit dee5182fc8
2 changed files with 14 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

23
litellm/proxy/management_endpoints/common_daily_activity.py
View file

@ -101,6 +101,19 @@ def update_breakdown_metrics(
return breakdown
async def get_api_key_metadata(
prisma_client: PrismaClient,
api_keys: set[str],
) -> Dict[str, Dict[str, Any]]:
"""Update api key metadata for a single record."""
key_records = await prisma_client.db.litellm_verificationtoken.find_many(
where={"token": {"in": list(api_keys)}}
)
return {
k.token: {"key_alias": k.key_alias, "team_id": k.team_id} for k in key_records
}
async def get_daily_activity(
prisma_client: Optional[PrismaClient],
table_name: str,
@ -177,15 +190,7 @@ async def get_daily_activity(
model_metadata: Dict[str, Dict[str, Any]] = {}
provider_metadata: Dict[str, Dict[str, Any]] = {}
if api_keys:
key_records = await prisma_client.db.litellm_verificationtoken.find_many(
where={"token": {"in": list(api_keys)}}
)
api_key_metadata.update(
{
k.token: {"key_alias": k.key_alias, "team_id": k.team_id}
for k in key_records
}
)
api_key_metadata = await get_api_key_metadata(prisma_client, api_keys)
# Process results
results = []

2
litellm/proxy/management_endpoints/internal_user_endpoints.py
View file

@ -36,8 +36,6 @@ from litellm.proxy.management_helpers.utils import management_endpoint_wrapper
from litellm.proxy.utils import handle_exception_on_proxy
from litellm.types.proxy.management_endpoints.common_daily_activity import (
BreakdownMetrics,
DailySpendData,
DailySpendMetadata,
KeyMetadata,
KeyMetricWithMetadata,
LiteLLM_DailyUserSpend,