feat: add fallback for folder_key in mcp service (#1484)

Author: dianagrecu-uipath

packages/uipath-platform/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-platform"
-version = "0.1.12"
+version = "0.1.13"
 description = "HTTP client library for programmatic access to UiPath Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

packages/uipath-platform/src/uipath/platform/common/_bindings.py

@@ -239,7 +239,7 @@ def process_args(args, kwargs) -> dict[str, Any]:
                         matched_overwrite.folder_identifier,
                     )
                 else:
-                    logger.debug(
+                    logger.info(
                         "No resource overwrite matched for %s key='%s' on %s",
                         resource_type,
                         key,

packages/uipath-platform/src/uipath/platform/orchestrator/_mcp_service.py

@@ -200,12 +200,19 @@ async def main():
     def custom_headers(self) -> dict[str, str]:
         return self.folder_headers
 
+    def _resolve_folder_key(self, folder_path: str | None) -> str | None:
+        """Resolve folder key from folder_path, falling back to FolderContext."""
+        if folder_path is not None:
+            return self._folders_service.retrieve_folder_key(folder_path)
+
+        return self._folder_key
+
     def _list_spec(
         self,
         *,
         folder_path: str | None,
     ) -> RequestSpec:
-        folder_key = self._folders_service.retrieve_folder_key(folder_path)
+        folder_key = self._resolve_folder_key(folder_path)
         return RequestSpec(
             method="GET",
             endpoint=Endpoint("/agenthub_/api/servers"),
@@ -220,7 +227,7 @@ def _retrieve_spec(
         *,
         folder_path: str | None,
     ) -> RequestSpec:
-        folder_key = self._folders_service.retrieve_folder_key(folder_path)
+        folder_key = self._resolve_folder_key(folder_path)
         return RequestSpec(
             method="GET",
             endpoint=Endpoint(f"/agenthub_/api/servers/{slug}"),

packages/uipath-platform/tests/services/test_mcp_service.py

@@ -105,14 +105,18 @@ def test_list_with_folder_path(
             assert HEADER_FOLDER_KEY in servers_request.headers
             assert servers_request.headers[HEADER_FOLDER_KEY] == "resolved-folder-key"
 
-        def test_list_without_folder_raises_error(
+        def test_list_without_folder_path_falls_back_to_folder_key(
             self,
+            httpx_mock: HTTPXMock,
             config: UiPathApiConfig,
             execution_context: UiPathExecutionContext,
+            base_url: str,
+            org: str,
+            tenant: str,
             monkeypatch: pytest.MonkeyPatch,
         ) -> None:
-            """Test that listing servers without a folder_path raises ValueError."""
-            monkeypatch.delenv("UIPATH_FOLDER_KEY", raising=False)
+            """Test that listing servers without folder_path falls back to folder_key from FolderContext."""
+            monkeypatch.setenv("UIPATH_FOLDER_KEY", "fallback-folder-key")
             monkeypatch.delenv("UIPATH_FOLDER_PATH", raising=False)
 
             folders_service = FolderService(
@@ -124,11 +128,19 @@ def test_list_without_folder_raises_error(
                 folders_service=folders_service,
             )
 
-            with pytest.raises(
-                ValueError,
-                match="Cannot obtain folder_key without providing folder_path",
-            ):
-                service.list()
+            httpx_mock.add_response(
+                url=f"{base_url}{org}{tenant}/agenthub_/api/servers",
+                status_code=200,
+                json=[],
+            )
+
+            servers = service.list()
+
+            assert servers == []
+            requests = httpx_mock.get_requests()
+            assert len(requests) == 1
+            assert HEADER_FOLDER_KEY in requests[0].headers
+            assert requests[0].headers[HEADER_FOLDER_KEY] == "fallback-folder-key"
 
         @pytest.mark.anyio
         async def test_list_async(

packages/uipath-platform/uv.lock

@@ -1,13 +1,13 @@
 [project]
 name = "uipath"
-version = "2.10.34"
+version = "2.10.35"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
   "uipath-core>=0.5.8, <0.6.0",
   "uipath-runtime>=0.10.0, <0.11.0",
-  "uipath-platform>=0.1.4, <0.2.0",
+  "uipath-platform>=0.1.13, <0.2.0",
   "click>=8.3.1",
   "httpx>=0.28.1",
   "pyjwt>=2.10.1",