phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 18:00:36 +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

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

@ -205,7 +205,11 @@ async def sse_generator(event_gen_coroutine):
except asyncio.CancelledError: except asyncio.CancelledError:
logger.info("Generator cancelled") logger.info("Generator cancelled")
if event_gen: if event_gen:
await event_gen.aclose() # 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: except Exception as e:
logger.exception("Error in sse_generator") logger.exception("Error in sse_generator")
yield create_sse_event( yield create_sse_event(