Merge pull request #37 from UiPath/fix/samples

fix: catch ProcessLookupError

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -239,8 +239,9 @@ async def _register(self) -> None:
 
                     # We don't continue with registration here - we'll do it after the context managers
 
-        except Exception as e:
+        except BaseException as e:
             # Just log the exception during cleanup - it's expected
+            # except (ProcessLookupError, ExceptionGroup) works with 3.11+
             if "ProcessLookupError" in str(e) or "ExceptionGroup" in str(e):
                 logger.info("Process already terminated during cleanup - this is expected")
             else:
@@ -257,7 +258,7 @@ async def _register(self) -> None:
             raise UiPathMcpRuntimeError(
                 "TIMEOUT_ERROR",
                 "Server initialization timed out",
-                "The server process did not respond in time. Verify environment variables are set correctly.",
+                "The server process failed to initialize. Verify environment variables are set correctly.",
                 UiPathErrorCategory.DEPLOYMENT,
             )