Merge pull request #3749 from BerriAI/litellm_raise_404_when_team_not…

…_exist [Fix] - raise 404 from `/team/info` when team does not exist
BerriAI · May 21, 2024 · ad91bff · ad91bff
2 parents c6bb6e3 + f574127
commit ad91bff
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
@@ -7819,6 +7819,12 @@ async def team_info(
         team_info = await prisma_client.get_data(
             team_id=team_id, table_name="team", query_type="find_unique"
         )
+        if team_info is None:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail={"message": f"Team not found, passed team id: {team_id}."},
+            )
+
         ## GET ALL KEYS ##
         keys = await prisma_client.get_data(
             team_id=team_id,