fix REDIS_DAILY_END_USER_SPEND_UPDATE_QUEUE

This commit is contained in:
Ishaan Jaffer 2025-12-05 16:32:52 -08:00
parent 82376d8b76
commit 58f8be60a1

View file

@ -1,10 +1,11 @@
import enum
from litellm._uuid import uuid
from typing import List, Optional
from pydantic import BaseModel, Field
from typing_extensions import TypedDict
from litellm._uuid import uuid
class ServiceMetrics(enum.Enum):
COUNTER = "counter"
@ -37,6 +38,7 @@ class ServiceTypes(str, enum.Enum):
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"
REDIS_DAILY_END_USER_SPEND_UPDATE_QUEUE = "redis_daily_end_user_spend_update_queue"
# spend update queue - current spend of key, user, team
IN_MEMORY_SPEND_UPDATE_QUEUE = "in_memory_spend_update_queue"
REDIS_SPEND_UPDATE_QUEUE = "redis_spend_update_queue"
@ -92,9 +94,6 @@ DEFAULT_SERVICE_CONFIGS = {
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]
},