lemming-cli 0.3.0__tar.gz → 0.3.1__tar.gz

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lemming-cli
-Version: 0.3.0
+Version: 0.3.1
 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.3.0 → lemming_cli-0.3.1}/pyproject.toml

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

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/models.py

@@ -60,6 +60,8 @@ class Task(pydantic.BaseModel):
     created_at: float = pydantic.Field(default_factory=time.time)
     run_time: float = 0.0
     execution_times: dict[str, float] | None = None
+    active_execution_component: str | None = None
+    active_execution_started_at: float | None = None
     pid: int | None = None
     last_heartbeat: float | None = None
     has_runner_log: bool = False

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/models_test.py

@@ -19,6 +19,8 @@ def test_task_model_defaults():
     assert task.progress == []
     assert task.run_time == 0.0
     assert task.execution_times is None
+    assert task.active_execution_component is None
+    assert task.active_execution_started_at is None
     assert task.superseded_at is None
     assert task.superseded_reason is None
     assert task.created_at > 0

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/runner.py

@@ -306,6 +306,15 @@ def run_with_heartbeat(
         if header and header.startswith("Hook: ")
         else "runner"
     )
+    try:
+        tasks.mark_execution_started(tasks_file, task_id, component)
+    except Exception:
+        logger.warning(
+            "Could not mark %s execution as active for task %s",
+            component,
+            task_id,
+            exc_info=True,
+        )
 
     full_log: list[str] = []
 

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/runner_test.py

@@ -302,6 +302,8 @@ def test_run_with_heartbeat_records_runner_and_hook_times(tmp_path):
     assert task.execution_times is not None
     assert task.execution_times["runner"] > 0
     assert task.execution_times["hook:testing"] > 0
+    assert task.active_execution_component is None
+    assert task.active_execution_started_at is None
 
 
 def test_run_with_heartbeat_interruption_cleanup(tmp_path):

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/tasks/__init__.py

