chore: initialize project repository

2026-09-09 21:55:37 +08:00
commit b1497fd2c6
167 changed files with 17690 additions and 0 deletions
@@ -0,0 +1,88 @@
+from app.api.views.agent_run_sse import encode_events, recovery_events
+
+
+def test_first_connection_emits_unicode_deltas_only_after_commit():
+    content = "基金😀\n结果" * 100
+    events = recovery_events(run_id="r", trace_id="t", status="succeeded", error_code=None,
+                             content=content, tool_calls=None, replay=False, chunk_size=11)
+    assert events[0][0] == "start" and events[-1][0] == "done"
+    assert "".join(payload["content"] for name, payload in events if name == "delta") == content
+    assert all(len(payload["content"]) <= 11 for name, payload in events if name == "delta")
+    ids = [line for event in encode_events("r", events)
+           for line in event.splitlines() if line.startswith("id:")]
+    assert len(set(ids)) == len(ids)
+    pending = recovery_events(run_id="r", trace_id="t", status="running", error_code=None,
+                              content=content, tool_calls=None, replay=False)
+    assert [name for name, _ in pending] == ["start"]
+
+
+def test_empty_result_still_emits_delta():
+    events = recovery_events(run_id="r", trace_id="t", status="succeeded", error_code=None,
+                             content="", tool_calls=None, replay=False)
+    assert [name for name, _ in events] == ["start", "delta", "done"]
+
+
+def test_pending_run_never_emits_result_events() -> None:
+    events = recovery_events(
+        run_id="run",
+        trace_id="trace",
+        status="running",
+        error_code=None,
+        content=None,
+        tool_calls=None,
+    )
+    assert [name for name, _ in events] == ["start"]
+
+
+def test_succeeded_run_emits_tools_and_result_after_terminal_state() -> None:
+    events = recovery_events(
+        run_id="run",
+        trace_id="trace",
+        status="succeeded",
+        error_code=None,
+        content="answer",
+        tool_calls={"calls": []},
+    )
+    assert [name for name, _ in events] == ["start", "tools", "replace", "done"]
+
+
+def test_failed_run_emits_error_without_result() -> None:
+    events = recovery_events(
+        run_id="run",
+        trace_id="trace",
+        status="failed",
+        error_code="MODEL_TIMEOUT",
+        content=None,
+        tool_calls=None,
+    )
+    assert [name for name, _ in events] == ["start", "error"]
+
+
+def test_client_disconnect_does_not_change_recoverable_terminal_events() -> None:
+    events = recovery_events(
+        run_id="run",
+        trace_id="trace",
+        status="succeeded",
+        error_code=None,
+        content="answer",
+        tool_calls=None,
+    )
+    stream = encode_events("run", events)
+    first = next(stream)
+    assert "event: start" in first
+    del stream
+    recovered = list(
+        encode_events(
+            "run",
+            recovery_events(
+                run_id="run",
+                trace_id="trace",
+                status="succeeded",
+                error_code=None,
+                content="answer",
+                tool_calls=None,
+            ),
+        )
+    )
+    assert any("event: replace" in item for item in recovered)
+    assert any("event: done" in item for item in recovered)