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

8
tests/test_keys.py
View file

@ -203,10 +203,13 @@ 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 Make sure only models user has access to are returned
""" """
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}" url = f"http://0.0.0.0:4000/key/info?key={get_key}"
headers = { headers = {
"Authorization": f"Bearer {call_key}", "Authorization": f"Bearer {call_key}",
@ -243,6 +246,9 @@ async def test_key_info():
await get_key_info(session=session, get_key=key, call_key="sk-1234") await get_key_info(session=session, get_key=key, call_key="sk-1234")
# as key itself # # as key itself #
await get_key_info(session=session, get_key=key, call_key=key) 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 # # as random key #
key_gen = await generate_key(session=session, i=0) key_gen = await generate_key(session=session, i=0)
random_key = key_gen["key"] random_key = key_gen["key"]