From 23589a398a0db7024b662ca8f56854d3f6f2ddc9 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 16 Feb 2024 09:52:25 -0800 Subject: [PATCH] (feat) log request_tags in SpendLogs --- litellm/proxy/_types.py | 1 + litellm/proxy/schema.prisma | 1 + litellm/proxy/utils.py | 1 + schema.prisma | 1 + 4 files changed, 4 insertions(+) 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