uipath 2.0.56__py3-none-any.whl → 2.0.57__py3-none-any.whl

uipath/_services/buckets_service.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, Optional, Union
 
-from httpx import request
+import httpx
 
 from .._config import Config
 from .._execution_context import ExecutionContext
@@ -25,6 +25,8 @@ class BucketsService(FolderContext, BaseService):
 
     def __init__(self, config: Config, execution_context: ExecutionContext) -> None:
         super().__init__(config=config, execution_context=execution_context)
+        self.custom_client = httpx.Client()
+        self.custom_client_async = httpx.AsyncClient()
 
     @traced(name="buckets_download", run_type="uipath")
     def download(
@@ -78,7 +80,7 @@ class BucketsService(FolderContext, BaseService):
             if result["RequiresAuth"]:
                 file_content = self.request("GET", read_uri, headers=headers).content
             else:
-                file_content = request("GET", read_uri, headers=headers).content
+                file_content = self.custom_client.get(read_uri, headers=headers).content
             file.write(file_content)
 
     @traced(name="buckets_upload", run_type="uipath")
@@ -89,7 +91,8 @@ class BucketsService(FolderContext, BaseService):
         name: Optional[str] = None,
         blob_file_path: str,
         content_type: str,
-        source_path: str,
+        source_path: Optional[str] = None,
+        content: Optional[Union[str, bytes]] = None,
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
     ) -> None:
@@ -100,7 +103,8 @@ class BucketsService(FolderContext, BaseService):
             name (Optional[str]): The name of the bucket.
             blob_file_path (str): The path where the file will be stored in the bucket.
             content_type (str): The MIME type of the file.
-            source_path (str): The local path of the file to upload.
+            source_path (Optional[str]): The local path of the file to upload.
+            content (Optional[Union[str, bytes]]): The content to upload (string or bytes).
             folder_key (Optional[str]): The key of the folder where the bucket resides.
             folder_path (Optional[str]): The path of the folder where the bucket resides.
 
@@ -108,6 +112,11 @@ class BucketsService(FolderContext, BaseService):
             ValueError: If neither key nor name is provided.
             Exception: If the bucket with the specified key or name is not found.
         """
+        if content is not None and source_path is not None:
+            raise ValueError("Content and source_path are mutually exclusive")
+        if content is None and source_path is None:
+            raise ValueError("Either content or source_path must be provided")
+
         bucket = self.retrieve(
             name=name, key=key, folder_key=folder_key, folder_path=folder_path
         )
@@ -136,11 +145,25 @@ class BucketsService(FolderContext, BaseService):
             )
         }
 
-        with open(source_path, "rb") as file:
+        if content is not None:
+            if isinstance(content, str):
+                content = content.encode("utf-8")
+
             if result["RequiresAuth"]:
-                self.request("PUT", write_uri, headers=headers, files={"file": file})
+                self.request("PUT", write_uri, headers=headers, content=content)
             else:
-                request("PUT", write_uri, headers=headers, files={"file": file})
+                self.custom_client.put(write_uri, headers=headers, content=content)
+
+        if source_path is not None:
+            with open(source_path, "rb") as file:
+                if result["RequiresAuth"]:
+                    self.request(
+                        "PUT", write_uri, headers=headers, files={"file": file}
+                    )
+                else:
+                    self.custom_client.put(
+                        write_uri, headers=headers, files={"file": file}
+                    )
 
     @traced(name="buckets_upload", run_type="uipath")
     async def upload_async(
@@ -150,7 +173,8 @@ class BucketsService(FolderContext, BaseService):
         name: Optional[str] = None,
         blob_file_path: str,
         content_type: str,
-        source_path: str,
+        source_path: Optional[str] = None,
+        content: Optional[Union[str, bytes]] = None,
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
     ) -> None:
@@ -169,6 +193,11 @@ class BucketsService(FolderContext, BaseService):
             ValueError: If neither key nor name is provided.
             Exception: If the bucket with the specified key or name is not found.
         """
