Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/agent/test_replay_cleanup.py

@@ -0,0 +1,95 @@
+"""Tests for agent.replay_cleanup — shared replay-tail sanitizers.
+
+These functions were extracted from gateway/run.py so every resume surface
+(messaging gateway AND TUI/WebUI gateway) strips poisoned tool-call tails the
+same way. Regression coverage for #29086 (WebUI session permanently stuck
+because the dangling tool-call tail was replayed on every resume).
+"""
+
+from agent.replay_cleanup import (
+    is_interrupted_tool_result,
+    strip_dangling_tool_call_tail,
+    strip_interrupted_tool_tails,
+    sanitize_replay_history,
+)
+
+
+def _user(text):
+    return {"role": "user", "content": text}
+
+
+def _assistant_tc(name):
+    return {
+        "role": "assistant",
+        "content": "",
+        "tool_calls": [
+            {"id": "c1", "type": "function", "function": {"name": name, "arguments": "{}"}}
+        ],
+    }
+
+
+def _tool(content):
+    return {"role": "tool", "tool_call_id": "c1", "content": content}
+
+
+
+
+
+
+
+
+
+
+def test_mixed_dangling_batch_uses_truthful_per_call_wording():
+    assistant = {
+        "role": "assistant",
+        "content": "",
+        "tool_calls": [
+            {"id": "read", "function": {"name": "read_file", "arguments": "{}"}},
+            {"id": "write", "function": {"name": "write_file", "arguments": "{}"}},
+        ],
+    }
+    out = strip_dangling_tool_call_tail([_user("hi"), assistant])
+
+    read_result, write_result = out[-2:]
+    assert read_result["effect_disposition"] == "none"
+    assert "no effect" in read_result["content"].lower()
+    assert "unknown" not in read_result["content"].lower()
+    assert write_result["effect_disposition"] == "unknown"
+    assert "unknown" in write_result["content"].lower()
+
+
+
+
+
+
+
+
+
+
+
+
+def test_sanitize_replay_history_combines_both():
+    # interrupted block is removed; a dangling read-only call is safe to erase
+    history = [
+        _user("first"),
+        _assistant_tc("terminal"), _tool("[Command interrupted]"),
+        _user("second"),
+        _assistant_tc("read_file"),  # dangling
+    ]
+    out = sanitize_replay_history(history)
+    assert out[:2] == [
+        _user("first"),
+        _assistant_tc("terminal"),
+    ]
+    assert out[2]["effect_disposition"] == "unknown"
+    assert out[-1] == _user("second")
+
+
+def test_sanitize_replay_history_noop_on_clean_history():
+    history = [_user("hi"), {"role": "assistant", "content": "hello"}]
+    assert sanitize_replay_history(history) == history
+
+
+def test_sanitize_replay_history_empty():
+    assert sanitize_replay_history([]) == []