test(test_models.py): fix delete model test

This commit is contained in:
Krrish Dholakia 2024-04-04 08:45:32 -07:00
parent 674499b41a
commit 4b56f08cbe
3 changed files with 31 additions and 26 deletions

View file

@ -186,5 +186,6 @@ async def test_delete_models():
key_gen = await generate_key(session=session)
key = key_gen["key"]
await add_models(session=session, model_id=model_id)
await asyncio.sleep(60)
await chat_completion(session=session, key=key)
await delete_model(session=session, model_id=model_id)