phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) /key/info

Browse source
This commit is contained in:
ishaan-jaff 2024-01-26 19:26:55 -08:00
parent ec3f497db8
commit 9c4d9c2f58
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/test_keys.py
View file

@ -203,11 +203,14 @@ async def test_key_delete():
)
async def get_key_info(session, get_key, call_key):
async def get_key_info(session, call_key, get_key=None):
"""
Make sure only models user has access to are returned
"""
url = f"http://0.0.0.0:4000/key/info?key={get_key}"
if get_key is None:
url = "http://0.0.0.0:4000/key/info"
else:
url = f"http://0.0.0.0:4000/key/info?key={get_key}"
headers = {
"Authorization": f"Bearer {call_key}",
"Content-Type": "application/json",
@ -243,6 +246,9 @@ async def test_key_info():
await get_key_info(session=session, get_key=key, call_key="sk-1234")
# as key itself #
await get_key_info(session=session, get_key=key, call_key=key)
# as key itself, use the auth param, and no query key needed
await get_key_info(session=session, call_key=key)
# as random key #
key_gen = await generate_key(session=session, i=0)
random_key = key_gen["key"]