agent-cli 0.88.0__tar.gz → 0.89.0__tar.gz

{agent_cli-0.88.0 → agent_cli-0.89.0}/.github/workflows/docs.yml

@@ -45,11 +45,11 @@ jobs:
 
       - name: Setup Pages
         if: github.event_name != 'pull_request'
-        uses: actions/configure-pages@v5
+        uses: actions/configure-pages@v6
 
       - name: Upload artifact
         if: github.event_name != 'pull_request'
-        uses: actions/upload-pages-artifact@v4
+        uses: actions/upload-pages-artifact@v5
         with:
           path: "./site"
 
@@ -63,4 +63,4 @@ jobs:
     steps:
       - name: Deploy to GitHub Pages
         id: deployment
-        uses: actions/deploy-pages@v4
+        uses: actions/deploy-pages@v5

{agent_cli-0.88.0 → agent_cli-0.89.0}/.github/workflows/release-drafter.yml

@@ -9,6 +9,6 @@ jobs:
   update_release_draft:
     runs-on: ubuntu-latest
     steps:
-      - uses: release-drafter/release-drafter@v6
+      - uses: release-drafter/release-drafter@v7
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

{agent_cli-0.88.0 → agent_cli-0.89.0}/.pre-commit-config.yaml

@@ -11,13 +11,13 @@ repos:
       - id: end-of-file-fixer
       - id: mixed-line-ending
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: "v0.15.8"
+    rev: "v0.15.11"
     hooks:
       - id: ruff
         args: ["--fix"]
       - id: ruff-format
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: "v1.19.1"
+    rev: "v1.20.1"
     hooks:
       - id: mypy
         additional_dependencies: ["types-PyYAML"]

{agent_cli-0.88.0 → agent_cli-0.89.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-cli
-Version: 0.88.0
+Version: 0.89.0
 Summary: A suite of AI-powered command-line tools for text correction, audio transcription, and voice assistance.
 Project-URL: Homepage, https://github.com/basnijholt/agent-cli
 Author-email: Bas Nijholt <bas@nijho.lt>

{agent_cli-0.88.0 → agent_cli-0.89.0}/agent_cli/server/whisper/backends/faster_whisper.py

@@ -6,6 +6,7 @@ import asyncio
 import logging
 import tempfile
 from concurrent.futures import ProcessPoolExecutor
+from concurrent.futures.process import BrokenProcessPool
 from dataclasses import dataclass
 from multiprocessing import get_context
 from pathlib import Path
@@ -209,12 +210,27 @@ class FasterWhisperBackend:
         }
 
         loop = asyncio.get_running_loop()
-        result = await loop.run_in_executor(
-            self._executor,
-            _transcribe_in_subprocess,
-            audio,
-            kwargs,
-        )
+        try:
+            result = await loop.run_in_executor(
+                self._executor,
+                _transcribe_in_subprocess,
+                audio,
+                kwargs,
+            )
+        except BrokenProcessPool:
+            logger.warning(
+                "faster-whisper process pool died during transcription; "
+                "reloading model %s and retrying once",
+                self._config.model_name,
+            )
+            await self.unload()
+            await self.load()
+            result = await loop.run_in_executor(
+                self._executor,
+                _transcribe_in_subprocess,
+                audio,
+                kwargs,
+            )
 
         return TranscriptionResult(
             text=result["text"],

agent_cli-0.89.0/tests/test_faster_whisper_backend.py

@@ -0,0 +1,66 @@
+"""Tests for the faster-whisper backend."""
+
+from __future__ import annotations
+
+from concurrent.futures.process import BrokenProcessPool
+from typing import TYPE_CHECKING, cast
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from agent_cli.server.whisper.backends.base import BackendConfig
+from agent_cli.server.whisper.backends.faster_whisper import FasterWhisperBackend
+
+if TYPE_CHECKING:
+    from concurrent.futures import ProcessPoolExecutor
+
+
+@pytest.mark.asyncio
+async def test_faster_whisper_transcribe_recovers_from_broken_process_pool() -> None:
+    """Reload the backend and retry once when the process pool is broken."""
+    config = BackendConfig(model_name="tiny", device="cpu", compute_type="int8")
+    backend = FasterWhisperBackend(config)
+    initial_executor = cast("ProcessPoolExecutor", object())
+    backend._executor = initial_executor
+    backend._device = "cpu"
+
+    recovered_executor = cast("ProcessPoolExecutor", object())
+    fake_result = {
+        "text": "hello world",
+        "language": "en",
+        "language_probability": 0.99,
+        "duration": 1.25,
+        "segments": [],
+    }
+    executors_seen: list[object] = []
+
+    async def mock_run_in_executor(
+        executor: object, _func: object, *_args: object
+    ) -> dict[str, object]:
+        executors_seen.append(executor)
+        if len(executors_seen) == 1:
+            msg = "worker died"
+            raise BrokenProcessPool(msg)
+        return fake_result
+
+    async def fake_unload() -> None:
+        backend._executor = None
+        backend._device = None
+
+    async def fake_load() -> float:
+        backend._executor = recovered_executor
+        backend._device = "cpu"
+        return 0.1
+
+    with (
+        patch("asyncio.get_running_loop") as mock_loop,
+        patch.object(backend, "unload", new=AsyncMock(side_effect=fake_unload)) as unload_mock,
+        patch.object(backend, "load", new=AsyncMock(side_effect=fake_load)) as load_mock,
+    ):
+        mock_loop.return_value.run_in_executor = mock_run_in_executor
+        result = await backend.transcribe(b"fake audio bytes")
+
+    assert result.text == "hello world"
+    unload_mock.assert_awaited_once()
+    load_mock.assert_awaited_once()
+    assert executors_seen == [initial_executor, recovered_executor]