phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 01:48:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(server.py): check attr sse_generator returned object

Browse source
This commit is contained in:
r-bit-rry 2025-11-26 10:55:28 +02:00
parent 91f1b352b4
commit 57f8f6d3af
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/llama_stack/core/server/server.py
View file

@ -205,7 +205,11 @@ async def sse_generator(event_gen_coroutine):
except asyncio.CancelledError:
logger.info("Generator cancelled")
if event_gen:
# Some generators (like OpenAI's AsyncStream) only have close()
if hasattr(event_gen, "aclose"):
await event_gen.aclose()
elif hasattr(event_gen, "close"):
await event_gen.close()
except Exception as e:
logger.exception("Error in sse_generator")
yield create_sse_event(