uipath 2.0.39__py3-none-any.whl → 2.0.40__py3-none-any.whl

uipath/_services/context_grounding_service.py

@@ -1,5 +1,5 @@
 import json
-from typing import Any, Dict, List, Optional
+from typing import Any, List, Optional
 
 from pydantic import TypeAdapter
 
@@ -8,8 +8,6 @@ from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
 from .._utils import Endpoint, RequestSpec, header_folder
 from .._utils.constants import (
-    HEADER_FOLDER_KEY,
-    HEADER_FOLDER_PATH,
     ORCHESTRATOR_STORAGE_BUCKET_DATA_SOURCE,
 )
 from ..models import IngestionInProgressException
@@ -440,36 +438,19 @@ class ContextGroundingService(FolderContext, BaseService):
             headers=spec.headers,
         )
 
-    @property
-    def custom_headers(self) -> Dict[str, str]:
-        self._folder_key = self._folder_key or (
-            self._folders_service.retrieve_key_by_folder_path(self._folder_path)
-            if self._folder_path
-            else None
-        )
-
-        if self._folder_key is None:
-            raise ValueError(
-                f"Neither the folder key nor the folder path is set ({HEADER_FOLDER_KEY}, {HEADER_FOLDER_PATH})"
-            )
-
-        return self.folder_headers
-
     def _ingest_spec(
         self,
         key: str,
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        if folder_key is None and folder_path is not None:
-            folder_key = self._folders_service.retrieve_key_by_folder_path(folder_path)
-            folder_path = None
+        folder_key = self._resolve_folder_key(folder_key, folder_path)
 
         return RequestSpec(
             method="POST",
             endpoint=Endpoint(f"/ecs_/v2/indexes/{key}/ingest"),
             headers={
-                **header_folder(folder_key, folder_path),
+                **header_folder(folder_key, None),
             },
         )
 
@@ -479,17 +460,14 @@ class ContextGroundingService(FolderContext, BaseService):
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        print(folder_key, folder_path)
-        if folder_key is None and folder_path is not None:
-            folder_key = self._folders_service.retrieve_key_by_folder_path(folder_path)
-            folder_path = None
-        print("~~~", name, folder_key, folder_path)
+        folder_key = self._resolve_folder_key(folder_key, folder_path)
+
         return RequestSpec(
             method="GET",
             endpoint=Endpoint("/ecs_/v2/indexes"),
             params={"$filter": f"Name eq '{name}'"},
             headers={
-                **header_folder(folder_key, folder_path),
+                **header_folder(folder_key, None),
             },
         )
 
@@ -503,9 +481,7 @@ class ContextGroundingService(FolderContext, BaseService):
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        if folder_key is None and folder_path is not None:
-            folder_key = self._folders_service.retrieve_key_by_folder_path(folder_path)
-            folder_path = None
+        folder_key = self._resolve_folder_key(folder_key, folder_path)
 
         storage_bucket_folder_path = (
             storage_bucket_folder_path
@@ -531,7 +507,7 @@ class ContextGroundingService(FolderContext, BaseService):
                 }
             ),
             headers={
-                **header_folder(folder_key, folder_path),
+                **header_folder(folder_key, None),
             },
         )
 
@@ -541,15 +517,13 @@ class ContextGroundingService(FolderContext, BaseService):
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        if folder_key is None and folder_path is not None:
-            folder_key = self._folders_service.retrieve_key_by_folder_path(folder_path)
-            folder_path = None
+        folder_key = self._resolve_folder_key(folder_key, folder_path)
 
         return RequestSpec(
             method="GET",
             endpoint=Endpoint(f"/ecs_/v2/indexes/{id}"),
             headers={
-                **header_folder(folder_key, folder_path),
+                **header_folder(folder_key, None),
             },
         )
 
