chore: version bump

Author: gcuip

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.6.26"
+version = "2.6.31"
 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"

src/uipath/platform/context_grounding/_context_grounding_service.py

@@ -782,14 +782,18 @@ def download_batch_transform_result(
 
         Path(destination_path).parent.mkdir(parents=True, exist_ok=True)
 
+        # SAS uris can be downloaded without authentication
+        # encrypted artifacts require authenticated DownloadBlob endpoint
         with open(destination_path, "wb") as file:
             if uri_response.is_encrypted:
-                # Use authenticated client for encrypted artifacts
-                file_content = self._client.get(
-                    uri_response.uri, headers=self.auth_headers
-                ).content
+                download_spec = self._batch_transform_download_blob_spec(id=id)
+                download_response = self.request(
+                    download_spec.method,
+                    download_spec.endpoint,
+                    headers=download_spec.headers,
+                )
+                file_content = download_response.content
             else:
-                # Use unauthenticated client for non-encrypted artifacts
                 with httpx.Client(**get_httpx_client_kwargs()) as client:
                     file_content = client.get(uri_response.uri).content
             file.write(file_content)
@@ -835,14 +839,17 @@ async def download_batch_transform_result_async(
         )
         uri_response = BatchTransformReadUriResponse.model_validate(response.json())
 
+        # SAS uris can be downloaded without authentication
+        # encrypted artifacts require authenticated DownloadBlob endpoint
         if uri_response.is_encrypted:
-            # Use authenticated client for encrypted artifacts
-            download_response = await self._client_async.get(
-                uri_response.uri, headers=self.auth_headers
+            download_spec = self._batch_transform_download_blob_spec(id=id)
+            download_response = await self.request_async(
+                download_spec.method,
+                download_spec.endpoint,
+                headers=download_spec.headers,
             )
             file_content = download_response.content
         else:
-            # Use unauthenticated client for non-encrypted artifacts
             async with httpx.AsyncClient(**get_httpx_client_kwargs()) as client:
                 download_response = await client.get(uri_response.uri)
                 file_content = download_response.content
@@ -1530,6 +1537,15 @@ def _batch_transform_get_read_uri_spec(
             endpoint=Endpoint(f"/ecs_/v2/batchRag/{id}/GetReadUri"),
         )
 
+    def _batch_transform_download_blob_spec(
+        self,
+        id: str,
+    ) -> RequestSpec:
+        return RequestSpec(
+            method="GET",
+            endpoint=Endpoint(f"/ecs_/v2/batchRag/{id}/DownloadBlob"),
+        )
+
     def _resolve_folder_key(self, folder_key, folder_path):
         if folder_key is None and folder_path is not None:
             folder_key = self._folders_service.retrieve_key(folder_path=folder_path)

tests/sdk/services/test_context_grounding_service.py

@@ -1794,13 +1794,13 @@ def test_download_batch_transform_result_encrypted(
             url=f"{base_url}{org}{tenant}/ecs_/v2/batchRag/test-batch-id/GetReadUri",
             status_code=200,
             json={
-                "uri": f"{base_url}{org}{tenant}/storage/encrypted/result.csv",
+                "uri": f"{base_url}{org}{tenant}/ecs_/v2/batchRag/test-batch-id/DownloadBlob",
                 "isEncrypted": True,
             },
         )
 
         httpx_mock.add_response(
-            url=f"{base_url}{org}{tenant}/storage/encrypted/result.csv",
+            url=f"{base_url}{org}{tenant}/ecs_/v2/batchRag/test-batch-id/DownloadBlob",
             status_code=200,
             content=b"encrypted,data\nval1,val2",
         )
@@ -1818,9 +1818,10 @@ def test_download_batch_transform_result_encrypted(
         if sent_requests is None:
             raise Exception("No request was sent")
 
-        # Verify the download request includes Authorization header
+        # Verify the DownloadBlob endpoint was called with Authorization header
         download_request = sent_requests[2]
         assert download_request.method == "GET"
+        assert "/DownloadBlob" in str(download_request.url)
         assert "Authorization" in download_request.headers
         assert download_request.headers["Authorization"].startswith("Bearer ")
 
@@ -2019,13 +2020,13 @@ async def test_download_batch_transform_result_async_encrypted(
             url=f"{base_url}{org}{tenant}/ecs_/v2/batchRag/test-batch-id/GetReadUri",
             status_code=200,
             json={
-                "uri": f"{base_url}{org}{tenant}/storage/encrypted/result.csv",
+                "uri": f"{base_url}{org}{tenant}/ecs_/v2/batchRag/test-batch-id/DownloadBlob",
                 "isEncrypted": True,
             },
         )
 
         httpx_mock.add_response(
-            url=f"{base_url}{org}{tenant}/storage/encrypted/result.csv",
+            url=f"{base_url}{org}{tenant}/ecs_/v2/batchRag/test-batch-id/DownloadBlob",
             status_code=200,
             content=b"encrypted,data\nval1,val2",
         )
@@ -2043,8 +2044,9 @@ async def test_download_batch_transform_result_async_encrypted(
         if sent_requests is None:
             raise Exception("No request was sent")
 
-        # Verify the download request includes Authorization header
+        # Verify the DownloadBlob endpoint was called with Authorization header
         download_request = sent_requests[2]
         assert download_request.method == "GET"
+        assert "/DownloadBlob" in str(download_request.url)
         assert "Authorization" in download_request.headers
         assert download_request.headers["Authorization"].startswith("Bearer ")