lemming-cli 0.1.3__tar.gz → 0.1.4__tar.gz

{lemming_cli-0.1.3 → lemming_cli-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lemming-cli
-Version: 0.1.3
+Version: 0.1.4
 Summary: An autonomous, iterative task runner for AI coding agents.
 Project-URL: Homepage, https://github.com/owahltinez/lemming
 Project-URL: Repository, https://github.com/owahltinez/lemming

{lemming_cli-0.1.3 → lemming_cli-0.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "lemming-cli"
-version = "0.1.3"
+version = "0.1.4"
 description = "An autonomous, iterative task runner for AI coding agents."
 readme = "README.md"
 authors = [

{lemming_cli-0.1.3 → lemming_cli-0.1.4}/src/lemming/orchestrator.py

@@ -140,6 +140,30 @@ def run_hooks(
             current = next((t for t in data.tasks if t.id == task_id), None)
             if current and current.status != tasks.TaskStatus.IN_PROGRESS:
                 return exit_codes
+
+            # If any hook run was killed or crashed (e.g. the model became
+            # unavailable mid-finalization), the completion is unverified.
+            # Revert the task to pending — keeping its attempt count — so
+            # the whole task is retried instead of being silently marked
+            # completed. Failure finalization proceeds regardless: those
+            # tasks have already exhausted their attempts.
+            failed_hooks = [h for h, code in exit_codes.items() if code != 0]
+            if final_status == tasks.TaskStatus.COMPLETED and failed_hooks:
+                tasks.add_progress(
+                    tasks_file,
+                    task_id,
+                    "Finalization hooks failed "
+                    f"({', '.join(failed_hooks)}); task reverted to "
+                    "pending for retry.",
+                )
+                tasks.revert_task_to_pending(tasks_file, task_id)
+                click.echo(
+                    f"Task {task_id} finalization hooks failed "
+                    f"({', '.join(failed_hooks)}). Reverting to pending "
+                    "for retry."
+                )
+                return exit_codes
+
             tasks.update_task(
                 tasks_file, task_id, status=final_status, force=True
             )

{lemming_cli-0.1.3 → lemming_cli-0.1.4}/src/lemming/orchestrator_test.py

@@ -1,3 +1,4 @@
+import os
 import time
 from unittest import mock
 
@@ -689,6 +690,74 @@ def test_run_hooks_skips_finalization_when_healed(
     assert data.tasks[0].attempts == 0
 
 
+@mock.patch("lemming.prompts.prepare_hook_prompt")
+@mock.patch("lemming.runner.run_with_heartbeat")
+def test_run_hooks_reverts_to_pending_when_hook_killed(
+    mock_run, mock_prepare, hooks_env
+):
+    """A killed/failed finalization hook must not mark the task completed."""
+    mock_prepare.return_value = "Mock Prompt"
+    # Simulate the hook runner being killed (e.g. model became unavailable).
+    mock_run.return_value = (-15, "", "")
+
+    # Put the task in the finalizing state: in progress with a requested
+    # completion, as if the agent had called `lemming complete`.
+    tasks.update_task(
+        hooks_env, "12345678", status=tasks.TaskStatus.IN_PROGRESS
+    )
+    tasks.claim_task(hooks_env, "12345678", pid=os.getpid())
+    tasks.update_task(hooks_env, "12345678", status=tasks.TaskStatus.COMPLETED)
+
+    run_hooks(
+        hooks_env,
+        "12345678",
+        "agy",
+        yolo=True,
+        runner_args=(),
+        no_defaults=False,
+        verbose=False,
+        hooks=["roadmap"],
+        final_status=tasks.TaskStatus.COMPLETED,
+    )
+
+    # The task should be retried from scratch, keeping its attempt count.
+    data = tasks.load_tasks(hooks_env)
+    task = data.tasks[0]
+    assert task.status == tasks.TaskStatus.PENDING
+    assert task.requested_status is None
+    assert task.attempts == 1
+    assert any("hook" in p.lower() for p in task.progress)
+
+
+@mock.patch("lemming.prompts.prepare_hook_prompt")
+@mock.patch("lemming.runner.run_with_heartbeat")
+def test_run_hooks_failure_finalization_ignores_hook_errors(
+    mock_run, mock_prepare, hooks_env
+):
+    """Failed hooks must not revert a task that is being marked FAILED."""
+    mock_prepare.return_value = "Mock Prompt"
+    mock_run.return_value = (1, "", "")
+
+    tasks.update_task(
+        hooks_env, "12345678", status=tasks.TaskStatus.IN_PROGRESS
+    )
+
+    run_hooks(
+        hooks_env,
+        "12345678",
+        "agy",
+        yolo=True,
+        runner_args=(),
+        no_defaults=False,
+        verbose=False,
+        hooks=["roadmap"],
+        final_status=tasks.TaskStatus.FAILED,
+    )
+
+    data = tasks.load_tasks(hooks_env)
+    assert data.tasks[0].status == tasks.TaskStatus.FAILED
+
+
 @mock.patch("lemming.prompts.prepare_hook_prompt")
 @mock.patch("lemming.runner.run_with_heartbeat")
 def test_run_hooks_reloads_tasks(mock_run, mock_prepare, hooks_env):

{lemming_cli-0.1.3 → lemming_cli-0.1.4}/src/lemming/tasks/__init__.py

@@ -31,6 +31,9 @@ from .lifecycle import (
 from .lifecycle import (
     reset_task_logs as reset_task_logs,
 )
+from .lifecycle import (
+    revert_task_to_pending as revert_task_to_pending,
+)
 from .lifecycle import (
     update_heartbeat as update_heartbeat,
 )

{lemming_cli-0.1.3 → lemming_cli-0.1.4}/src/lemming/tasks/lifecycle.py

@@ -218,6 +218,37 @@ def finish_task_attempt(
         return task
 
 
+def revert_task_to_pending(
+    tasks_file: pathlib.Path, task_id: str
+) -> models.Task | None:
+    """Reverts an in-progress or finalizing task to pending for a retry.
+
+    Unlike reset_task, this preserves the attempt count and progress so
+    retry accounting still applies.
+
+    Args:
+        tasks_file: Path to the tasks YAML file.
+        task_id: The ID of the task to revert.
+
+    Returns:
+        The updated Task, or None if not found.
+    """
+    with persistence.lock_tasks(tasks_file):
+        data = persistence.load_tasks(tasks_file)
+        task = next((t for t in data.tasks if t.id == task_id), None)
+        if not task:
+            return None
+
+        update_run_time(task)
+        task.status = models.TaskStatus.PENDING
+        task.requested_status = None
+        task.pid = None
+        task.last_heartbeat = None
+
+        persistence.save_tasks(tasks_file, data)
+        return task
+
+
 def update_heartbeat(
     tasks_file: pathlib.Path, task_id: str, pid: int | None = None
 ) -> bool:

{lemming_cli-0.1.3 → lemming_cli-0.1.4}/uv.lock

@@ -222,7 +222,7 @@ wheels = [
 
 [[package]]
 name = "lemming-cli"
-version = "0.1.3"
+version = "0.1.4"
 source = { editable = "." }
 dependencies = [
     { name = "beautifulsoup4" },