scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

UI fix

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-11 19:34:23 -08:00
parent a8f2a4932f
commit fa11ca3fb0
2 changed files with 19 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -3146,6 +3146,10 @@ async def _build_ui_spend_logs_response(
{row.session_id for row in data if getattr(row, "session_id", None)}
)
if session_ids:
# NOTE: This GROUP BY runs on every v1/UI page load. The IN clause
# is bounded by page_size (typically 25-50 distinct session IDs).
# If performance degrades at scale, consider short-lived caching or
# folding the count into the main query via a window function.
counts = await prisma_client.db.litellm_spendlogs.group_by(
by=["session_id"],
where={"session_id": {"in": session_ids}},

24
ui/litellm-dashboard/src/components/view_logs/index.tsx
View file

@ -331,6 +331,18 @@ export default function SpendLogsTable({
return acc;
}, {});
// Build a single-pass map of session_id → representative request_id.
// Prefers an LLM row over an MCP row as the representative.
const sessionRepresentativeMap = new Map<string, { requestId: string; isMcp: boolean }>();
for (const log of searchedLogs) {
if (!log.session_id || (log.session_total_count || 1) <= 1) continue;
const isMcp = MCP_CALL_TYPES.includes(log.call_type);
const existing = sessionRepresentativeMap.get(log.session_id);
if (!existing || (existing.isMcp && !isMcp)) {
sessionRepresentativeMap.set(log.session_id, { requestId: log.request_id, isMcp });
}
}
const filteredData =
searchedLogs
.map((log) => {
@ -350,16 +362,10 @@ export default function SpendLogsTable({
},
};
})
// Deduplicate multi-call sessions:
// Prefer showing an LLM row as the root representative when available.
.filter((log, _index, arr) => {
// Deduplicate multi-call sessions using the pre-built map (O(1) per row).
.filter((log) => {
if (!log.session_id || (log.session_total_count || 1) <= 1) return true;
const rowsForSession = arr.filter((l) => l.session_id === log.session_id);
const llmRepresentative = rowsForSession.find(
(l) => !MCP_CALL_TYPES.includes(l.call_type),
);
const representative = llmRepresentative || rowsForSession[0];
return representative.request_id === log.request_id;
return sessionRepresentativeMap.get(log.session_id)?.requestId === log.request_id;
}) || [];
// Add this function to handle manual refresh