test: use stdio instead of streamable client

Author: radugheo

testcases/ground-to-cloud/test.py

@@ -2,33 +2,24 @@
 import os
 import sys
 
-from dotenv import load_dotenv
-from mcp.client.session import ClientSession
-from mcp.client.streamable_http import streamablehttp_client
+from mcp import ClientSession, StdioServerParameters, stdio_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")
+    # Get the server script path
+    server_script = os.path.join(os.path.dirname(__file__), "server.py")
 
-    # Construct the MCP server URL
-    mcp_url = f"{base_url}/agenthub_/mcp/{folder_key}/{mcp_server_name}"
+    # Create server parameters for stdio connection
+    server_params = StdioServerParameters(
+        command="python",
+        args=[server_script]
+    )
 
     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, _):
+        # Use stdio client to connect to the MCP server
+        async with stdio_client(server_params) as (read_stream, write_stream):
             async with ClientSession(read_stream, write_stream) as session:
                 # Initialize the session
                 try:
@@ -75,8 +66,6 @@ async def call_add_tool():
 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}")