Merge pull request #10 from UiPath/fix/samples

fix: add more logging

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -185,18 +185,19 @@ async def handle_signalr_open(self) -> None:
             data = response.json()
             session_id = data["sessionId"]
             message = data["message"]
-            logger.info(f"Received message from UiPath MCP: {message}")
-            if session_id not in self.session_servers:
-                # Create and start a new session server
-                session_server = SessionServer(self.server, session_id)
-                self.session_servers[session_id] = session_server
-                await session_server.start(self.signalr_client)
-
-            # Get the session server for this session
-            session_server = self.session_servers[session_id]
-
-            # Forward the message to the session's MCP server
-            await session_server.send_message(message)
+            logger.info(f"Received message from UiPath MCP: {data}")
+            if session_id:
+                if session_id not in self.session_servers:
+                    # Create and start a new session server
+                    session_server = SessionServer(self.server, session_id)
+                    self.session_servers[session_id] = session_server
+                    await session_server.start(self.signalr_client)
+
+                # Get the session server for this session
+                session_server = self.session_servers[session_id]
+
+                # Forward the message to the session's MCP server
+                await session_server.send_message(message)
 
     async def handle_signalr_close(self) -> None:
         """Handle SignalR connection close event."""