tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-21 22:24:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

(test) setting team_id

Browse Source
This commit is contained in:
ishaan-jaff
2024-01-18 14:33:13 -08:00
parent f405a827e3
commit 5beef6dbcd
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_key_generate_prisma.py
+3 -1
View File
@@ -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")