Replay observations and auto-approve pending calls in TUI resume

- Extend TUI resume command to replay tool call observations from
  the prior run and auto-approve any pending approval request
- Report replayed_observations count in the resume result payload

Author: johnteee

teaagent/tui.py

@@ -3,7 +3,7 @@
 import json
 import shlex
 from pathlib import Path
-from typing import Callable, Optional
+from typing import Any, Callable, Optional
 
 from teaagent import __version__
 from teaagent.chat_agent import ChatAgentConfig, run_chat_agent
@@ -294,13 +294,22 @@ def handle_command(self, raw_command: str) -> bool:
             if len(args) != 1:
                 self.output_fn("error: resume requires a run id")
                 return True
+            store = RunStore(self.root)
             try:
-                original_task = RunStore(self.root).task_for_run(args[0])
+                original_task = store.task_for_run(args[0])
+                observations = store.observations_for_run(args[0])
+                pending = store.pending_approval_for_run(args[0])
             except (FileNotFoundError, ValueError) as exc:
                 self.output_fn(f"error: {exc}")
                 return True
+            if pending:
+                self.approved_call_ids.add(pending["call_id"])
+                self.output_fn(f"auto-approved pending call: {pending['call_id']}")
             self.output_fn(f"resume: {args[0]}")
-            self._run_agent_task(original_task)
+            self._run_agent_task(
+                original_task,
+                initial_observations=observations if observations else None,
+            )
             return True
         if action == "use":
             if len(args) != 1:
@@ -355,7 +364,7 @@ def _handle_memory(self, args: list[str]) -> None:
             return
         self.output_fn(f"error: unknown memory command '{action}'")
 
-    def _run_agent_task(self, task: str, *, clarify_first: bool = False) -> None:
+    def _run_agent_task(self, task: str, *, clarify_first: bool = False, initial_observations: Optional[list[dict[str, Any]]] = None) -> None:
         task_spec = None
         if clarify_first:
             clarification = clarify_task(task)
@@ -388,9 +397,13 @@ def _run_agent_task(self, task: str, *, clarify_first: bool = False) -> None:
             ),
             audit=audit,
             task_spec=task_spec,
+            initial_observations=initial_observations,
         )
         store.logger_for_result(result, audit)
-        self._print_json(self._run_result_payload(result, routing=routing.to_dict() if routing else None))
+        payload = self._run_result_payload(result, routing=routing.to_dict() if routing else None)
+        if initial_observations:
+            payload["replayed_observations"] = len(initial_observations)
+        self._print_json(payload)
 
     def _approval_handler(self, request: ApprovalRequest) -> bool:
         self._print_json({"status": "approval_required", "approval": request.to_dict()})

tests/test_tui.py

@@ -250,6 +250,37 @@ def test_tui_resume_replays_persisted_run_task(self) -> None:
             self.assertEqual(resume_payload["final_answer"], "second")
             self.assertIn(f"resume: {run_id}", output)
 
+    def test_tui_resume_replays_observations_for_non_destructive_run(self) -> None:
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp)
+            (root / "note.txt").write_text("hello", encoding="utf-8")
+            output = []
+            adapter = FakeAdapter(
+                [
+                    '{"type":"tool","tool_name":"workspace_read_file","arguments":{"path":"note.txt"},"call_id":"read-1"}',
+                    '{"type":"final","content":"first-done"}',
+                    '{"type":"final","content":"second-done"}',
+                ]
+            )
+            tui = TeaAgentTUI(
+                root=tmp,
+                input_fn=lambda _prompt: "exit",
+                output_fn=output.append,
+                adapter_factory=lambda _provider, _model: adapter,
+            )
+
+            self.assertTrue(tui.handle_command("ask read note"))
+            first_payload = json.loads(output[-1])
+            run_id = first_payload["run_id"]
+
+            self.assertTrue(tui.handle_command(f"resume {run_id}"))
+            resume_payload = json.loads(output[-1])
+
+            self.assertEqual(resume_payload["status"], "completed")
+            self.assertEqual(resume_payload["final_answer"], "second-done")
+            self.assertEqual(resume_payload["replayed_observations"], 1)
+            self.assertIn(f"resume: {run_id}", output)
+
     def test_tui_preflight_command_uses_current_settings(self) -> None:
         with tempfile.TemporaryDirectory() as tmp:
             output = []