fix: pass env vars to coded mcp servers

Author: radu-mocanu

pyproject.toml

@@ -7,7 +7,7 @@ requires-python = ">=3.11"
 dependencies = [
     "mcp==1.11.0",
     "pysignalr==1.3.0",
-    "uipath>=2.0.78, <2.1.0",
+    "uipath>=2.1.0, <2.2.0",
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -368,11 +368,10 @@ async def _register(self) -> None:
         env_vars = dict(server.env)
 
         # if server is Coded, include environment variables
-        if self.server_type is UiPathServerType.Coded:
-            for name, value in os.environ.items():
-                # config env variables should have precedence over system ones
-                if name not in env_vars:
-                    env_vars[name] = value
+        for name, value in os.environ.items():
+            # config env variables should have precedence over system ones
+            if name not in env_vars:
+                env_vars[name] = value
 
         try:
             # Create a temporary session to get tools

src/uipath_mcp/_cli/cli_run.py

@@ -17,7 +17,7 @@
 
 
 def mcp_run_middleware(
-    entrypoint: Optional[str], input: Optional[str], resume: bool
+    entrypoint: Optional[str], input: Optional[str], resume: bool, **kwargs
 ) -> MiddlewareResult:
     """Middleware to handle MCP server execution"""
 
@@ -37,6 +37,7 @@ async def execute():
             context.entrypoint = entrypoint
             context.input = input
             context.resume = resume
+            context.debug = kwargs.get("debug", False)
             context.logs_min_level = env.get("LOG_LEVEL", "INFO")
             context.job_id = env.get("UIPATH_JOB_KEY")
             context.folder_key = env.get("UIPATH_FOLDER_KEY")