refactor: 重构聊天模块支持 SSE 断线重连及用户上下文隔离
This commit is contained in:
@@ -0,0 +1,22 @@
|
||||
from v1.agent.router import _is_target_run_event, _is_terminal_run_event
|
||||
|
||||
|
||||
def test_is_target_run_event_matches_expected_run_id() -> None:
|
||||
event: dict[str, object] = {"type": "STEP_STARTED", "runId": "run_123"}
|
||||
assert _is_target_run_event(event, target_run_id="run_123") is True
|
||||
|
||||
|
||||
def test_is_target_run_event_rejects_other_run_id() -> None:
|
||||
event: dict[str, object] = {"type": "STEP_STARTED", "runId": "run_999"}
|
||||
assert _is_target_run_event(event, target_run_id="run_123") is False
|
||||
|
||||
|
||||
def test_is_target_run_event_rejects_missing_run_id() -> None:
|
||||
event: dict[str, object] = {"type": "STEP_STARTED"}
|
||||
assert _is_target_run_event(event, target_run_id="run_123") is False
|
||||
|
||||
|
||||
def test_is_terminal_run_event_only_accepts_terminal_types() -> None:
|
||||
assert _is_terminal_run_event({"type": "RUN_FINISHED"}) is True
|
||||
assert _is_terminal_run_event({"type": "RUN_ERROR"}) is True
|
||||
assert _is_terminal_run_event({"type": "STEP_STARTED"}) is False
|
||||
Reference in New Issue
Block a user