mirror of
https://github.com/langgenius/dify.git
synced 2026-05-26 13:00:51 -04:00
55f95dbc36
Co-authored-by: Copilot <copilot@github.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
30 lines
1020 B
Python
30 lines
1020 B
Python
"""Server-sent event formatting for run event replay.
|
|
|
|
SSE frames use the run event id as ``id`` and the run event type as ``event`` so
|
|
browsers can resume with ``Last-Event-ID`` while clients can subscribe by event
|
|
name. Payload data is the full public ``RunEvent`` JSON object.
|
|
"""
|
|
|
|
from collections.abc import AsyncIterable, AsyncIterator
|
|
|
|
from dify_agent.protocol.schemas import RUN_EVENT_ADAPTER, RunEvent
|
|
|
|
|
|
def format_sse_event(event: RunEvent) -> str:
|
|
"""Serialize one event as an SSE frame."""
|
|
lines: list[str] = []
|
|
if event.id is not None:
|
|
lines.append(f"id: {event.id}")
|
|
lines.append(f"event: {event.type}")
|
|
lines.append(f"data: {RUN_EVENT_ADAPTER.dump_json(event).decode()}")
|
|
return "\n".join(lines) + "\n\n"
|
|
|
|
|
|
async def sse_event_stream(events: AsyncIterable[RunEvent]) -> AsyncIterator[str]:
|
|
"""Yield formatted SSE frames from public run events."""
|
|
async for event in events:
|
|
yield format_sse_event(event)
|
|
|
|
|
|
__all__ = ["format_sse_event", "sse_event_stream"]
|