(feat) log request_tags in SpendLogs

This commit is contained in:
ishaan-jaff 2024-02-16 09:52:25 -08:00
parent 097b46d948
commit 23589a398a
4 changed files with 4 additions and 0 deletions

View file

@ -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):

View file

@ -88,4 +88,5 @@ model LiteLLM_SpendLogs {
metadata Json @default("{}")
cache_hit String @default("")
cache_key String @default("")
request_tags Json @default("[]")
}

View file

@ -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")

View file

@ -88,4 +88,5 @@ model LiteLLM_SpendLogs {
metadata Json @default("{}")
cache_hit String @default("")
cache_key String @default("")
request_tags Json @default("[]")
}