Merge pull request #409 from UiPath/chore/optional-ingestion

chore: context grounding, optional index ingestion

Author: radu-mocanu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.0.68"
+version = "2.0.69"
 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.10"

src/uipath/_services/context_grounding_service.py

@@ -56,6 +56,7 @@ def add_to_index(
         source_path: Optional[str] = None,
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
+        ingest_data: bool = True,
     ) -> None:
         """Add content to the index.
 
@@ -67,6 +68,7 @@ def add_to_index(
             source_path (Optional[str]): The source path of the content if it is being uploaded from a file.
             folder_key (Optional[str]): The key of the folder where the index resides.
             folder_path (Optional[str]): The path of the folder where the index resides.
+            ingest_data (bool): Whether to ingest data in the index after content is uploaded. Defaults to True.
 
         Raises:
             ValueError: If neither content nor source_path is provided, or if both are provided.
@@ -94,7 +96,9 @@ def add_to_index(
                 folder_path=bucket_folder_path,
                 content_type=content_type,
             )
-        self.ingest_data(index, folder_key=folder_key, folder_path=folder_path)
+
+        if ingest_data:
+            self.ingest_data(index, folder_key=folder_key, folder_path=folder_path)
 
     @traced(name="add_to_index", run_type="uipath")
     @infer_bindings(resource_type="index")
@@ -107,6 +111,7 @@ async def add_to_index_async(
         source_path: Optional[str] = None,
         folder_key: Optional[str] = None,
         folder_path: Optional[str] = None,
+        ingest_data: bool = True,
     ) -> None:
         """Asynchronously add content to the index.
 
@@ -118,6 +123,7 @@ async def add_to_index_async(
             source_path (Optional[str]): The source path of the content if it is being uploaded from a file.
             folder_key (Optional[str]): The key of the folder where the index resides.
             folder_path (Optional[str]): The path of the folder where the index resides.
+            ingest_data (bool): Whether to ingest data in the index after content is uploaded. Defaults to True.
 
         Raises:
             ValueError: If neither content nor source_path is provided, or if both are provided.
@@ -148,9 +154,10 @@ async def add_to_index_async(
                 content_type=content_type,
             )
 
-        await self.ingest_data_async(
-            index, folder_key=folder_key, folder_path=folder_path
-        )
+        if ingest_data:
+            await self.ingest_data_async(
+                index, folder_key=folder_key, folder_path=folder_path
+            )
 
     @traced(name="contextgrounding_retrieve", run_type="uipath")
     @infer_bindings(resource_type="index")