phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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 043d0c0c89
commit 3f045501c2
1 changed files with 3 additions and 6 deletions
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