Merge pull request #142 from UiPath/fix/env-vars

fix: pass env vars to coded mcp servers

Author: radu-mocanu

pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "uipath-mcp"
-version = "0.0.101"
+version = "0.0.102"
 description = "UiPath MCP SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 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

@@ -365,7 +365,7 @@ async def _register(self) -> None:
         initialization_successful = False
         tools_result: Optional[ListToolsResult] = None
         server_stderr_output = ""
-        env_vars = dict(server.env)
+        env_vars = server.env
 
         # if server is Coded, include environment variables
         if self.server_type is UiPathServerType.Coded:

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")