perf(cli): reduce health poll interval for local langgraph dev server

mdrxy · mdrxy · commit 9606d1dfc1a3 · 2026-03-27T20:52:28.000-04:00
diff --git a/libs/cli/deepagents_cli/server.py b/libs/cli/deepagents_cli/server.py
@@ -26,7 +26,8 @@
 
 _DEFAULT_HOST = "127.0.0.1"
 _DEFAULT_PORT = 2024
-_HEALTH_POLL_INTERVAL = 0.3
+_HEALTH_POLL_INTERVAL_LOCAL = 0.1
+_HEALTH_POLL_INTERVAL_REMOTE = 0.3
 _HEALTH_TIMEOUT = 60
 _SHUTDOWN_TIMEOUT = 5
 
@@ -167,6 +168,7 @@ async def wait_for_server_healthy(
     timeout: float = _HEALTH_TIMEOUT,  # noqa: ASYNC109
     process: subprocess.Popen | None = None,
     read_log: Callable[[], str] | None = None,
+    local: bool = False,
 ) -> None:
     """Poll a LangGraph server health endpoint until it responds.
 
@@ -177,38 +179,42 @@ async def wait_for_server_healthy(
             we fail fast instead of waiting for the timeout.
         read_log: Optional callable returning log file contents (for
             error messages on early exit).
+        local: Use a shorter poll interval for local servers.
 
     Raises:
         RuntimeError: If the server doesn't become healthy in time.
     """
     import httpx
 
+    poll_interval = (
+        _HEALTH_POLL_INTERVAL_LOCAL if local else _HEALTH_POLL_INTERVAL_REMOTE
+    )
     health_url = f"{url}/ok"
     deadline = time.monotonic() + timeout
     last_status: int | None = None
     last_exc: Exception | None = None
 
-    while time.monotonic() < deadline:
-        if process and process.poll() is not None:
-            output = read_log() if read_log else ""
-            msg = f"Server process exited with code {process.returncode}"
-            if output:
-                msg += f"\n{output[-3000:]}"
-            raise RuntimeError(msg)
+    async with httpx.AsyncClient() as client:
+        while time.monotonic() < deadline:
+            if process and process.poll() is not None:
+                output = read_log() if read_log else ""
+                msg = f"Server process exited with code {process.returncode}"
+                if output:
+                    msg += f"\n{output[-3000:]}"
+                raise RuntimeError(msg)
 
-        try:
-            async with httpx.AsyncClient() as client:
+            try:
                 resp = await client.get(health_url, timeout=2)
                 if resp.status_code == 200:  # noqa: PLR2004
                     logger.info("Server is healthy at %s", url)
                     return
                 last_status = resp.status_code
                 logger.debug("Health check returned status %d", resp.status_code)
-        except (httpx.TransportError, OSError) as exc:
-            logger.debug("Health check attempt failed: %s", exc)
-            last_exc = exc
+            except (httpx.TransportError, OSError) as exc:
+                logger.debug("Health check attempt failed: %s", exc)
+                last_exc = exc
 
-        await asyncio.sleep(_HEALTH_POLL_INTERVAL)
+            await asyncio.sleep(poll_interval)
 
     msg = f"Server did not become healthy within {timeout}s"
     if last_status is not None:
@@ -404,6 +410,7 @@ async def start(
                 timeout=timeout,
                 process=self._process,
                 read_log=self._read_log_file,
+                local=True,
             )
         except Exception:
             self.stop()
diff --git a/libs/cli/tests/unit_tests/test_server.py b/libs/cli/tests/unit_tests/test_server.py
@@ -145,7 +145,8 @@ async def test_raises_on_timeout(self) -> None:
 
         with (
             patch("httpx.AsyncClient", return_value=mock_client),
-            patch("deepagents_cli.server._HEALTH_POLL_INTERVAL", 0),
+            patch("deepagents_cli.server._HEALTH_POLL_INTERVAL_LOCAL", 0),
+            patch("deepagents_cli.server._HEALTH_POLL_INTERVAL_REMOTE", 0),
             pytest.raises(RuntimeError, match="did not become healthy"),
         ):
             await wait_for_server_healthy("http://localhost:2024", timeout=0.01)
@@ -161,7 +162,8 @@ async def test_timeout_reports_last_status(self) -> None:
 
         with (
             patch("httpx.AsyncClient", return_value=mock_client),
-            patch("deepagents_cli.server._HEALTH_POLL_INTERVAL", 0),
+            patch("deepagents_cli.server._HEALTH_POLL_INTERVAL_LOCAL", 0),
+            patch("deepagents_cli.server._HEALTH_POLL_INTERVAL_REMOTE", 0),
             pytest.raises(RuntimeError, match="last status: 503"),
         ):
             await wait_for_server_healthy("http://localhost:2024", timeout=0.01)
