Merge pull request #35 from UiPath/fix/samples

fix: gracefully handle subprocess init failure

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -62,7 +62,6 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
             with tracer.start_as_current_span(self.server.name) as root_span:
                 root_span.set_attribute("session_id", self.server.session_id)
                 root_span.set_attribute("command", self.server.command)
-                root_span.set_attribute("env", self.server.env)
                 root_span.set_attribute("args", self.server.args)
                 self.signalr_client = SignalRClient(
                     signalr_url,
@@ -222,7 +221,6 @@ async def _register(self) -> None:
             async with stdio_client(server_params) as (read, write):
                 async with ClientSession(read, write) as session:
                     logger.info("Initializing client session...")
-
                     # Try to initialize with timeout
                     try:
                         await asyncio.wait_for(
@@ -242,14 +240,17 @@ async def _register(self) -> None:
                     # We don't continue with registration here - we'll do it after the context managers
 
         except Exception as e:
-            # Handle any other exceptions that occur
-            logger.error(f"Error during server initialization: {e}")
-            raise UiPathMcpRuntimeError(
-                "SERVER_ERROR",
-                "Server initialization failed",
-                str(e),
-                UiPathErrorCategory.DEPLOYMENT,
-            ) from e
+            # Just log the exception during cleanup - it's expected
+            if "ProcessLookupError" in str(e) or "ExceptionGroup" in str(e):
+                logger.info("Process already terminated during cleanup - this is expected")
+            else:
+                logger.error(f"Error during server initialization: {e}")
+                raise UiPathMcpRuntimeError(
+                    "SERVER_ERROR",
+                    "Server initialization failed",
+                    str(e),
+                    UiPathErrorCategory.DEPLOYMENT,
+                ) from e
 
         # Now that we're outside the context managers, check if initialization succeeded
         if not initialization_successful:

src/uipath_mcp/_cli/_runtime/_session.py

@@ -76,10 +76,8 @@ async def _run_server(self, server_params: StdioServerParameters) -> None:
                 # Process incoming messages from the server
                 try:
                     while True:
-                        print("Waiting for messages...")
+                        logger.info("Waiting for messages...")
                         message = await self.read_stream.receive()
-                        json_str = message.model_dump_json()
-                        print(f"Received message from local server: {json_str}")
                         await self.send_outgoing_message(message)
                 finally:
                     # Cancel the consumer when we exit the loop