From 31cb1269f5d7db69e03e0649bdf633fb1a46ed1d Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 12 Jun 2024 08:42:35 -0700
Subject: [PATCH] fix(proxy/_types.py): support logging k,v pairs to spend logs
 with `spend_logs_metadata` param

---
 litellm/proxy/_types.py | 3 +++
 litellm/proxy/utils.py  | 1 +
 2 files changed, 4 insertions(+)

diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index 72065e5b970..1930a0faa34 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -1489,6 +1489,9 @@ class SpendLogsMetadata(TypedDict):
     user_api_key_team_id: Optional[str]
     user_api_key_user_id: Optional[str]
     user_api_key_team_alias: Optional[str]
+    spend_logs_metadata: Optional[
+        dict
+    ]  # special param to log k,v pairs to spendlogs for a call
 
 
 class SpendLogsPayload(TypedDict):
diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index 93790e76413..54782c0887e 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -2018,6 +2018,7 @@ def get_logging_payload(
         user_api_key_team_id=None,
         user_api_key_user_id=None,
         user_api_key_team_alias=None,
+        spend_logs_metadata=None,
     )
     if isinstance(metadata, dict):
         verbose_proxy_logger.debug(