aria-code 0.116.0__tar.gz → 0.117.0__tar.gz

{aria_code-0.116.0/src/aria_code.egg-info → aria_code-0.117.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aria-code
-Version: 0.116.0
+Version: 0.117.0
 Summary: Open-source AI agent for coding, financial analysis and logistics operations: numbers you can check, client data kept apart, actions a person approves
 License: Apache-2.0
 Project-URL: Homepage, https://github.com/artheras/aria-code
@@ -172,7 +172,7 @@ Numbers you can check, client data kept apart, actions a person approves.</p>
 <p align="center">
   <a href="https://www.npmjs.com/package/@artheras/aria-code"><img src="https://img.shields.io/npm/v/@artheras/aria-code/latest?style=flat-square&logo=npm&label=npm" alt="npm version"></a>
   <!-- Release automation pins PyPI: the default endpoint still selects legacy 4.x. -->
-  <a href="https://pypi.org/project/aria-code/0.116.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.116.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
+  <a href="https://pypi.org/project/aria-code/0.117.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.117.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
   <a href="https://github.com/artheras/aria-code/actions/workflows/ci.yml"><img src="https://img.shields.io/github/actions/workflow/status/artheras/aria-code/ci.yml?branch=main&style=flat-square&logo=githubactions&label=CI" alt="CI status"></a>
   <a href="LICENSE"><img src="https://img.shields.io/badge/license-Apache%202.0-64748b?style=flat-square" alt="Apache License 2.0"></a>
 </p>

{aria_code-0.116.0 → aria_code-0.117.0}/README.md

@@ -15,7 +15,7 @@ Numbers you can check, client data kept apart, actions a person approves.</p>
 <p align="center">
   <a href="https://www.npmjs.com/package/@artheras/aria-code"><img src="https://img.shields.io/npm/v/@artheras/aria-code/latest?style=flat-square&logo=npm&label=npm" alt="npm version"></a>
   <!-- Release automation pins PyPI: the default endpoint still selects legacy 4.x. -->
-  <a href="https://pypi.org/project/aria-code/0.116.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.116.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
+  <a href="https://pypi.org/project/aria-code/0.117.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.117.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
   <a href="https://github.com/artheras/aria-code/actions/workflows/ci.yml"><img src="https://img.shields.io/github/actions/workflow/status/artheras/aria-code/ci.yml?branch=main&style=flat-square&logo=githubactions&label=CI" alt="CI status"></a>
   <a href="LICENSE"><img src="https://img.shields.io/badge/license-Apache%202.0-64748b?style=flat-square" alt="Apache License 2.0"></a>
 </p>

{aria_code-0.116.0 → aria_code-0.117.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aria-code"
-version = "0.116.0"
+version = "0.117.0"
 description = "Open-source AI agent for coding, financial analysis and logistics operations: numbers you can check, client data kept apart, actions a person approves"
 readme = "README.md"
 license = { text = "Apache-2.0" }

{aria_code-0.116.0 → aria_code-0.117.0}/src/aria_code/_version.py

@@ -17,4 +17,4 @@ from __future__ import annotations
 
 __all__ = ["__version__"]
 
-__version__ = "0.116.0"
+__version__ = "0.117.0"

{aria_code-0.116.0 → aria_code-0.117.0}/src/aria_code/runtime/subagent.py

@@ -114,6 +114,7 @@ def restore_tasks() -> int:
     explicit interrupted task rather than silently disappearing.
     """
     restored = 0
+    interrupted = []
     for record in _ledger().restore():
         task_id = str(record.get("task_id") or "").strip()
         if not task_id or task_id in _TASKS:
@@ -141,8 +142,16 @@ def restore_tasks() -> int:
             handoff=dict(record.get("handoff") or {}),
         )
         _TASKS[task_id] = task
-        _persist(task)
+        if status == "interrupted" and record.get("status") == "running":
+            interrupted.append(task.snapshot())
         restored += 1
+    if interrupted:
+        # Completed/queued records have not changed. Persist all interrupted
+        # records together; per-task upserts made restoration quadratic.
+        try:
+            _ledger().upsert_many(interrupted)
+        except Exception:
+            pass  # Persistence failure must not discard restored live state.
     return restored
 
 

{aria_code-0.116.0 → aria_code-0.117.0}/src/aria_code/runtime/task_ledger.py

@@ -69,18 +69,32 @@ class TaskLedger:
                 os.unlink(temp_name)
 
     def upsert(self, record: Mapping[str, Any]) -> None:
-        task_id = str(record.get("task_id") or "").strip()
-        if not task_id:
-            raise ValueError("Task record requires task_id")
-        records = self.load()
-        snapshot = dict(record)
-        snapshot["updated_at"] = time.time()
-        result = str(snapshot.get("result") or "")
-        if len(result) > MAX_RESULT_CHARS:
-            snapshot["result"] = result[:MAX_RESULT_CHARS] + "\n…[truncated in task ledger]"
-            snapshot["result_truncated"] = True
-        records[task_id] = snapshot
-        self.save(records)
+        self.upsert_many((record,))
+
+    def upsert_many(self, records: Iterable[Mapping[str, Any]]) -> None:
+        """Persist a batch with one read and one atomic replacement.
+
+        Validate the entire batch before touching disk. Restoring interrupted
+        tasks at startup must not rewrite the full ledger for every task.
+        """
+        snapshots = {}
+        updated_at = time.time()
+        for record in records:
+            task_id = str(record.get("task_id") or "").strip()
+            if not task_id:
+                raise ValueError("Task record requires task_id")
+            snapshot = dict(record)
+            snapshot["updated_at"] = updated_at
+            result = str(snapshot.get("result") or "")
+            if len(result) > MAX_RESULT_CHARS:
+                snapshot["result"] = result[:MAX_RESULT_CHARS] + "\n…[truncated in task ledger]"
+                snapshot["result_truncated"] = True
+            snapshots[task_id] = snapshot
+        if not snapshots:
+            return
+        stored = self.load()
+        stored.update(snapshots)
+        self.save(stored)
 
     def restore(self) -> Iterable[dict[str, Any]]:
         return self.load().values()

{aria_code-0.116.0 → aria_code-0.117.0/src/aria_code.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aria-code
-Version: 0.116.0
+Version: 0.117.0
 Summary: Open-source AI agent for coding, financial analysis and logistics operations: numbers you can check, client data kept apart, actions a person approves
 License: Apache-2.0
 Project-URL: Homepage, https://github.com/artheras/aria-code
@@ -172,7 +172,7 @@ Numbers you can check, client data kept apart, actions a person approves.</p>
 <p align="center">
   <a href="https://www.npmjs.com/package/@artheras/aria-code"><img src="https://img.shields.io/npm/v/@artheras/aria-code/latest?style=flat-square&logo=npm&label=npm" alt="npm version"></a>
   <!-- Release automation pins PyPI: the default endpoint still selects legacy 4.x. -->
-  <a href="https://pypi.org/project/aria-code/0.116.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.116.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
+  <a href="https://pypi.org/project/aria-code/0.117.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.117.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
   <a href="https://github.com/artheras/aria-code/actions/workflows/ci.yml"><img src="https://img.shields.io/github/actions/workflow/status/artheras/aria-code/ci.yml?branch=main&style=flat-square&logo=githubactions&label=CI" alt="CI status"></a>
   <a href="LICENSE"><img src="https://img.shields.io/badge/license-Apache%202.0-64748b?style=flat-square" alt="Apache License 2.0"></a>
 </p>

{aria_code-0.116.0 → aria_code-0.117.0}/src/aria_code.egg-info/SOURCES.txt

@@ -769,6 +769,7 @@ tests/test_supervisor_routing.py
 tests/test_system_prompts.py
 tests/test_system_tools_output.py
 tests/test_task_orchestrator.py
+tests/test_task_restore_startup.py
 tests/test_team_analysts.py
 tests/test_team_synthesis.py
 tests/test_terminal_mixins.py

{aria_code-0.116.0 → aria_code-0.117.0}/tests/test_subagent.py

@@ -16,7 +16,7 @@ from aria_code.runtime.task_ledger import TaskLedger
 
 
 @pytest.fixture(autouse=True)
-def clear_tasks():
+def clear_tasks(tmp_path):
     """Clear the task registry and runner between tests."""
     # Same module object the names at the top of this file come from:
     # `runtime.subagent` (via src/aria_code) and `aria_code.runtime.subagent`
@@ -28,6 +28,7 @@ def clear_tasks():
     _orig_runner = _sa._RUNNER
     _orig_ledger = _sa._LEDGER
     _sa._RUNNER = None  # ensure no runner is registered during tests
+    _sa._LEDGER = TaskLedger(tmp_path / "tasks.json")
     yield
     _TASKS.clear()
     _sa._RUNNER = _orig_runner

aria_code-0.117.0/tests/test_task_restore_startup.py

@@ -0,0 +1,106 @@
+"""Restoring thousands of tasks must not do thousands of full-file writes."""
+
+from unittest.mock import Mock
+
+import pytest
+
+import aria_code.runtime.subagent as subagent
+from aria_code.runtime.task_ledger import MAX_RESULT_CHARS, TaskLedger
+
+
+@pytest.fixture
+def ledger(tmp_path, monkeypatch):
+    ledger = TaskLedger(tmp_path / "tasks.json")
+    monkeypatch.setattr(subagent, "_LEDGER", ledger)
+    monkeypatch.setattr(subagent, "_TASKS", {})
+    return ledger
+
+
+def track_io(ledger, monkeypatch):
+    load = Mock(wraps=ledger.load)
+    save = Mock(wraps=ledger.save)
+    monkeypatch.setattr(ledger, "load", load)
+    monkeypatch.setattr(ledger, "save", save)
+    return load, save
+
+
+def test_thousands_of_unchanged_tasks_are_read_once_without_writes(ledger, monkeypatch):
+    statuses = ("pending", "done", "failed", "cancelled", "interrupted")
+    records = {
+        str(i): {**subagent.SubagentTask(task_id=str(i), prompt="inspect",
+                                      status=statuses[i % len(statuses)]).snapshot(),
+                 "updated_at": 123.0, "future_metadata": {"preserve": i}}
+        for i in range(2500)
+    }
+    ledger.save(records)
+    original = ledger.path.read_bytes()
+    load, save = track_io(ledger, monkeypatch)
+
+    assert subagent.restore_tasks() == 2500
+    assert len(subagent._TASKS) == 2500
+    assert load.call_count == 1
+    save.assert_not_called()
+    assert ledger.path.read_bytes() == original
+    assert subagent.restore_tasks() == 0
+    save.assert_not_called()
+
+
+def test_running_tasks_are_interrupted_in_one_atomic_batch(ledger, monkeypatch):
+    records = {
+        str(i): subagent.SubagentTask(task_id=str(i), prompt="inspect", status="running",
+                                    result="x" * (MAX_RESULT_CHARS + 1) if i == 0 else "partial",
+                                    handoff={"verification": "unfinished"}).snapshot()
+        for i in range(300)
+    }
+    records["done"] = {"task_id": "done", "status": "done", "prompt": "finished",
+                       "updated_at": 123.0, "future_metadata": "keep"}
+    ledger.save(records)
+    load, save = track_io(ledger, monkeypatch)
+
+    assert subagent.restore_tasks() == 301
+    assert load.call_count == 2  # Initial restoration, then the batch merge.
+    assert save.call_count == 1
+    persisted = TaskLedger(ledger.path).load()
+    for i in range(300):
+        task = subagent._TASKS[str(i)]
+        assert task.status == "interrupted"
+        assert task.async_task is None
+        assert "restarted" in task.error
+        assert persisted[str(i)]["status"] == "interrupted"
+        assert persisted[str(i)]["handoff"] == {"verification": "unfinished"}
+    assert persisted["done"] == records["done"]
+    assert persisted["0"]["result_truncated"] is True
+    assert persisted["0"]["result"].startswith("x" * MAX_RESULT_CHARS)
+
+
+def test_live_tasks_are_not_overwritten_by_saved_running_state(ledger, monkeypatch):
+    ledger.save({"live": {"task_id": "live", "status": "running", "prompt": "saved"}})
+    live = subagent.SubagentTask(task_id="live", prompt="new", status="done", result="new result")
+    subagent._TASKS["live"] = live
+    _, save = track_io(ledger, monkeypatch)
+
+    assert subagent.restore_tasks() == 0
+    assert subagent._TASKS["live"] is live
+    save.assert_not_called()
+
+
+def test_failed_recovery_write_keeps_restored_state(ledger, monkeypatch):
+    ledger.save({"running": {"task_id": "running", "status": "running", "prompt": "inspect"}})
+    monkeypatch.setattr(ledger, "save", Mock(side_effect=OSError("disk full")))
+
+    assert subagent.restore_tasks() == 1
+    assert subagent._TASKS["running"].status == "interrupted"
+    assert "restarted" in subagent._TASKS["running"].error
+
+
+def test_invalid_batch_does_not_partially_write(ledger, monkeypatch):
+    ledger.save({"existing": {"task_id": "existing", "status": "done"}})
+    original = ledger.path.read_bytes()
+    load, save = track_io(ledger, monkeypatch)
+
+    with pytest.raises(ValueError, match="task_id"):
+        ledger.upsert_many([{"task_id": "valid"}, {"task_id": ""}])
+    ledger.upsert_many([])
+    load.assert_not_called()
+    save.assert_not_called()
+    assert ledger.path.read_bytes() == original