From 6ec0efab502b8da0f5d8fa9dfc25124df6026e39 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 25 May 2024 15:42:52 -0700
Subject: [PATCH] fix - test team_alias

---
 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 8bfc3314a8..7182444c45 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -8036,6 +8036,7 @@ async def new_team(
     ## ADD TO TEAM TABLE
     complete_team_data = LiteLLM_TeamTable(
         **data.json(),
+        model_id=_model_id,
     )
 
     # If budget_duration is set, set `budget_reset_at`