Merge pull request #114 from UiPath/fix/update_3_11

fix: proper error handling

Author: cristipufu

.python-version

@@ -1 +1 @@
-3.10
+3.11

pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "uipath-mcp"
-version = "0.0.92"
+version = "0.0.93"
 description = "UiPath MCP SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
-requires-python = ">=3.10"
+requires-python = ">=3.11"
 dependencies = [
-    "mcp==1.9.0",
+    "mcp==1.10.1",
     "pysignalr==1.3.0",
-    "uipath>=2.0.71",
+    "uipath>=2.0.73",
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -3,10 +3,10 @@
 import os
 import sys
 import tempfile
-import traceback
 import uuid
 from typing import Any, Dict, Optional
 
+from httpx import HTTPStatusError
 from mcp import ClientSession, StdioServerParameters, stdio_client
 from mcp.types import JSONRPCResponse
 from opentelemetry import trace
@@ -329,35 +329,11 @@ async def _register(self) -> None:
                             # We'll handle this after exiting the context managers
                         # We don't continue with registration here - we'll do it after the context managers
 
-        except BaseException as e:
-            # In Python 3.10, ExceptionGroup is in the 'exceptiongroup' module
-            # and asyncio.TaskGroup wraps exceptions in ExceptionGroup
-            if hasattr(e, "__context__") and e.__context__ is not None:
-                logger.error("Sub-exception details:")
+        except* Exception as eg:
+            for e in eg.exceptions:
                 logger.error(
-                    "".join(
-                        traceback.format_exception(
-                            type(e.__context__),
-                            e.__context__,
-                            e.__context__.__traceback__,
-                        )
-                    )
-                )
-            elif hasattr(e, "exceptions"):  # For ExceptionGroup
-                for i, sub_exc in enumerate(e.exceptions):
-                    logger.error(f"Sub-exception {i + 1}:")
-                    logger.error(
-                        "".join(
-                            traceback.format_exception(
-                                type(sub_exc), sub_exc, sub_exc.__traceback__
-                            )
-                        )
-                    )
-            else:
-                # Log the full traceback of the exception itself
-                logger.error("Full traceback:")
-                logger.error(
-                    "".join(traceback.format_exception(type(e), e, e.__traceback__))
+                    f"Unexpected error: {e}",
+                    exc_info=True,
                 )
 
         # Now that we're outside the context managers, check if initialization succeeded
@@ -395,8 +371,6 @@ async def _register(self) -> None:
                 }
                 client_info["tools"].append(tool_info)
 
-            logger.info(client_info)
-
             # Register with UiPath MCP Server
             await self._uipath.api_client.request_async(
                 "POST",
@@ -407,8 +381,10 @@ async def _register(self) -> None:
             logger.info("Registered MCP Server type successfully")
         except Exception as e:
             logger.error(f"Error during registration: {e}")
-            if e.status_code == 400:
-                logger.error(f"Error details: {e.response.text}")
+            if isinstance(e, HTTPStatusError):
+                logger.error(
+                    f"HTTP error details: {e.response.text} status code: {e.response.status_code}"
+                )
 
             raise UiPathMcpRuntimeError(
                 "REGISTRATION_ERROR",

src/uipath_mcp/_cli/_runtime/_session.py

@@ -143,15 +143,16 @@ async def _run_server(self, server_params: StdioServerParameters) -> None:
                         # Cancel the consumer when we exit the loop
                         consumer_task.cancel()
                         try:
-                            await consumer_task
-                        except asyncio.CancelledError:
+                            await asyncio.wait_for(consumer_task, timeout=2.0)
+                        except (asyncio.CancelledError, asyncio.TimeoutError):
                             pass
 
-            except Exception as e:
-                logger.error(
-                    f"Error in server process for session {self._session_id}: {e}",
-                    exc_info=True,
-                )
+            except* Exception as eg:
+                for e in eg.exceptions:
+                    logger.error(
+                        f"Unexpected error for session {self._session_id}: {e}",
+                        exc_info=True,
+                    )
             finally:
                 stderr_temp.seek(0)
                 self._server_stderr_output = stderr_temp.read().decode(