diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 3f8f1944e..372b953e0 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -437,6 +437,7 @@ class LiteLLM_SpendLogs(LiteLLMBase): metadata: Optional[dict] = {} cache_hit: Optional[str] = "False" cache_key: Optional[str] = None + request_tags: Optional[Json] = None class LiteLLM_SpendLogs_ResponseObject(LiteLLMBase): diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma index a047951dc..5a57b8808 100644 --- a/litellm/proxy/schema.prisma +++ b/litellm/proxy/schema.prisma @@ -88,4 +88,5 @@ model LiteLLM_SpendLogs { metadata Json @default("{}") cache_hit String @default("") cache_key String @default("") + request_tags Json @default("[]") } \ No newline at end of file diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index b14727d1a..616f99f40 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -1270,6 +1270,7 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time): "total_tokens": usage.get("total_tokens", 0), "prompt_tokens": usage.get("prompt_tokens", 0), "completion_tokens": usage.get("completion_tokens", 0), + "request_tags": metadata.get("tags", []), } verbose_proxy_logger.debug(f"SpendTable: created payload - payload: {payload}\n\n") diff --git a/schema.prisma b/schema.prisma index a047951dc..5a57b8808 100644 --- a/schema.prisma +++ b/schema.prisma @@ -88,4 +88,5 @@ model LiteLLM_SpendLogs { metadata Json @default("{}") cache_hit String @default("") cache_key String @default("") + request_tags Json @default("[]") } \ No newline at end of file