Merge branch 'main' into feature/add-list-models-command

Author: GabrielVasilescu04

packages/uipath-platform/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-platform"
-version = "0.1.36"
+version = "0.1.38"
 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/context_grounding/_context_grounding_service.py

@@ -17,6 +17,7 @@
     ORCHESTRATOR_STORAGE_BUCKET_DATA_SOURCE,
 )
 from ..errors import (
+    BatchTransformFailedException,
     BatchTransformNotCompleteException,
     IngestionInProgressException,
     UnsupportedDataSourceException,
@@ -1050,6 +1051,10 @@ def download_batch_transform_result(
             batch_transform = self.retrieve_batch_transform(
                 id=id, index_name=index_name
             )
+            if batch_transform.last_batch_rag_status == BatchTransformStatus.FAILED:
+                raise BatchTransformFailedException(
+                    batch_transform_id=id,
+                )
             if batch_transform.last_batch_rag_status != BatchTransformStatus.SUCCESSFUL:
                 raise BatchTransformNotCompleteException(
                     batch_transform_id=id,
@@ -1109,6 +1114,10 @@ async def download_batch_transform_result_async(
             batch_transform = await self.retrieve_batch_transform_async(
                 id=id, index_name=index_name
             )
+            if batch_transform.last_batch_rag_status == BatchTransformStatus.FAILED:
+                raise BatchTransformFailedException(
+                    batch_transform_id=id,
+                )
             if batch_transform.last_batch_rag_status != BatchTransformStatus.SUCCESSFUL:
                 raise BatchTransformNotCompleteException(
                     batch_transform_id=id,

packages/uipath-platform/src/uipath/platform/errors/__init__.py

@@ -8,13 +8,15 @@
 - FolderNotFoundException: Raised when a folder cannot be found
 - UnsupportedDataSourceException: Raised when an operation is attempted on an unsupported data source type
 - IngestionInProgressException: Raised when a search is attempted on an index during ingestion
+- BatchTransformFailedException: Raised when a batch transform has failed
 - BatchTransformNotCompleteException: Raised when attempting to get results from an incomplete batch transform
 - OperationNotCompleteException: Raised when attempting to get results from an incomplete operation
 - OperationFailedException: Raised when an operation has failed
 - EnrichedException: Enriched HTTP error with detailed request/response information
 """
 
 from ._base_url_missing_error import BaseUrlMissingError
+from ._batch_transform_failed_exception import BatchTransformFailedException
 from ._batch_transform_not_complete_exception import BatchTransformNotCompleteException
 from ._enriched_exception import EnrichedException, ExtractedErrorInfo
 from ._folder_not_found_exception import FolderNotFoundException
@@ -26,6 +28,7 @@
 
 __all__ = [
     "BaseUrlMissingError",
+    "BatchTransformFailedException",
     "BatchTransformNotCompleteException",
     "EnrichedException",
     "ExtractedErrorInfo",

packages/uipath-platform/src/uipath/platform/errors/_batch_transform_failed_exception.py

@@ -0,0 +1,10 @@
+class BatchTransformFailedException(Exception):
+    """Raised when a batch transform has failed.
+
+    This exception is raised when a batch transform task has completed
+    with a failed status, as opposed to still being in progress.
+    """
+
+    def __init__(self, batch_transform_id: str):
+        self.message = f"Batch transform '{batch_transform_id}' failed."
+        super().__init__(self.message)

packages/uipath-platform/src/uipath/platform/memory/_memory_service.py

@@ -10,6 +10,7 @@
 from uipath.core.tracing import traced
 
 from ..common._base_service import BaseService
+from ..common._bindings import resource_override
 from ..common._config import UiPathApiConfig
 from ..common._execution_context import UiPathExecutionContext
 from ..common._folder_context import FolderContext, header_folder
@@ -50,13 +51,15 @@ def __init__(
 
     # ── Memory space operations (ECS) ──────────────────────────────────
 
+    @resource_override(resource_type="memorySpace")
     @traced(name="memory_create", run_type="uipath")
     def create(
         self,
         name: str,
         description: Optional[str] = None,
         is_encrypted: Optional[bool] = None,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> MemorySpace:
         """Create a new memory space.
 
@@ -65,11 +68,14 @@ def create(
             description: Optional description (max 1024 chars).
             is_encrypted: Whether the memory space should be encrypted.
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
 
         Returns:
             MemorySpace: The created memory space.
         """
-        spec = self._create_spec(name, description, is_encrypted, folder_key)
+        spec = self._create_spec(
+            name, description, is_encrypted, folder_key, folder_path
+        )
         response = self.request(
             spec.method,
             spec.endpoint,
@@ -78,13 +84,15 @@ def create(
         ).json()
         return MemorySpace.model_validate(response)
 
+    @resource_override(resource_type="memorySpace")
     @traced(name="memory_create", run_type="uipath")
     async def create_async(
         self,
         name: str,
         description: Optional[str] = None,
         is_encrypted: Optional[bool] = None,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> MemorySpace:
         """Asynchronously create a new memory space.
 
@@ -93,11 +101,14 @@ async def create_async(
             description: Optional description (max 1024 chars).
             is_encrypted: Whether the memory space should be encrypted.
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
 
         Returns:
             MemorySpace: The created memory space.
         """
-        spec = self._create_spec(name, description, is_encrypted, folder_key)
+        spec = self._create_spec(
+            name, description, is_encrypted, folder_key, folder_path
+        )
         response = (
             await self.request_async(
                 spec.method,
@@ -116,6 +127,7 @@ def list(
         top: Optional[int] = None,
         skip: Optional[int] = None,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> MemorySpaceListResponse:
         """List memory spaces with optional OData query parameters.
 
@@ -125,11 +137,12 @@ def list(
             top: Maximum number of results.
             skip: Number of results to skip.
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
 
         Returns:
             MemorySpaceListResponse: The list of memory spaces.
         """
-        spec = self._list_spec(filter, orderby, top, skip, folder_key)
+        spec = self._list_spec(filter, orderby, top, skip, folder_key, folder_path)
         response = self.request(
             spec.method,
             spec.endpoint,
@@ -146,6 +159,7 @@ async def list_async(
         top: Optional[int] = None,
         skip: Optional[int] = None,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> MemorySpaceListResponse:
         """Asynchronously list memory spaces.
 
@@ -155,11 +169,12 @@ async def list_async(
             top: Maximum number of results.
             skip: Number of results to skip.
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
 
         Returns:
             MemorySpaceListResponse: The list of memory spaces.
         """
-        spec = self._list_spec(filter, orderby, top, skip, folder_key)
+        spec = self._list_spec(filter, orderby, top, skip, folder_key, folder_path)
         response = (
             await self.request_async(
                 spec.method,
@@ -178,6 +193,7 @@ def search(
         memory_space_id: str,
         request: MemorySearchRequest,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> MemorySearchResponse:
         """Search a memory space via LLMOps.
 
@@ -188,11 +204,12 @@ def search(
             memory_space_id: The GUID of the memory space.
             request: The search request payload.
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
 
         Returns:
             MemorySearchResponse: Results, metadata, and system prompt injection.
         """
-        spec = self._search_spec(memory_space_id, folder_key)
+        spec = self._search_spec(memory_space_id, folder_key, folder_path)
         response = self.request(
             spec.method,
             spec.endpoint,
@@ -207,6 +224,7 @@ async def search_async(
         memory_space_id: str,
         request: MemorySearchRequest,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> MemorySearchResponse:
         """Asynchronously search a memory space via LLMOps.
 
@@ -217,11 +235,12 @@ async def search_async(
             memory_space_id: The GUID of the memory space.
             request: The search request payload.
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
 
         Returns:
             MemorySearchResponse: Results, metadata, and system prompt injection.
         """
-        spec = self._search_spec(memory_space_id, folder_key)
+        spec = self._search_spec(memory_space_id, folder_key, folder_path)
         response = (
             await self.request_async(
                 spec.method,
@@ -240,6 +259,7 @@ def escalation_search(
         memory_space_id: str,
         request: MemorySearchRequest,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> EscalationMemorySearchResponse:
         """Search escalation memory for previously resolved outcomes.
 
@@ -250,11 +270,12 @@ def escalation_search(
             memory_space_id: The GUID of the memory space.
             request: The search request payload (same as regular search).
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
 
         Returns:
             EscalationMemorySearchResponse: Matched escalation outcomes.
         """
-        spec = self._escalation_search_spec(memory_space_id, folder_key)
+        spec = self._escalation_search_spec(memory_space_id, folder_key, folder_path)
         response = self.request(
             spec.method,
             spec.endpoint,
@@ -269,6 +290,7 @@ async def escalation_search_async(
         memory_space_id: str,
         request: MemorySearchRequest,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> EscalationMemorySearchResponse:
         """Asynchronously search escalation memory for previously resolved outcomes.
 
@@ -279,11 +301,12 @@ async def escalation_search_async(
             memory_space_id: The GUID of the memory space.
             request: The search request payload (same as regular search).
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
 
         Returns:
             EscalationMemorySearchResponse: Matched escalation outcomes.
         """
-        spec = self._escalation_search_spec(memory_space_id, folder_key)
+        spec = self._escalation_search_spec(memory_space_id, folder_key, folder_path)
         response = (
             await self.request_async(
                 spec.method,
@@ -300,6 +323,7 @@ def escalation_ingest(
         memory_space_id: str,
         request: EscalationMemoryIngestRequest,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> None:
         """Ingest a resolved escalation outcome into memory.
 
@@ -310,8 +334,9 @@ def escalation_ingest(
             memory_space_id: The GUID of the memory space.
             request: The escalation ingest payload.
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
         """
-        spec = self._escalation_ingest_spec(memory_space_id, folder_key)
+        spec = self._escalation_ingest_spec(memory_space_id, folder_key, folder_path)
         self.request(
             spec.method,
             spec.endpoint,
@@ -325,6 +350,7 @@ async def escalation_ingest_async(
         memory_space_id: str,
         request: EscalationMemoryIngestRequest,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> None:
         """Asynchronously ingest a resolved escalation outcome into memory.
 
@@ -335,8 +361,9 @@ async def escalation_ingest_async(
             memory_space_id: The GUID of the memory space.
             request: The escalation ingest payload.
             folder_key: The folder key for the operation.
+            folder_path: The folder path for the operation.
         """
-        spec = self._escalation_ingest_spec(memory_space_id, folder_key)
+        spec = self._escalation_ingest_spec(memory_space_id, folder_key, folder_path)
         await self.request_async(
             spec.method,
             spec.endpoint,
@@ -379,8 +406,9 @@ def _create_spec(
         description: Optional[str],
         is_encrypted: Optional[bool],
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        folder_key = self._resolve_folder(folder_key)
+        folder_key = self._resolve_folder(folder_key, folder_path)
         body = MemorySpaceCreateRequest(
             name=name,
             description=description,
@@ -400,8 +428,9 @@ def _list_spec(
         top: Optional[int],
         skip: Optional[int],
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        folder_key = self._resolve_folder(folder_key)
+        folder_key = self._resolve_folder(folder_key, folder_path)
         params: dict[str, Any] = {}
         if filter is not None:
             params["$filter"] = filter
@@ -424,8 +453,9 @@ def _search_spec(
         self,
         memory_space_id: str,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        folder_key = self._resolve_folder(folder_key)
+        folder_key = self._resolve_folder(folder_key, folder_path)
         return RequestSpec(
             method="POST",
             endpoint=Endpoint(f"{_LLMOPS_AGENT_BASE}/{memory_space_id}/search"),
@@ -436,8 +466,9 @@ def _escalation_search_spec(
         self,
         memory_space_id: str,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        folder_key = self._resolve_folder(folder_key)
+        folder_key = self._resolve_folder(folder_key, folder_path)
         return RequestSpec(
             method="POST",
             endpoint=Endpoint(
@@ -450,8 +481,9 @@ def _escalation_ingest_spec(
         self,
         memory_space_id: str,
         folder_key: Optional[str] = None,
+        folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        folder_key = self._resolve_folder(folder_key)
+        folder_key = self._resolve_folder(folder_key, folder_path)
         return RequestSpec(
             method="POST",
             endpoint=Endpoint(

packages/uipath-platform/src/uipath/platform/resume_triggers/_protocol.py

@@ -53,6 +53,7 @@
     ContextGroundingIndex,
 )
 from uipath.platform.errors import (
+    BatchTransformFailedException,
     BatchTransformNotCompleteException,
     OperationNotCompleteException,
 )
@@ -323,6 +324,11 @@ async def read_trigger(self, trigger: UiPathResumeTrigger) -> Any | None:
                                 "index_name", trigger.payload
                             ),
                         )
+                    except BatchTransformFailedException as e:
+                        raise UiPathFaultedTriggerError(
+                            ErrorCategory.SYSTEM,
+                            f"{e.message}",
+                        ) from e
                     except BatchTransformNotCompleteException as e:
                         raise UiPathPendingTriggerError(
                             ErrorCategory.SYSTEM,