fix: session output

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_logger.py

@@ -120,7 +120,8 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
                     # If there are multiple sessions, use the sessionId as the key
                     output_result = session_outputs
 
-                return UiPathRuntimeResult(output=output_result)
+                self.result = UiPathRuntimeResult(output=output_result)
+                return self.result
 
         except Exception as e:
             if isinstance(e, UiPathMcpRuntimeError):

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -87,11 +87,6 @@ async def _run_server(self, server_params: StdioServerParameters) -> None:
                             message = await self.read_stream.receive()
                             await self.send_outgoing_message(message)
                     finally:
-                        stderr_temp.seek(0)
-                        self._server_stderr_output = stderr_temp.read()
-                        logger.debug(
-                            f"Session {self.session_id} - Server stderr output:\n{self._server_stderr_output}"
-                        )
                         # Cancel the consumer when we exit the loop
                         consumer_task.cancel()
                         try:
@@ -105,6 +100,11 @@ async def _run_server(self, server_params: StdioServerParameters) -> None:
                     exc_info=True,
                 )
             finally:
+                stderr_temp.seek(0)
+                self._server_stderr_output = stderr_temp.read()
+                logger.info(
+                    f"Session {self.session_id} - Server stderr output:\n{self._server_stderr_output}"
+                )
                 # The context managers will handle cleanup of resources
                 logger.info(f"Server process for session {self.session_id} has ended")
                 self.read_stream = None

src/uipath_mcp/_cli/_runtime/_session.py

@@ -53,6 +53,8 @@ def create_span_for_message(self, message: types.JSONRPCMessage, **context) -> S
                 span.set_attribute("type", "response")
                 span.set_attribute("span_type", "response")
                 span.set_attribute("id", str(root_value.id))
+                if isinstance(root_value.result, dict):
+                    parent_span.set_attribute("output", json.dumps(root_value.result))
                 self._add_response_attributes(span, root_value)
             else:  # JSONRPCError
                 span = self.tracer.start_span("error", context=span_context)
@@ -126,6 +128,9 @@ def _add_request_attributes(
                 if "arguments" in request.params and isinstance(
                     request.params["arguments"], dict
                 ):
+                    span.set_attribute(
+                        "input", json.dumps(request.params["arguments"])
+                    )
                     span.set_attribute(
                         "tool_args", json.dumps(request.params["arguments"])
                     )