@@ -25,6 +25,9 @@ from .lifecycle import (
 from .lifecycle import (
     is_task_active as is_task_active,
 )
+from .lifecycle import (
+    mark_execution_started as mark_execution_started,
+)
 from .lifecycle import (
     mark_task_in_progress as mark_task_in_progress,
 )

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/tasks/lifecycle.py

@@ -83,6 +83,34 @@ def record_execution_time(
         task.execution_times[component] = (
             task.execution_times.get(component, 0.0) + duration
         )
+        if task.active_execution_component == component:
+            task.active_execution_component = None
+            task.active_execution_started_at = None
+        persistence.save_tasks(tasks_file, data)
+
+
+def mark_execution_started(
+    tasks_file: pathlib.Path,
+    task_id: str,
+    component: str,
+    started_at: float | None = None,
+) -> None:
+    """Records the component whose elapsed time is currently accumulating.
+
+    Args:
+        tasks_file: Path to the tasks YAML file.
+        task_id: ID of the task whose subprocess started.
+        component: Stable component key, such as ``runner``.
+        started_at: Optional wall-clock timestamp, defaulting to now.
+    """
+    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 task is None:
+            return
+
+        task.active_execution_component = component
+        task.active_execution_started_at = started_at or time.time()
         persistence.save_tasks(tasks_file, data)
 
 
@@ -141,6 +169,8 @@ def _mark_task_in_progress(
                 if task.started_at is None:
                     task.started_at = now
                 task.last_started_at = now
+                task.active_execution_component = None
+                task.active_execution_started_at = None
                 if pid is not None:
                     task.pid = pid
                 return True
@@ -259,6 +289,8 @@ def revert_task_to_pending(
         task.requested_status = None
         task.pid = None
         task.last_heartbeat = None
+        task.active_execution_component = None
+        task.active_execution_started_at = None
 
         persistence.save_tasks(tasks_file, data)
         return task
@@ -376,6 +408,8 @@ def cancel_task(tasks_file: pathlib.Path, task_id: str) -> bool:
         task.pid = None
         task.last_heartbeat = None
         task.requested_status = None
+        task.active_execution_component = None
+        task.active_execution_started_at = None
 
         persistence.save_tasks(tasks_file, data)
 
@@ -414,6 +448,8 @@ def reset_task(tasks_file: pathlib.Path, task_id: str) -> models.Task:
         target.progress = []
         target.run_time = 0.0
         target.execution_times = None
+        target.active_execution_component = None
+        target.active_execution_started_at = None
         target.completed_at = None
         target.superseded_at = None
         target.superseded_reason = None

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/tasks/lifecycle_test.py

@@ -61,6 +61,27 @@ def test_record_execution_time_accumulates_by_component(tmp_path):
     }
 
 
+def test_active_execution_is_recorded_then_cleared(tmp_path):
+    tasks_file = tmp_path / "tasks.yml"
+    persistence.save_tasks(
+        tasks_file,
+        models.Roadmap(tasks=[models.Task(id="1", description="test")]),
+    )
+
+    lifecycle.mark_execution_started(
+        tasks_file, "1", "hook:testing", started_at=123.0
+    )
+    active_task = persistence.load_tasks(tasks_file).tasks[0]
+    assert active_task.active_execution_component == "hook:testing"
+    assert active_task.active_execution_started_at == 123.0
+
+    lifecycle.record_execution_time(tasks_file, "1", "hook:testing", 5.0)
+    finished_task = persistence.load_tasks(tasks_file).tasks[0]
+    assert finished_task.execution_times == {"hook:testing": 5.0}
+    assert finished_task.active_execution_component is None
+    assert finished_task.active_execution_started_at is None
+
+
 def test_mark_task_in_progress(tmp_path):
     tasks_file = tmp_path / "tasks.yml"
     data = models.Roadmap(

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/tasks/operations.py

@@ -216,6 +216,8 @@ def supersede_task(
         target.pid = None
         target.last_heartbeat = None
         target.requested_status = None
+        target.active_execution_component = None
+        target.active_execution_started_at = None
 
         persistence.save_tasks(tasks_file, data)
         return target
@@ -320,12 +322,16 @@ def update_task(
                     target.pid = None
                     target.last_heartbeat = None
                     target.requested_status = None
+                    target.active_execution_component = None
+                    target.active_execution_started_at = None
                 elif status == models.TaskStatus.PENDING:
                     target.completed_at = None
                     target.superseded_at = None
                     target.superseded_reason = None
                     target.attempts = 0
                     target.requested_status = None
+                    target.active_execution_component = None
+                    target.active_execution_started_at = None
                 elif target.completed_at is not None:
                     target.completed_at = None
 

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/web/dashboard.spec.js

@@ -163,6 +163,43 @@ test.describe('Dashboard E2E', () => {
     expect(new Set(measurements.map((segment) => segment.color)).size).toBe(3);
   });
 
+  test('advances the active execution segment while a task is running', async ({
+    page,
+  }) => {
+    const activeStartedAt = Date.now() / 1000 - 5;
+    await page.route('**/api/data', async (route) => {
+      await route.fulfill({
+        contentType: 'application/json',
+        json: {
+          cwd: '/mock/cwd',
+          loop_running: true,
+          goal: 'Live timing project',
+          tasks: [
+            {
+              id: 'running',
+              description: 'Task with live execution timing',
+              status: 'in_progress',
+              attempts: 1,
+              progress: [],
+              active_execution_component: 'runner',
+              active_execution_started_at: activeStartedAt,
+            },
+          ],
+          config: { retries: 3, runner: 'agy', time_limit: 60 },
+        },
+      });
+    });
+
+    await gotoAndAwaitMancha(page);
+    await page.getByRole('button', { name: 'Show details' }).click();
+
+    const timingBar = page.getByTestId('execution-breakdown').getByRole('img');
+    const initialLabel = await timingBar.getAttribute('aria-label');
+    await expect
+      .poll(() => timingBar.getAttribute('aria-label'))
+      .not.toBe(initialLabel);
+  });
+
   // --- Environment Overrides Tests ---
 
   test('adds an environment override, stores it in localStorage, and sends it on run', async ({

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/web/dashboard.test.js

@@ -98,7 +98,21 @@ const createInitialState = (overrides = {}) => {
         'bg-purple-600',
         'bg-red-600',
       ];
-      const entries = Object.entries(task.execution_times || {}).filter(
+      const executionTimes = { ...(task.execution_times || {}) };
+      if (
+        task.status === 'in_progress' &&
+        task.active_execution_component &&
+        task.active_execution_started_at
+      ) {
+        const activeDuration = Math.max(
+          0,
+          Date.now() / 1000 - task.active_execution_started_at,
+        );
+        executionTimes[task.active_execution_component] =
+          Number(executionTimes[task.active_execution_component] || 0) +
+          activeDuration;
+      }
+      const entries = Object.entries(executionTimes).filter(
         ([, duration]) => Number.isFinite(Number(duration)) && duration > 0,
       );
       entries.sort(([a], [b]) => {
@@ -894,6 +908,37 @@ describe('Lemming Web Dashboard', () => {
     assert.ok(breakdown.textContent.includes('testing 20s'));
   });
 
+  test('adds live elapsed time to the active execution segment', () => {
+    const originalDateNow = Date.now;
+    let now = 1_000_000;
+    Date.now = () => now;
+    try {
+      const task = {
+        status: 'in_progress',
+        execution_times: { runner: 30 },
+        active_execution_component: 'hook:testing',
+        active_execution_started_at: 990,
+      };
+      const initialState = createInitialState();
+
+      let segments = initialState.getExecutionSegments(task);
+      assert.deepStrictEqual(
+        segments.map(({ key, duration }) => [key, duration]),
+        [
+          ['runner', 30],
+          ['hook:testing', 10],
+        ],
+      );
+
+      now += 5_000;
+      segments = initialState.getExecutionSegments(task);
+      assert.strictEqual(segments[1].duration, 15);
+      assert.ok(Math.abs(segments[1].percent - 100 / 3) < 0.001);
+    } finally {
+      Date.now = originalDateNow;
+    }
+  });
+
   test('hides execution breakdown when timing data is unavailable', async () => {
     const tasks = [
       {

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/src/lemming/web/index.js

@@ -193,7 +193,21 @@
           'bg-purple-600',
           'bg-red-600',
         ];
-        const entries = Object.entries(task.execution_times || {}).filter(
+        const executionTimes = { ...(task.execution_times || {}) };
+        if (
+          task.status === 'in_progress' &&
+          task.active_execution_component &&
+          task.active_execution_started_at
+        ) {
+          const activeDuration = Math.max(
+            0,
+            Date.now() / 1000 - task.active_execution_started_at,
+          );
+          executionTimes[task.active_execution_component] =
+            Number(executionTimes[task.active_execution_component] || 0) +
+            activeDuration;
+        }
+        const entries = Object.entries(executionTimes).filter(
           ([, duration]) => Number.isFinite(Number(duration)) && duration > 0,
         );
         entries.sort(([a], [b]) => {

{lemming_cli-0.3.0 → lemming_cli-0.3.1}/uv.lock

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