fix: use custom stdio client for session

Author: cristipufu

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_session.py

@@ -5,11 +5,11 @@
 
 import mcp.types as types
 from mcp import StdioServerParameters
-from mcp.client.stdio import stdio_client
 from opentelemetry import trace
 from uipath import UiPath
 
 from .._utils._config import McpServer
+from ._stdio_client import stdio_client
 from ._tracer import McpTracer
 
 logger = logging.getLogger(__name__)
@@ -98,7 +98,7 @@ async def _run_server(self, server_params: StdioServerParameters) -> None:
         """Run the local MCP server process."""
         logger.info(f"Starting local MCP Server process for session {self._session_id}")
         self._server_stderr_output = None
-        with tempfile.TemporaryFile(mode='w+b') as stderr_temp:
+        with tempfile.TemporaryFile(mode="w+b") as stderr_temp:
             try:
                 async with stdio_client(server_params, errlog=stderr_temp) as (
                     read,
@@ -129,7 +129,9 @@ async def _run_server(self, server_params: StdioServerParameters) -> None:
                 )
             finally:
                 stderr_temp.seek(0)
-                self._server_stderr_output = stderr_temp.read().decode('utf-8', errors='replace')
+                self._server_stderr_output = stderr_temp.read().decode(
+                    "utf-8", errors="replace"
+                )
                 # The context managers will handle cleanup of resources
 
     def _run_server_callback(self, task):

src/uipath_mcp/_cli/_runtime/_stdio_client.py

@@ -101,15 +101,15 @@ async def stdin_writer():
                 # Then terminate the process with escalating signals
                 process.terminate()
                 try:
-                    with anyio.fail_after(2.0):
+                    with anyio.fail_after(1.0):
                         await process.wait()
                 except TimeoutError:
                     try:
                         if sys.platform == "win32":
                             process.send_signal(signal.CTRL_C_EVENT)
                         else:
                             process.send_signal(signal.SIGINT)
-                        with anyio.fail_after(2.0):
+                        with anyio.fail_after(1.0):
                             await process.wait()
                     except TimeoutError:
                         # Force kill if it doesn't terminate