scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix row

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-11 19:43:40 -08:00
parent 3ab8944104
commit 13699b27f3
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/proxy/spend_tracking/spend_management_endpoints.py
View file

@ -3161,16 +3161,22 @@ async def _build_ui_spend_logs_response(
if r.get("session_id")
}
enriched: List[dict] = []
for row in data:
row_dict = row.model_dump()
if enrich_session_counts:
if enrich_session_counts:
enriched: List[dict] = []
for row in data:
row_dict = row.model_dump()
sid = row_dict.get("session_id")
row_dict["session_total_count"] = count_map.get(sid, 1) if sid else 1
enriched.append(row_dict)
enriched.append(row_dict)
response_data: list = enriched
else:
# v2 path: return raw Prisma model instances so FastAPI applies its
# own Pydantic-aware serialisation (preserves alias handling, custom
# serializers, etc.).
response_data = data # type: ignore[assignment]
return {
"data": enriched,
"data": response_data,
"total": total_records,
"page": page,
"page_size": page_size,