Merge pull request #11 from UiPath/fix/samples

fix: proper ws cleanup

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -273,9 +273,13 @@ async def cleanup(self) -> None:
 
         self.session_servers.clear()
 
-        if self.signalr_client:
-            # Close the SignalR connection
-            await self.signalr_client._transport._ws.close()
+        if self.signalr_client and hasattr(self.signalr_client, "_transport"):
+            transport = self.signalr_client._transport
+            if transport and hasattr(transport, "_ws") and transport._ws:
+                try:
+                    await transport._ws.close()
+                except Exception as e:
+                    logger.error(f"Error closing SignalR WebSocket: {str(e)}")
 
         # Add a small delay to allow the server to shut down gracefully
         if sys.platform == "win32":