[MODIFY] MEM SEE OUTPUT

This commit is contained in:
Mark
2025-12-15 20:50:15 +08:00
parent 7bbef35b7d
commit 9b8db9a001
15 changed files with 863 additions and 144 deletions

View File

@@ -0,0 +1,27 @@
"""
Server-Sent Events (SSE) Utility Functions
Provides shared utilities for formatting and handling SSE messages.
"""
import json
from typing import Dict, Any
def format_sse_message(event_type: str, data: Dict[str, Any]) -> str:
"""
Format a message in Server-Sent Events (SSE) format.
Args:
event_type: Type of event (stage name, result, error, done)
data: Event data dictionary to be serialized as JSON
Returns:
SSE formatted string: "event: <type>\\ndata: <json>\\n\\n"
Example:
>>> format_sse_message("loading", {"message": "Loading..."})
'event: loading\\ndata: {"message": "Loading..."}\\n\\n'
"""
json_data = json.dumps(data, ensure_ascii=False)
return f"event: {event_type}\ndata: {json_data}\n\n"