From 5beef6dbcd32271b92d8c12a4e200c90cfc73cde Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 18 Jan 2024 14:33:13 -0800 Subject: [PATCH] (test) setting team_id --- litellm/tests/test_key_generate_prisma.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 5accd03c6..ae51e4e96 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -543,7 +543,8 @@ def test_generate_and_update_key(prisma_client): async def test(): await litellm.proxy.proxy_server.prisma_client.connect() request = NewUserRequest( - metadata={"team": "litellm-team3", "project": "litellm-project3"} + metadata={"team": "litellm-team3", "project": "litellm-project3"}, + team_id="litellm-core-infra@gmail.com", ) key = await new_user(request) print(key) @@ -560,6 +561,7 @@ def test_generate_and_update_key(prisma_client): "team": "litellm-team3", "project": "litellm-project3", } + assert result["info"].team_id == "litellm-core-infra@gmail.com" request = Request(scope={"type": "http"}) request._url = URL(url="/update/key")