+        if content is not None and source_path is not None:
+            raise ValueError("Content and source_path are mutually exclusive")
+        if content is None and source_path is None:
+            raise ValueError("Either content or source_path must be provided")
+
         bucket = await self.retrieve_async(
             name=name, key=key, folder_key=folder_key, folder_path=folder_path
         )
@@ -199,148 +228,29 @@ class BucketsService(FolderContext, BaseService):
             )
         }
 
-        with open(source_path, "rb") as file:
+        if content is not None:
+            if isinstance(content, str):
+                content = content.encode("utf-8")
+
             if result["RequiresAuth"]:
                 await self.request_async(
-                    "PUT", write_uri, headers=headers, files={"file": file}
+                    "PUT", write_uri, headers=headers, content=content
                 )
             else:
-                request("PUT", write_uri, headers=headers, files={"file": file})
-
-    @traced(
-        name="buckets_upload_from_memory",
-        run_type="uipath",
-        input_processor=_upload_from_memory_input_processor,
-    )
-    def upload_from_memory(
-        self,
-        *,
-        key: Optional[str] = None,
-        name: Optional[str] = None,
-        blob_file_path: str,
-        content_type: str,
-        content: Union[str, bytes],
-        folder_key: Optional[str] = None,
-        folder_path: Optional[str] = None,
-    ) -> None:
-        """Upload content from memory to a bucket.
-
-        Args:
-            key (Optional[str]): The key of the bucket.
-            name (Optional[str]): The name of the bucket.
-            blob_file_path (str): The path where the content will be stored in the bucket.
-            content_type (str): The MIME type of the content.
-            content (Union[str, bytes]): The content to upload (string or bytes).
-            folder_key (Optional[str]): The key of the folder where the bucket resides.
-            folder_path (Optional[str]): The path of the folder where the bucket resides.
-
-        Raises:
-            ValueError: If neither key nor name is provided.
-            Exception: If the bucket with the specified key or name is not found.
-        """
-        bucket = self.retrieve(
-            name=name, key=key, folder_key=folder_key, folder_path=folder_path
-        )
-
-        spec = self._retrieve_writeri_spec(
-            bucket.id,
-            content_type,
-            blob_file_path,
-            folder_key=folder_key,
-            folder_path=folder_path,
-        )
-
-        result = self.request(
-            spec.method,
-            url=spec.endpoint,
-            params=spec.params,
-            headers=spec.headers,
-        ).json()
-
-        write_uri = result["Uri"]
-
-        headers = {
-            key: value
-            for key, value in zip(
-                result["Headers"]["Keys"], result["Headers"]["Values"], strict=False
-            )
-        }
-
-        # Convert string to bytes if needed
-        if isinstance(content, str):
-            content = content.encode("utf-8")
-
-        if result["RequiresAuth"]:
-            self.request("PUT", write_uri, headers=headers, content=content)
-        else:
-            request("PUT", write_uri, headers=headers, content=content)
-
-    async def upload_from_memory_async(
-        self,
-        *,
-        key: Optional[str] = None,
-        name: Optional[str] = None,
-        blob_file_path: str,
-        content_type: str,
-        content: Union[str, bytes],
-        folder_key: Optional[str] = None,
-        folder_path: Optional[str] = None,
-    ) -> None:
-        """Asynchronously upload content from memory to a bucket.
-
-        Args:
-            key (Optional[str]): The key of the bucket.
-            name (Optional[str]): The name of the bucket.
-            blob_file_path (str): The path where the content will be stored in the bucket.
-            content_type (str): The MIME type of the content.
-            content (Union[str, bytes]): The content to upload (string or bytes).
-            folder_key (Optional[str]): The key of the folder where the bucket resides.
-            folder_path (Optional[str]): The path of the folder where the bucket resides.
-
-        Raises:
-            ValueError: If neither key nor name is provided.
-            Exception: If the bucket with the specified key or name is not found.
-        """
-        bucket = await self.retrieve_async(
-            name=name, key=key, folder_key=folder_key, folder_path=folder_path
-        )
-
-        bucket_id = bucket["Id"]
-
-        spec = self._retrieve_writeri_spec(
-            bucket_id,
-            content_type,
-            blob_file_path,
-            folder_key=folder_key,
-            folder_path=folder_path,
-        )
-
-        result = (
-            await self.request_async(
-                spec.method,
-                url=spec.endpoint,
-                params=spec.params,
-                headers=spec.headers,
-            )
-        ).json()
-
-        write_uri = result["Uri"]
-
-        headers = {
-            key: value
-            for key, value in zip(
-                result["Headers"]["Keys"], result["Headers"]["Values"], strict=False
-            )
-        }
-
-        # Convert string to bytes if needed
-        if isinstance(content, str):
-            content = content.encode("utf-8")
+                await self.custom_client_async.put(
+                    write_uri, headers=headers, content=content
+                )
 
