fix: add request id to span names

Author: cristipufu

pyproject.toml

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

samples/mcp-server/pyproject.toml

@@ -1,10 +1,10 @@
 [project]
 name = "mcp-server"
-version = "0.0.1"
+version = "0.0.3"
 description = "Demo MCP Server"
 authors = [{ name = "John Doe" }]
 dependencies = [
     "mcp>=1.6.0",
-    "uipath-mcp>=0.0.1",
+    "uipath-mcp==0.0.3",
 ]
 requires-python = ">=3.10"

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -63,7 +63,6 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
                 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)
-                root_span.set_attribute("type", self.server.type)
                 self.signalr_client = SignalRClient(
                     signalr_url,
                     headers={

src/uipath_mcp/_cli/_runtime/_session.py

@@ -189,7 +189,7 @@ async def send_outgoing_message(self, message: types.JSONRPCMessage) -> None:
                     json=message.model_dump(),
                 )
 
-                span.set_attribute("http.status_code", response.status_code)
+                span.set_attribute("http_status_code", response.status_code)
 
                 if response.status_code == 202:
                     logger.info(

src/uipath_mcp/_cli/_runtime/_tracer.py

@@ -36,7 +36,7 @@ def create_span_for_message(
         root_value = message.root
 
         if isinstance(root_value, types.JSONRPCRequest):
-            span = self.tracer.start_span(root_value.method)
+            span = self.tracer.start_span(f"{root_value.method} [{root_value.id}]")
             span.set_attribute("type", "request")
             span.set_attribute("id", str(root_value.id))
             span.set_attribute("method", root_value.method)
@@ -49,13 +49,13 @@ def create_span_for_message(
             self._add_notification_attributes(span, root_value)
 
         elif isinstance(root_value, types.JSONRPCResponse):
-            span = self.tracer.start_span("response")
+            span = self.tracer.start_span(f"response [{root_value.id}]")
             span.set_attribute("type", "response")
             span.set_attribute("id", str(root_value.id))
             self._add_response_attributes(span, root_value)
 
         elif isinstance(root_value, types.JSONRPCError):
-            span = self.tracer.start_span("error")
+            span = self.tracer.start_span(f"error [{root_value.id}]")
             span.set_attribute("type", "error")
             span.set_attribute("id", str(root_value.id))
             span.set_attribute("error_code", root_value.error.code)
@@ -83,6 +83,7 @@ def _add_request_attributes(
             if request.method == "tools/call" and isinstance(request.params, dict):
                 if "name" in request.params:
                     span.set_attribute("tool_name", request.params["name"])
+                    span.update_name(f"{request.method}/{request.params['name']} [{request.id}]")
                 if "arguments" in request.params and isinstance(
                     request.params["arguments"], dict
                 ):