diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index 83407a25f..ba8bfb7a9 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(