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] },