From f14e14d5461ecfe7dc147beb01677f6889eacc4b Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Tue, 17 Feb 2026 17:17:20 -0800 Subject: [PATCH] fix: session grouping for dict rows from query_raw --- litellm/proxy/spend_tracking/spend_management_endpoints.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/spend_tracking/spend_management_endpoints.py b/litellm/proxy/spend_tracking/spend_management_endpoints.py index 08aaa851691..b4aeda62004 100644 --- a/litellm/proxy/spend_tracking/spend_management_endpoints.py +++ b/litellm/proxy/spend_tracking/spend_management_endpoints.py @@ -3278,7 +3278,11 @@ async def _build_ui_spend_logs_response( count_map: dict[str, int] = {} if enrich_session_counts: session_ids = list( - {row.session_id for row in data if getattr(row, "session_id", None)} + { + (row.get("session_id") if isinstance(row, dict) else getattr(row, "session_id", None)) + for row in data + if (row.get("session_id") if isinstance(row, dict) else getattr(row, "session_id", None)) + } ) if session_ids: # NOTE: This GROUP BY runs on every v1/UI page load. The IN clause