Merge pull request #122 from UiPath/feat/cli-new

Feat/cli new

Author: radu-mocanu

src/uipath_mcp/_cli/_runtime/_runtime.py

@@ -14,7 +14,6 @@
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
 from pysignalr.client import CompletionMessage, SignalRClient
-from uipath import UiPath
 from uipath._cli._runtime._contracts import (
     UiPathBaseRuntime,
     UiPathErrorCategory,
@@ -26,6 +25,7 @@
 from ._context import UiPathMcpRuntimeContext, UiPathServerType
 from ._exception import UiPathMcpRuntimeError
 from ._session import SessionServer
+from uipath import UiPath
 
 logger = logging.getLogger(__name__)
 tracer = trace.get_tracer(__name__)
@@ -274,13 +274,18 @@ async def _register(self) -> None:
         """Register the MCP server with UiPath."""
 
         folder_key = os.environ.get("UIPATH_FOLDER_KEY")
-        if not folder_key:
+        folder_path = os.environ.get("UIPATH_FOLDER_PATH")
+        if not folder_key and not folder_path:
             raise UiPathMcpRuntimeError(
                 "REGISTRATION_ERROR",
-                "No UIPATH_FOLDER_KEY environment variable set.",
-                "Please set the UIPATH_FOLDER_KEY environment variable.",
+                "No UIPATH_FOLDER_PATH or UIPATH_FOLDER_KEY environment variable set.",
+                "Please set the UIPATH_FOLDER_PATH or UIPATH_FOLDER_KEY environment variable.",
                 UiPathErrorCategory.USER,
             )
+        if not folder_key:
+            uipath = UiPath()
+            folder_key = uipath.folders.retrieve_key(folder_path=folder_path)
+
         logger.info(f"Folder key: {folder_key}")
 
         initialization_successful = False

src/uipath_mcp/_cli/_templates/mcp.json.template

@@ -0,0 +1,9 @@
+{
+  "servers": {
+    "$server_name": {
+        "transport": "stdio",
+        "command": "python",
+        "args": ["server.py"]
+    }
+  }
+}

src/uipath_mcp/_cli/_templates/pyproject.toml.template

@@ -0,0 +1,9 @@
+[project]
+name = "$project_name"
+version = "0.0.1"
+description = "Description for '$project_name' project"
+authors = [{ name = "John Doe", email = "[email protected]" }]
+dependencies = [
+    "uipath-mcp>=0.0.96",
+]
+requires-python = ">=3.11"