fix: close session with invalid version on failure

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -218,7 +218,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:
-                await 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:
-            await 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}"
@@ -324,7 +324,7 @@ async def _register(self) -> None:
                 UiPathErrorCategory.SYSTEM,
             ) from e
 
-    async def dispose_session(self) -> None:
+    async def _dispose_session(self) -> None:
         """Dispose of the session on the server."""
         try:
             response = self._uipath.api_client.request(
@@ -334,36 +334,23 @@ async def dispose_session(self) -> None:
                     jsonrpc="2.0",
                     id=0,
                     result={
-                        "protocolVersion": "invalid-version",
+                        "protocolVersion": "initiliaze-failure",
                         "capabilities": {},
-                        "serverInfo": {"name": "server", "version": "1.0"}
-                    }
+                        "serverInfo": {"name": self.server.name, "version": "1.0"},
+                    },
                 ).model_dump(),
             )
             if response.status_code == 202:
                 logger.info(
-                    f"Sent outgoing cancelled message to UiPath MCP Server: {self.server.session_id}"
-                )
-            else:
-                logger.error(
-                    f"Error sending outgoing cancelled message to to UiPath MCP Server: {response.status_code} - {response.text}"
-                )
-
-            response = self._uipath.api_client.request(
-                "POST",
-                f"mcp_/mcp/{self.server.name}/dispose?sessionId={self.server.session_id}",
-            )
-            if response.status_code == 202:
-                logger.info(
-                    f"Sent session dispose signalr to UiPath MCP Server: {self.server.session_id}"
+                    f"Sent outgoing session dispose message to UiPath MCP Server: {self.server.session_id}"
                 )
             else:
                 logger.error(
-                    f"Error sending session dispose signalr to UiPath MCP Server: {response.status_code} - {response.text}"
+                    f"Error sending session dispose message to UiPath MCP Server: {response.status_code} - {response.text}"
                 )
         except Exception as e:
             logger.error(
-                f"Error sending session dispose signalr to UiPath MCP Server: {e}"
+                f"Error sending session dispose signal to UiPath MCP Server: {e}"
             )
 
     async def cleanup(self) -> None: