mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 17:30:23 +08:00
1d9c77522a
- 删除全部 Python 源码 (src/) 及 Alembic 迁移脚本，归档至 _deprecated_py_src/ - 重构 Rust gateway ai_pipeline: 拆分 planner/finalize 模块，新增 contracts/adaptation 层 - 重组 handlers 模块为 admin/public/proxy/internal/shared 子模块结构 - 新增 executor 模块，引入 Rust 原生数据库迁移 (aether-data/migrations) - 简化 CI/Docker 构建流程，移除 base image 二级构建，统一为单一 app image - 移除 Python 相关基础设施文件 (entrypoint.sh, gunicorn_conf.py, Dockerfile.base)
110 lines
3.9 KiB
Python
110 lines
3.9 KiB
Python
class SSEEventParser:
|
||
"""轻量SSE解析器，按行接收输入并输出完整事件。"""
|
||
|
||
def __init__(self) -> None:
|
||
self._reset_buffer()
|
||
|
||
def _reset_buffer(self) -> None:
|
||
self._buffer: dict[str, str | None | list[str]] = {
|
||
"event": None,
|
||
"data": [],
|
||
"id": None,
|
||
"retry": None,
|
||
}
|
||
|
||
def _finalize_event(self) -> dict[str, str | None] | None:
|
||
data_lines = self._buffer.get("data", [])
|
||
if not isinstance(data_lines, list) or not data_lines:
|
||
self._reset_buffer()
|
||
return None
|
||
|
||
data_str = "\n".join(data_lines)
|
||
event_val = self._buffer.get("event")
|
||
id_val = self._buffer.get("id")
|
||
retry_val = self._buffer.get("retry")
|
||
event: dict[str, str | None] = {
|
||
"event": event_val if isinstance(event_val, str) else None,
|
||
"data": data_str,
|
||
"id": id_val if isinstance(id_val, str) else None,
|
||
"retry": retry_val if isinstance(retry_val, str) else None,
|
||
}
|
||
|
||
self._reset_buffer()
|
||
return event
|
||
|
||
def feed_line(self, line: str | None) -> list[dict[str, str | None]]:
|
||
"""处理单行SSE文本，返回所有完成的事件。"""
|
||
|
||
normalized_line = (line or "").rstrip("\r")
|
||
events: list[dict[str, str | None]] = []
|
||
|
||
# 空行表示事件结束
|
||
if normalized_line == "":
|
||
event = self._finalize_event()
|
||
if event:
|
||
events.append(event)
|
||
return events
|
||
|
||
# 注释行直接忽略
|
||
if normalized_line.startswith(":") and not normalized_line.startswith("::"):
|
||
return events
|
||
|
||
if normalized_line.startswith("event:"):
|
||
_, rest = normalized_line.split(":", 1)
|
||
value = rest.lstrip()
|
||
|
||
if " data:" in value:
|
||
event_part, data_part = value.split("data:", 1)
|
||
event_name = event_part.strip() or None
|
||
data_value = data_part.lstrip()
|
||
self._buffer["event"] = event_name
|
||
if data_value:
|
||
self._append_data_line(data_value)
|
||
event = self._finalize_event()
|
||
if event:
|
||
events.append(event)
|
||
else:
|
||
event_name = value.strip() or None
|
||
self._buffer["event"] = event_name
|
||
return events
|
||
|
||
if normalized_line.startswith("data:"):
|
||
# 如果已经有缓存的 data，先完成上一个事件
|
||
# 这样可以处理没有空行分隔的连续 data 行
|
||
existing_data = self._buffer.get("data", [])
|
||
if existing_data and len(existing_data) > 0:
|
||
event = self._finalize_event()
|
||
if event:
|
||
events.append(event)
|
||
|
||
_, rest = normalized_line.split(":", 1)
|
||
self._append_data_line(rest[1:] if rest.startswith(" ") else rest)
|
||
return events
|
||
|
||
if normalized_line.startswith("id:"):
|
||
_, rest = normalized_line.split(":", 1)
|
||
self._buffer["id"] = rest.strip() or None
|
||
return events
|
||
|
||
if normalized_line.startswith("retry:"):
|
||
_, rest = normalized_line.split(":", 1)
|
||
self._buffer["retry"] = rest.strip() or None
|
||
return events
|
||
|
||
# 未知行：视作数据追加（部分实现会缺少 data: 前缀）
|
||
self._append_data_line(normalized_line)
|
||
return events
|
||
|
||
def flush(self) -> list[dict[str, str | None]]:
|
||
"""在流结束时调用，输出尚未完成的事件。"""
|
||
|
||
event = self._finalize_event()
|
||
return [event] if event else []
|
||
|
||
def _append_data_line(self, value: str) -> None:
|
||
data_lines = self._buffer.get("data")
|
||
if isinstance(data_lines, list):
|
||
data_lines.append(value)
|
||
else:
|
||
self._buffer["data"] = [value]
|