From cf27d19dd550d10033f591d5100ac1d379b5688b Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 21 Oct 2024 14:03:32 -0700 Subject: [PATCH] fix sse_generator async --- llama_stack/distribution/server/server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index 6154432b6..84c369364 100644 --- a/llama_stack/distribution/server/server.py +++ b/llama_stack/distribution/server/server.py @@ -203,7 +203,7 @@ async def maybe_await(value): async def sse_generator(event_gen): try: - async for item in event_gen: + async for item in await event_gen: yield create_sse_event(item) await asyncio.sleep(0.01) except asyncio.CancelledError: