fix: runtime error handliing

Author: radugheo

pyproject.toml

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

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -402,7 +402,7 @@ async def _register(self) -> None:
                             # Only proceed if initialization was successful
                             tools_result = await session.list_tools()
                             # logger.info(tools_result)
-                        except asyncio.TimeoutError:
+                        except Exception:
                             logger.error("Initialization timed out")
                             # Capture stderr output here, after the timeout
                             stderr_temp.seek(0)
@@ -418,7 +418,7 @@ async def _register(self) -> None:
         # Now that we're outside the context managers, check if initialization succeeded
         if not initialization_successful:
             await self._on_runtime_abort()
-            error_message = "The server process failed to initialize. Verify environment variables are set correctly."
+            error_message = "The server process failed to initialize."
             if server_stderr_output:
                 error_message += f"\nServer error output:\n{server_stderr_output}"
             raise UiPathMcpRuntimeError(

testcases/ground-to-cloud/test.py

@@ -1,76 +1,62 @@
-import asyncio
-import os
-import sys
-
-from dotenv import load_dotenv
-from mcp.client.session import ClientSession
-from mcp.client.streamable_http import streamablehttp_client
-from retry import retry
-
-
-def get_required_env_var(name: str) -> str:
-    """Get required environment variable or raise an error if not set."""
-    value = os.getenv(name)
-    if not value:
-        raise ValueError(f"Required environment variable {name} is not set")
-    return value
-
-@retry(tries=3, delay=2, backoff=2)
-async def call_add_tool():
-    # Load configuration from environment variables
-    base_url = get_required_env_var("BASE_URL")
-    folder_key = get_required_env_var("UIPATH_FOLDER_KEY")
-    token = get_required_env_var("UIPATH_ACCESS_TOKEN")
-    mcp_server_name = get_required_env_var("MCP_SERVER_NAME")
-    
-    # Construct the MCP server URL
-    mcp_url = f"{base_url}/agenthub_/mcp/{folder_key}/{mcp_server_name}"
-    
-    try:
-        # Use streamable HTTP client to connect to the MCP server
-        async with streamablehttp_client(mcp_url, headers={ 'Authorization': f'Bearer {token}' }) as (read_stream, write_stream, _):
-            async with ClientSession(read_stream, write_stream) as session:
-                # Initialize the session
-                await session.initialize()
-                
-                # List available tools
-                tools_result = await session.list_tools()
-                available_tools = [tool.name for tool in tools_result.tools]
-                expected_available_tools = [
-                    "add", "subtract", "multiply", "divide", "power", "square_root", "nth_root", 
-                    "sin", "cos", "tan", "log10", "natural_log", "log_base", "mean", "median", "standard_deviation", 
-                    "complex_add", "complex_multiply", "convert_temperature", "solve_quadratic", "get_constants" 
-                ]
-
-                print (f"Available tools: {available_tools}")
-
-                if set(available_tools) != set(expected_available_tools):
-                    raise AssertionError(f"Tool sets don't match. Expected: {set(expected_available_tools)}, Got: {set(available_tools)}")
-
-                # Call the add tool directly
-                call_tool_result = await session.call_tool(name="add", arguments={"a": 7, "b": 5})
-
-                expected_result = "12.0"
-                actual_result = call_tool_result.content[0].text if call_tool_result.content else None
-
-                if actual_result != expected_result:
-                    raise AssertionError(f"Expected {expected_result}, got {actual_result}")
-
-                print("Test completed successfully")
-    except Exception as e:
-        print(f"Unexpected error connecting to MCP server: {e}")
-        raise AssertionError(f"Connection error, {e}") from e
-
-async def main():
-    """Main async function to run the test."""
-    try:
-        load_dotenv()
-
-        await call_add_tool()
-    except Exception as e:
-        print(f"Test failed with error: {e}")
-        sys.exit(1)
-
-
-if __name__ == '__main__':
-    asyncio.run(main())
+import asyncio
+import os
+import sys
+
+from mcp import ClientSession, StdioServerParameters, stdio_client
+from retry import retry
+
+
+@retry(tries=3, delay=2, backoff=2)
+async def call_add_tool():
+    server_script = os.path.join(os.path.dirname(__file__), "server.py")
+
+    server_params = StdioServerParameters(
+        command="python",
+        args=[server_script]
+    )
+
+    try:
+        async with stdio_client(server_params) as (read_stream, write_stream):
+            async with ClientSession(read_stream, write_stream) as session:
+                # Initialize the session
+                await session.initialize()
+
+                # List available tools
+                tools_result = await session.list_tools()
+                available_tools = [tool.name for tool in tools_result.tools]
+                expected_available_tools = [
+                    "add", "subtract", "multiply", "divide", "power", "square_root", "nth_root",
+                    "sin", "cos", "tan", "log10", "natural_log", "log_base", "mean", "median", "standard_deviation",
+                    "complex_add", "complex_multiply", "convert_temperature", "solve_quadratic", "get_constants"
+                ]
+
+                print (f"Available tools: {available_tools}")
+
+                if set(available_tools) != set(expected_available_tools):
+                    raise AssertionError(f"Tool sets don't match. Expected: {set(expected_available_tools)}, Got: {set(available_tools)}")
+
+                # Call the add tool directly
+                call_tool_result = await session.call_tool(name="add", arguments={"a": 7, "b": 5})
+
+                expected_result = "12.0"
+                actual_result = call_tool_result.content[0].text if call_tool_result.content else None
+
+                if actual_result != expected_result:
+                    raise AssertionError(f"Expected {expected_result}, got {actual_result}")
+
+                print("Test completed successfully")
+    except Exception as e:
+        print(f"Unexpected error connecting to MCP server: {e}")
+        raise AssertionError(f"Connection error, {e}") from e
+
+async def main():
+    """Main async function to run the test."""
+    try:
+        await call_add_tool()
+    except Exception as e:
+        print(f"Test failed with error: {e}")
+        sys.exit(1)
+
+
+if __name__ == '__main__':
+    asyncio.run(main())