@@ -559,15 +533,13 @@ class ContextGroundingService(FolderContext, BaseService):
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        if folder_key is None and folder_path is not None:
-            folder_key = self._folders_service.retrieve_key_by_folder_path(folder_path)
-            folder_path = None
+        folder_key = self._resolve_folder_key(folder_key, folder_path)
 
         return RequestSpec(
             method="DELETE",
             endpoint=Endpoint(f"/ecs_/v2/indexes/{id}"),
             headers={
-                **header_folder(folder_key, folder_path),
+                **header_folder(folder_key, None),
             },
         )
 
@@ -579,9 +551,7 @@ class ContextGroundingService(FolderContext, BaseService):
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
     ) -> RequestSpec:
-        if folder_key is None and folder_path is not None:
-            folder_key = self._folders_service.retrieve_key_by_folder_path(folder_path)
-            folder_path = None
+        folder_key = self._resolve_folder_key(folder_key, folder_path)
 
         return RequestSpec(
             method="POST",
@@ -593,6 +563,22 @@ class ContextGroundingService(FolderContext, BaseService):
                 }
             ),
             headers={
-                **header_folder(folder_key, folder_path),
+                **header_folder(folder_key, None),
             },
         )
+
+    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_by_folder_path(folder_path)
+
+        if folder_key is None and folder_path is None:
+            folder_key = self._folder_key or (
+                self._folders_service.retrieve_key_by_folder_path(self._folder_path)
+                if self._folder_path
+                else None
+            )
+
+        if folder_key is None:
+            raise ValueError("Folder key or folder path is required")
+
+        return folder_key

{uipath-2.0.39.dist-info → uipath-2.0.40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath
-Version: 2.0.39
+Version: 2.0.40
 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.39.dist-info → uipath-2.0.40.dist-info}/RECORD

@@ -47,7 +47,7 @@ uipath/_services/assets_service.py,sha256=4TsHd_b3VQZlN195ecrCHbQcaNew0GWEk90SFh
 uipath/_services/buckets_service.py,sha256=m0HMWBkooUhjTtna_ZXcw4QOzKmaibuepWlC8wPGtlA,9330
 uipath/_services/connections_service.py,sha256=bE-t-YS_C67bcyEA9xNwKqv5b0dN7qh0McZdGETcEAQ,7338
 uipath/_services/connections_service.pyi,sha256=6OOnh0aCfxhETL8n_JZ6Xoe2BE3ST_7Vz-FgLZc53lM,2465
-uipath/_services/context_grounding_service.py,sha256=nZDntNcpPNHcEfgw-Aq-9t2YpVx1uPTphBnVwnztM6E,19287
+uipath/_services/context_grounding_service.py,sha256=eRBe5a3uubFCYVHwt6KWCYNW8MMjID5ri_uHefIt1KY,18573
 uipath/_services/folder_service.py,sha256=HtsBoBejvMuIZ-9gocAG9B8uKOFsAAD4WUozta-isXk,1673
 uipath/_services/jobs_service.py,sha256=MsJlu1egvHKZhHdammp4Xo9iJzceWquW4qIWT-nPBws,8214
 uipath/_services/llm_gateway_service.py,sha256=ySg3sflIoXmY9K7txlSm7bkuI2qzBT0kAKmGlFBk5KA,12032
@@ -79,8 +79,8 @@ uipath/tracing/__init__.py,sha256=GimSzv6qkCOlHOG1WtjYKJsZqcXpA28IgoXfR33JhiA,13
 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.39.dist-info/METADATA,sha256=Jzqb2X_hxkHUsfEQJHRtcvnINIWsz_q0qg-OUYlrzSY,6254
-uipath-2.0.39.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath-2.0.39.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
-uipath-2.0.39.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
-uipath-2.0.39.dist-info/RECORD,,
+uipath-2.0.40.dist-info/METADATA,sha256=lzQ5m9oN0aWLZFLw5U_FLUTmKThpnDEySj6axOTky2A,6254
+uipath-2.0.40.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath-2.0.40.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
+uipath-2.0.40.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
+uipath-2.0.40.dist-info/RECORD,,