test: fix test

This commit is contained in:
Krrish Dholakia 2024-07-11 20:09:24 -07:00
parent 1a57e49e46
commit ba4b94105b

View file

@ -279,7 +279,7 @@ async def test_aarun_thread_litellm(sync_mode, provider, is_streaming):
AsyncAssistantEventHandler,
)
print(run)
run.until_done()
await run.until_done()
else:
run = await litellm.arun_thread(
custom_llm_provider=provider,