Merge pull request #43 from UiPath/fix/samples

fix: return output

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -106,7 +106,6 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
                     try:
                         await session_server.cleanup()
                         stderr_output = session_server.get_server_stderr()
-                        logger.info(f"stderr_output: {stderr_output}")
                         if stderr_output:
                             session_outputs[session_id] = stderr_output
                     except Exception as e:
@@ -121,10 +120,9 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
                     # If there are multiple sessions, use the sessionId as the key
                     output_result = session_outputs
 
-                logger.info(f"Output: {output_result}")
-                self.result = UiPathRuntimeResult(output=output_result)
-                logger.info(f"Result: {self.result.model_dump_json()}")
-                return self.result
+                self.context.result = UiPathRuntimeResult(output=output_result)
+
+                return self.context.result
 
         except Exception as e:
             if isinstance(e, UiPathMcpRuntimeError):

src/uipath_mcp/_cli/_runtime/_session.py

@@ -102,9 +102,6 @@ async def _run_server(self, server_params: StdioServerParameters) -> None:
             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/_tracer.py

@@ -53,14 +53,9 @@ def create_span_for_message(self, message: types.JSONRPCMessage, **context) -> S
                 span.set_attribute("type", "response")
                 span.set_attribute("span_type", "MCP 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))
+                #if isinstance(root_value.result, dict):
+                    #parent_span.set_attribute("output", json.dumps(root_value.result))
                 self._add_response_attributes(span, root_value)
-                parent_span.add_event(
-                    "response",
-                    {"output": json.dumps(root_value.result)}
-                )
-                parent_span.end()
             else:  # JSONRPCError
                 span = self.tracer.start_span("error", context=span_context)
                 span.set_attribute("type", "error")