fix: handle keep alive errors

Author: cristipufu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-mcp"
-version = "0.0.65"
+version = "0.0.66"
 description = "UiPath MCP SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -380,9 +380,12 @@ async def _keep_alive(self) -> None:
         """
         Heartbeat to keep the runtime available.
         """
-        while True:
+        while not self._cancel_event.is_set():
             try:
                 async def on_keep_alive_response(response: CompletionMessage) -> None:
+                    if response.error:
+                        logger.error(f"Error during keep-alive: {response.error}")
+                        return
                     session_ids = response.result
                     logger.info(f"Active sessions: {session_ids}")
                     # If there are no active sessions and this is a sandbox environment