scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(test_router.py): run as asyncio.run

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-09 12:28:28 -08:00
parent dd6caa8b55
commit a250489405
1 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_router.py
View file

@ -226,12 +226,9 @@ def test_call_one_endpoint():
)
print("\n response", response)
tasks = []
tasks.append(call_azure_completion())
tasks.append(call_bedrock_claude())
tasks.append(call_azure_embedding())
asyncio.gather(**tasks)
asyncio.run(call_azure_completion())
asyncio.run(call_bedrock_claude())
asyncio.run(call_azure_embedding()
os.environ["AZURE_API_BASE"] = old_api_base
os.environ["AZURE_API_KEY"] = old_api_key