Merge pull request #24 from UiPath/fix/samples

fix: tracing

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_session.py

@@ -163,7 +163,7 @@ async def send_message(self, message: types.JSONRPCMessage) -> None:
         await self._message_queue.put(message)
         logger.debug(f"Session {self.session_id} - message queued for processing")
 
-    @traced
+    @traced()
     async def get_incoming_messages(self) -> None:
         """Get new messages from UiPath MCP Server."""
         response = self._uipath.api_client.request(
@@ -178,7 +178,7 @@ async def get_incoming_messages(self) -> None:
                 logger.info(f"Forwarding message to local MCP Server: {message}")
                 await self.send_message(json_message)
 
-    @traced
+    @traced()
     async def send_outgoing_message(self, message: types.JSONRPCMessage) -> None:
         """Send new message to UiPath MCP Server."""
         response = self._uipath.api_client.request(