-        if result["RequiresAuth"]:
-            await self.request_async("PUT", write_uri, headers=headers, content=content)
-        else:
-            request("PUT", write_uri, headers=headers, content=content)
+        if source_path is not None:
+            with open(source_path, "rb") as file:
+                if result["RequiresAuth"]:
+                    await self.request_async(
+                        "PUT", write_uri, headers=headers, files={"file": file}
+                    )
+                else:
+                    await self.custom_client_async.put(
+                        write_uri, headers=headers, files={"file": file}
+                    )
 
     @infer_bindings()
     @traced(name="buckets_retrieve", run_type="uipath")

uipath/_services/context_grounding_service.py

@@ -15,8 +15,8 @@ from ..models import IngestionInProgressException
 from ..models.context_grounding import ContextGroundingQueryResponse
 from ..models.context_grounding_index import ContextGroundingIndex
 from ..tracing._traced import traced
-from . import BucketsService
 from ._base_service import BaseService
+from .buckets_service import BucketsService
 from .folder_service import FolderService
 
 
@@ -85,7 +85,7 @@ class ContextGroundingService(FolderContext, BaseService):
                 content_type=content_type,
             )
         else:
-            self._buckets_service.upload_from_memory(
+            self._buckets_service.upload(
                 name=bucket_name,
                 content=content,
                 blob_file_path=blob_file_path,
@@ -137,9 +137,9 @@ class ContextGroundingService(FolderContext, BaseService):
                 content_type=content_type,
             )
         else:
-            await self._buckets_service.upload_from_memory_async(
+            await self._buckets_service.upload_async(
                 name=bucket_name,
-                content=content,  # type: ignore
+                content=content,
                 blob_file_path=blob_file_path,
                 folder_path=bucket_folder_path,
                 content_type=content_type,

uipath/telemetry/_constants.py

@@ -1,4 +1,4 @@
-_CONNECTION_STRING = ""
+_CONNECTION_STRING = "InstrumentationKey=d3da6dfc-5231-4d72-9696-f680a64683b8;IngestionEndpoint=https://westeurope-5.in.applicationinsights.azure.com/;LiveEndpoint=https://westeurope.livediagnostics.monitor.azure.com/;ApplicationId=9b73b209-6c6a-4d4a-9549-b922cef418d5"
 
 _APP_INSIGHTS_EVENT_MARKER_ATTRIBUTE = "APPLICATION_INSIGHTS_EVENT_MARKER_ATTRIBUTE"
 _OTEL_RESOURCE_ATTRIBUTES = "OTEL_RESOURCE_ATTRIBUTES"

{uipath-2.0.56.dist-info → uipath-2.0.57.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath
-Version: 2.0.56
+Version: 2.0.57
 Summary: Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools.
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-python

{uipath-2.0.56.dist-info → uipath-2.0.57.dist-info}/RECORD

@@ -45,9 +45,9 @@ uipath/_services/actions_service.py,sha256=LYKvG4VxNGQgZ46AzGK9kI1Txb-YmVvZj5ScP
 uipath/_services/api_client.py,sha256=hcof0EMa4-phEHD1WlO7Tdfzq6aL18Sbi2aBE7lJm1w,1821
 uipath/_services/assets_service.py,sha256=gfQLCchT6evsmhip1-coX6oFbshoKUWlxwGrS6DGcHU,13200
 uipath/_services/attachments_service.py,sha256=8iRdauPFzhJv65Uus69BBmA3jPckqfwBE4iGbXvktCQ,19653
-uipath/_services/buckets_service.py,sha256=xTIAEs7EbpyZYqd7PG1q0emOOBM_Ca0rVoH417g2bl0,17521
+uipath/_services/buckets_service.py,sha256=736isvj01_-Srh0JiorS02pOwjF0_Loaf-xR6u0B7Ww,14932
 uipath/_services/connections_service.py,sha256=qh-HNL_GJsyPUD0wSJZRF8ZdrTE9l4HrIilmXGK6dDk,4581
-uipath/_services/context_grounding_service.py,sha256=wRYPnpTFeZunS88OggRZ9qRaILHKdoEP_6VUCaF-Xw0,24097
+uipath/_services/context_grounding_service.py,sha256=nE-RZuean_DXK7a6pcA4S2fcWcfWmQ-rmOihi5_qErU,24072
 uipath/_services/folder_service.py,sha256=HtsBoBejvMuIZ-9gocAG9B8uKOFsAAD4WUozta-isXk,1673
 uipath/_services/jobs_service.py,sha256=Z1CmglIe6pFt0XovPLbuLCp13mWQw92MgjfRivqsodE,16745
 uipath/_services/llm_gateway_service.py,sha256=ySg3sflIoXmY9K7txlSm7bkuI2qzBT0kAKmGlFBk5KA,12032
@@ -80,14 +80,14 @@ uipath/models/llm_gateway.py,sha256=0sl5Wtve94V14H3AHwmJSoXAhoc-Fai3wJxP8HrnBPg,
 uipath/models/processes.py,sha256=Atvfrt6X4TYST3iA62jpS_Uxc3hg6uah11p-RaKZ6dk,2029
 uipath/models/queues.py,sha256=N_s0GKucbyjh0RnO8SxPk6wlRgvq8KIIYsfaoIY46tM,6446
 uipath/telemetry/__init__.py,sha256=Wna32UFzZR66D-RzTKlPWlvji9i2HJb82NhHjCCXRjY,61
-uipath/telemetry/_constants.py,sha256=UftH_cYzB260IwlDOafDJHYRbIUXL6_K0rdavG6cxjM,432
+uipath/telemetry/_constants.py,sha256=RVymOX07CRzcGMn1OWzh-TR6bZMvLG3vdCA6fvRrrCk,678
 uipath/telemetry/_track.py,sha256=v0e3hgwtetMsUco4yosBzNU00Ek5SI9RxUTumrTTNyo,3872
 uipath/tracing/__init__.py,sha256=GimSzv6qkCOlHOG1WtjYKJsZqcXpA28IgoXfR33JhiA,139
 uipath/tracing/_otel_exporters.py,sha256=x0PDPmDKJcxashsuehVsSsqBCzRr6WsNFaq_3_HS5F0,3014
 uipath/tracing/_traced.py,sha256=GFxOp73jk0vGTN_H7YZOOsEl9rVLaEhXGztMiYKIA-8,16634
 uipath/tracing/_utils.py,sha256=5SwsTGpHkIouXBndw-u8eCLnN4p7LM8DsTCCuf2jJgs,10165
-uipath-2.0.56.dist-info/METADATA,sha256=-6oLullDWbwXLA9RDKohTdX3H6qupXwdkyl8hK6940o,6304
-uipath-2.0.56.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath-2.0.56.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
-uipath-2.0.56.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
-uipath-2.0.56.dist-info/RECORD,,
+uipath-2.0.57.dist-info/METADATA,sha256=OKwYNKZD-y2kOHlywNqDxh1b-lU0NpUCb1btkd6IwHY,6304
+uipath-2.0.57.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath-2.0.57.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
+uipath-2.0.57.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
+uipath-2.0.57.dist-info/RECORD,,