Merge pull request #45 from UiPath/fix/samples

fix: await dispose

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -217,7 +217,7 @@ async def handle_signalr_open(self) -> None:
                 self.session_servers[self.server.session_id] = session_server
                 await session_server.get_incoming_messages()
             except Exception as e:
-                self.dispose_session()
+                await self.dispose_session()
                 logger.error(f"Error starting session server: {str(e)}")
 
     async def handle_signalr_close(self) -> None:
@@ -275,7 +275,7 @@ async def _register(self) -> None:
 
         # Now that we're outside the context managers, check if initialization succeeded
         if not initialization_successful:
-            self.dispose_session()
+            await self.dispose_session()
             error_message = "The server process failed to initialize. Verify environment variables are set correctly."
             if server_stderr_output:
                 error_message += f"\nServer error output:\n{server_stderr_output}"