diff --git a/litellm/main.py b/litellm/main.py index 0fad87d6d7..764ee5bb8a 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -368,6 +368,8 @@ async def acompletion( async def _async_streaming(response, model, custom_llm_provider, args): try: print_verbose(f"received response in _async_streaming: {response}") + if asyncio.iscoroutine(response): + response = await response async for line in response: print_verbose(f"line in async streaming: {line}") yield line