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

tests/gateway/test_cached_agent_history_guard.py

@@ -0,0 +1,67 @@
+"""Regression tests for cached gateway transcript selection."""
+
+from gateway.run import _build_gateway_agent_history, _select_cached_agent_history
+
+
+def test_select_cached_history_keeps_expected_replay_cleanup():
+    """A durable read-only tool tail removed by stock replay cleanup stays removed."""
+    live = [
+        {"role": "user", "content": "check status", "_db_persisted": True},
+        {
+            "role": "assistant",
+            "content": "",
+            "tool_calls": [
+                {
+                    "id": "call-1",
+                    "type": "function",
+                    "function": {"name": "web_search", "arguments": "{}"},
+                }
+            ],
+            "_db_persisted": True,
+        },
+        {
+            "role": "tool",
+            "tool_call_id": "call-1",
+            "content": "[Command interrupted]",
+            "_db_persisted": True,
+        },
+    ]
+
+    persisted, observed = _build_gateway_agent_history(live)
+
+    assert observed is None
+    assert len(persisted) < len(live)
+    assert persisted == [{"role": "user", "content": "check status"}]
+    assert _select_cached_agent_history(persisted, live) is persisted
+
+
+def test_select_cached_history_preserves_live_when_a_real_row_is_unpersisted():
+    persisted = [
+        {"role": "user", "content": "hello"},
+        {"role": "assistant", "content": "hi"},
+    ]
+    live = [
+        {"role": "user", "content": "hello", "_db_persisted": True},
+        {"role": "assistant", "content": "hi", "_db_persisted": True},
+        {"role": "user", "content": "not written"},
+    ]
+
+    selected = _select_cached_agent_history(persisted, live)
+
+    assert selected is not persisted
+    assert selected == live
+
+
+def test_select_cached_history_ignores_unmarked_ephemeral_scaffolding():
+    """Internal retry nudges are deliberately unpersisted, not evidence of DB lag."""
+    persisted = [{"role": "user", "content": "ship the fix"}]
+    live = [
+        {"role": "user", "content": "ship the fix", "_db_persisted": True},
+        {
+            "role": "user",
+            "content": "Verify your work before stopping.",
+            "_pre_verify_synthetic": True,
+        },
+    ]
+
+    assert _select_cached_agent_history(persisted, live) is persisted