fix: trace name and null logger

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_logger.py

@@ -1,7 +1,7 @@
 import os
 
 
-class LoggerAdapter:
+class NullLogger:
     def __init__(self):
         self.devnull = open(os.devnull, "w")
 

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -14,6 +14,7 @@
     UiPathErrorCategory,
     UiPathRuntimeResult,
 )
+from uipath.tracing import wait_for_tracers
 
 from .._utils._config import McpServer
 from ._context import UiPathMcpRuntimeContext
@@ -58,7 +59,7 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
 
             self.cancel_event = asyncio.Event()
 
-            with tracer.start_as_current_span("MCP Server") as root_span:
+            with tracer.start_as_current_span(self.server.name) as root_span:
                 root_span.set_attribute("session_id", self.server.session_id)
                 root_span.set_attribute("command", self.server.command)
                 root_span.set_attribute("args", self.server.args)
@@ -115,6 +116,7 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
             ) from e
 
         finally:
+            wait_for_tracers()
             await self.cleanup()
 
     async def validate(self) -> None:

src/uipath_mcp/_cli/_runtime/_session.py

@@ -6,10 +6,9 @@
 from mcp.client.stdio import stdio_client
 from opentelemetry import trace
 from uipath import UiPath
-from uipath.tracing import wait_for_tracers
 
 from .._utils._config import McpServer
-from ._logger import LoggerAdapter
+from ._logger import NullLogger
 from ._tracer import McpTracer
 
 logger = logging.getLogger(__name__)
@@ -61,7 +60,7 @@ async def _run_server(self, server_params: StdioServerParameters) -> None:
         """Run the server in proper context managers."""
         logger.info(f"Starting server process for session {self.session_id}")
         try:
-            stderr_null = LoggerAdapter(logger)
+            stderr_null = NullLogger()
 
             async with stdio_client(server_params, errlog=stderr_null) as (
                 read,
@@ -171,19 +170,17 @@ async def get_incoming_messages(self) -> None:
                 logger.info(f"Incoming message from UiPath MCP Server: {message}")
                 json_message = types.JSONRPCMessage.model_validate(message)
                 with self._mcp_tracer.create_span_for_message(
-                        json_message,
-                        session_id=self.session_id,
-                        server_name=self.server_config.name
-                    ) as _:
+                    json_message,
+                    session_id=self.session_id,
+                    server_name=self.server_config.name,
+                ) as _:
                     logger.info(f"Forwarding message to local MCP Server: {message}")
                     await self.send_message(json_message)
 
     async def send_outgoing_message(self, message: types.JSONRPCMessage) -> None:
         """Send new message to UiPath MCP Server."""
         with self._mcp_tracer.create_span_for_message(
-            message,
-            session_id=self.session_id,
-            server_name=self.server_config.name
+            message, session_id=self.session_id, server_name=self.server_config.name
         ) as span:
             try:
                 response = self._uipath.api_client.request(
@@ -195,9 +192,13 @@ async def send_outgoing_message(self, message: types.JSONRPCMessage) -> None:
                 span.set_attribute("http.status_code", response.status_code)
 
                 if response.status_code == 202:
-                    logger.info(f"Outgoing message sent to UiPath MCP Server: {message}")
+                    logger.info(
+                        f"Outgoing message sent to UiPath MCP Server: {message}"
+                    )
                 else:
-                    self._mcp_tracer.record_http_error(span, response.status_code, response.text)
+                    self._mcp_tracer.record_http_error(
+                        span, response.status_code, response.text
+                    )
             except Exception as e:
                 self._mcp_tracer.record_exception(span, e)
                 raise
@@ -229,5 +230,4 @@ async def cleanup(self) -> None:
         self.read_stream = None
         self.write_stream = None
         self.mcp_session = None
-        wait_for_tracers()
         logger.info(f"Cleanup completed for session {self.session_id}")