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

test(test_models.py): combine add + delete model test

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-08 13:29:03 -07:00
parent 3c6bc031de
commit a652a9b3e9
1 changed files with 4 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

21
tests/test_models.py
View file

@ -127,20 +127,6 @@ async def chat_completion(session, key):
raise Exception(f"Request did not return a 200 status code: {status}") raise Exception(f"Request did not return a 200 status code: {status}")
@pytest.mark.asyncio
async def test_add_models():
"""
Add model
Call new model
"""
async with aiohttp.ClientSession() as session:
key_gen = await generate_key(session=session)
key = key_gen["key"]
await add_models(session=session)
await asyncio.sleep(60)
await chat_completion(session=session, key=key)
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_get_models(): async def test_get_models():
""" """
@ -178,14 +164,15 @@ async def delete_model(session, model_id="123"):
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_delete_models(): async def test_add_and_delete_models():
""" """
Get models user has access to Add model
Call new model
""" """
model_id = "12345"
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
key_gen = await generate_key(session=session) key_gen = await generate_key(session=session)
key = key_gen["key"] key = key_gen["key"]
model_id = "1234"
await add_models(session=session, model_id=model_id) await add_models(session=session, model_id=model_id)
await asyncio.sleep(60) await asyncio.sleep(60)
await chat_completion(session=session, key=key) await chat_completion(session=session, key=key)