diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index 88b64ef2e..fc0f93a7c 100644 --- a/llama_stack/distribution/server/server.py +++ b/llama_stack/distribution/server/server.py @@ -180,7 +180,7 @@ async def sse_generator(event_gen_coroutine): except asyncio.CancelledError: logger.info("Generator cancelled") if event_gen: - await event_gen.aclose() + await event_gen.close() except Exception as e: logger.exception("Error in sse_generator") yield create_sse_event(