From 90509a159a4518b92ac0f1c0beec0feb7688066b Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 18 Jan 2024 13:54:08 -0800
Subject: [PATCH] (fix) write team_id to key table

---
 litellm/proxy/proxy_server.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index e77223861ed..f951376306b 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -1098,6 +1098,7 @@ async def generate_key_helper_fn(
             "config": config_json,
             "spend": spend,
             "user_id": user_id,
+            "team_id": team_id,
             "max_parallel_requests": max_parallel_requests,
             "metadata": metadata_json,
         }