steerdev 1.0.37__tar.gz → 1.0.39__tar.gz

{steerdev-1.0.37 → steerdev-1.0.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: steerdev
-Version: 1.0.37
+Version: 1.0.39
 Summary: Backend task runner for steerdev.com - orchestrates CLI coding agents with activity reporting
 Project-URL: Homepage, https://github.com/pentoai/steerdev-agent
 Project-URL: Repository, https://github.com/pentoai/steerdev-agent

{steerdev-1.0.37 → steerdev-1.0.39}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "steerdev"
-version = "1.0.37"
+version = "1.0.39"
 description = "Backend task runner for steerdev.com - orchestrates CLI coding agents with activity reporting"
 readme = "README.md"
 authors = [

{steerdev-1.0.37 → steerdev-1.0.39}/src/steerdev_agent/agent_loop.py

@@ -80,6 +80,8 @@ class CommandExecutor:
     _commands_client: CommandsClient | None
     _sessions_client: SessionsClient | None
     _shutdown_event: asyncio.Event
+    _current_task: asyncio.Task[bool] | None
+    _signal_count: int
     _commands_executed: int
     _commands_succeeded: int
     _commands_failed: int
@@ -172,6 +174,7 @@ class CommandExecutor:
             enable_worktrees=self._enable_worktrees,
             executor_config=self._executor_config,
             force_workflow_id=None,
+            shutdown_event=self._shutdown_event,
         )
 
         # Fetch task details
@@ -452,6 +455,8 @@ class AgentLoop(CommandExecutor):
         # State
         self._agent_id: str | None = None
         self._shutdown_event = asyncio.Event()
+        self._current_task: asyncio.Task[bool] | None = None
+        self._signal_count = 0
         self._is_busy = False
         self._consecutive_errors = 0
         self._commands_executed = 0
@@ -577,7 +582,14 @@ class AgentLoop(CommandExecutor):
 
         while not self._shutdown_event.is_set():
             try:
-                executed = await self._poll_once()
+                task = asyncio.create_task(self._poll_once())
+                self._current_task = task
+                try:
+                    executed = await task
+                except asyncio.CancelledError:
+                    break
+                finally:
+                    self._current_task = None
 
                 if executed:
                     self._consecutive_errors = 0
@@ -696,13 +708,27 @@ class AgentLoop(CommandExecutor):
         """Set up SIGINT/SIGTERM handlers for graceful shutdown."""
         loop = asyncio.get_running_loop()
 
-        def _signal_handler() -> None:
-            logger.info("Received shutdown signal")
-            console.print("\n[yellow]Received shutdown signal, finishing current work...[/yellow]")
+        def _sigint_handler() -> None:
+            self._signal_count += 1
+            if self._signal_count == 1:
+                logger.info("Received shutdown signal")
+                console.print(
+                    "\n[yellow]Received shutdown signal, finishing current work...[/yellow]"
+                )
+                console.print("[dim]Press Ctrl-C again to force stop[/dim]")
+                self._shutdown_event.set()
+            else:
+                logger.info("Force stop requested")
+                console.print("\n[red]Force stopping...[/red]")
+                if self._current_task and not self._current_task.done():
+                    self._current_task.cancel()
+
+        def _sigterm_handler() -> None:
+            logger.info("Received SIGTERM")
             self._shutdown_event.set()
 
-        for sig in (signal.SIGINT, signal.SIGTERM):
-            loop.add_signal_handler(sig, _signal_handler)
+        loop.add_signal_handler(signal.SIGINT, _sigint_handler)
+        loop.add_signal_handler(signal.SIGTERM, _sigterm_handler)
 
     async def _shutdown(self) -> None:
         """Clean up resources and send offline heartbeat."""
@@ -790,6 +816,8 @@ class WorkspaceAgentLoop(CommandExecutor):
         # State
         self._agent_id: str | None = None
         self._shutdown_event = asyncio.Event()
+        self._current_task: asyncio.Task[bool] | None = None
+        self._signal_count = 0
         self._is_busy = False
         self._consecutive_errors = 0
         self._commands_executed = 0
@@ -956,7 +984,14 @@ class WorkspaceAgentLoop(CommandExecutor):
 
         while not self._shutdown_event.is_set():
             try:
-                executed = await self._poll_once()
+                task = asyncio.create_task(self._poll_once())
+                self._current_task = task
+                try:
+                    executed = await task
+                except asyncio.CancelledError:
+                    break
+                finally:
+                    self._current_task = None
 
                 if executed:
                     self._consecutive_errors = 0
@@ -1099,13 +1134,27 @@ class WorkspaceAgentLoop(CommandExecutor):
         """Set up SIGINT/SIGTERM handlers for graceful shutdown."""
         loop = asyncio.get_running_loop()
 
-        def _signal_handler() -> None:
-            logger.info("Received shutdown signal")
-            console.print("\n[yellow]Received shutdown signal, finishing current work...[/yellow]")
+        def _sigint_handler() -> None:
+            self._signal_count += 1
+            if self._signal_count == 1:
+                logger.info("Received shutdown signal")
+                console.print(
+                    "\n[yellow]Received shutdown signal, finishing current work...[/yellow]"
+                )
+                console.print("[dim]Press Ctrl-C again to force stop[/dim]")
+                self._shutdown_event.set()
+            else:
+                logger.info("Force stop requested")
+                console.print("\n[red]Force stopping...[/red]")
+                if self._current_task and not self._current_task.done():
+                    self._current_task.cancel()
+
+        def _sigterm_handler() -> None:
+            logger.info("Received SIGTERM")
             self._shutdown_event.set()
 
-        for sig in (signal.SIGINT, signal.SIGTERM):
-            loop.add_signal_handler(sig, _signal_handler)
+        loop.add_signal_handler(signal.SIGINT, _sigint_handler)
+        loop.add_signal_handler(signal.SIGTERM, _sigterm_handler)
 
     async def _shutdown(self) -> None:
         """Clean up resources and send offline heartbeat."""

{steerdev-1.0.37 → steerdev-1.0.39}/src/steerdev_agent/runner.py

@@ -235,6 +235,7 @@ class Runner:
         force_workflow_id: str | None = None,
         dry_run: bool = False,
         retry_config: RetryConfig | None = None,
+        shutdown_event: asyncio.Event | None = None,
     ) -> None:
         """Initialize the runner.
 
@@ -268,6 +269,7 @@ class Runner:
         self.workflow_id = force_workflow_id
         self.dry_run = dry_run
         self._retry_config = retry_config or RetryConfig()
+        self._shutdown_event = shutdown_event
 
         # Executor configuration
         self._executor_config = executor_config or ExecutorConfig()
@@ -673,6 +675,7 @@ class Runner:
             project_id=self.project_id,
             agent_type=self.agent_type,
             agent_name=self.agent_name,
+            shutdown_event=self._shutdown_event,
         )
 
         try:

{steerdev-1.0.37 → steerdev-1.0.39}/src/steerdev_agent/workflow/executor.py

@@ -50,6 +50,7 @@ class WorkflowExecutor:
         project_id: str | None = None,
         agent_type: str = "claude",
         agent_name: str | None = None,
+        shutdown_event: asyncio.Event | None = None,
     ) -> None:
         """Initialize the workflow executor.
 
@@ -63,6 +64,7 @@ class WorkflowExecutor:
             project_id: Project ID for session tracking.
             agent_type: Agent type for session tracking.
             agent_name: Agent name for session tracking (resolves to agentId).
+            shutdown_event: Optional event to signal graceful shutdown between phases.
         """
         self.working_directory = Path(working_directory)
         self._api_key = api_key
@@ -73,6 +75,7 @@ class WorkflowExecutor:
         self.project_id = project_id
         self.agent_type = agent_type
         self.agent_name = agent_name
+        self._shutdown_event = shutdown_event
 
         self._memory_manager = WorkflowMemoryManager(working_directory)
         self._workflows_client: WorkflowsClient | None = None
@@ -492,6 +495,11 @@ Complete this phase according to its type: {phase.phase_type}
             phases_failed = 0
 
             for phase in phases:
+                if self._shutdown_event and self._shutdown_event.is_set():
+                    console.print("[yellow]Shutdown requested, stopping workflow[/yellow]")
+                    phases_failed += 1
+                    break
+
                 result = await self.execute_phase(
                     workflow_run=workflow_run,
                     phase=phase,