From 86baa9e5fbd0aa28665089062564beb0ce531d2f Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Fri, 5 Dec 2025 16:38:09 -0800 Subject: [PATCH] Adding end user and org to service types --- litellm/types/services.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/litellm/types/services.py b/litellm/types/services.py index 78cb0cb3ceb..2525d9cbaff 100644 --- a/litellm/types/services.py +++ b/litellm/types/services.py @@ -33,6 +33,8 @@ class ServiceTypes(str, enum.Enum): # daily spend update queue - actual transaction events IN_MEMORY_DAILY_SPEND_UPDATE_QUEUE = "in_memory_daily_spend_update_queue" REDIS_DAILY_SPEND_UPDATE_QUEUE = "redis_daily_spend_update_queue" + REDIS_DAILY_END_USER_SPEND_UPDATE_QUEUE = "redis_daily_end_user_spend_update_queue" + REDIS_DAILY_ORG_SPEND_UPDATE_QUEUE = "redis_daily_org_spend_update_queue" REDIS_DAILY_TEAM_SPEND_UPDATE_QUEUE = "redis_daily_team_spend_update_queue" REDIS_DAILY_TAG_SPEND_UPDATE_QUEUE = "redis_daily_tag_spend_update_queue" # spend update queue - current spend of key, user, team @@ -87,6 +89,12 @@ DEFAULT_SERVICE_CONFIGS = { ServiceTypes.REDIS_DAILY_SPEND_UPDATE_QUEUE.value: { "metrics": [ServiceMetrics.GAUGE] }, + ServiceTypes.REDIS_DAILY_END_USER_SPEND_UPDATE_QUEUE.value: { + "metrics": [ServiceMetrics.GAUGE] + }, + ServiceTypes.REDIS_DAILY_ORG_SPEND_UPDATE_QUEUE.value: { + "metrics": [ServiceMetrics.GAUGE] + }, ServiceTypes.IN_MEMORY_SPEND_UPDATE_QUEUE.value: { "metrics": [ServiceMetrics.GAUGE] },