Skip to content

Commit 29d38d9

Browse files
committed
feat: make _retrieve_system_indexes private
1 parent f0eaefe commit 29d38d9

2 files changed

Lines changed: 11 additions & 37 deletions

File tree

packages/uipath-platform/src/uipath/platform/context_grounding/_context_grounding_service.py

Lines changed: 4 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -258,23 +258,10 @@ async def retrieve_across_folders_async(
258258
]
259259

260260
@traced(name="contextgrounding_retrieve_system_indexes", run_type="uipath")
261-
def retrieve_system_indexes(
261+
def _retrieve_system_indexes(
262262
self,
263263
name: Optional[str] = None,
264264
) -> List[ContextGroundingIndex]:
265-
"""Retrieve tenant-wide system context grounding indexes.
266-
267-
System indexes are shared across all folders and tenants and do not
268-
require a folder key. They are typically used by bundled StudioWeb
269-
templates that reference shared indexes.
270-
271-
Args:
272-
name (Optional[str]): Optional name filter. If provided, only indexes
273-
matching this name will be returned.
274-
275-
Returns:
276-
List[ContextGroundingIndex]: A list of system indexes.
277-
"""
278265
spec = self._retrieve_system_indexes_spec(name=name)
279266

280267
response = self.request(
@@ -288,23 +275,10 @@ def retrieve_system_indexes(
288275
]
289276

290277
@traced(name="contextgrounding_retrieve_system_indexes", run_type="uipath")
291-
async def retrieve_system_indexes_async(
278+
async def _retrieve_system_indexes_async(
292279
self,
293280
name: Optional[str] = None,
294281
) -> List[ContextGroundingIndex]:
295-
"""Asynchronously retrieve tenant-wide system context grounding indexes.
296-
297-
System indexes are shared across all folders and tenants and do not
298-
require a folder key. They are typically used by bundled StudioWeb
299-
templates that reference shared indexes.
300-
301-
Args:
302-
name (Optional[str]): Optional name filter. If provided, only indexes
303-
matching this name will be returned.
304-
305-
Returns:
306-
List[ContextGroundingIndex]: A list of system indexes.
307-
"""
308282
spec = self._retrieve_system_indexes_spec(name=name)
309283

310284
response = (
@@ -446,7 +420,7 @@ async def retrieve_async(
446420
raise ContextGroundingIndexNotFoundError(name) from None
447421

448422
def _retrieve_from_system_indexes(self, name: str) -> ContextGroundingIndex:
449-
indexes = self.retrieve_system_indexes(name=name)
423+
indexes = self._retrieve_system_indexes(name=name)
450424
try:
451425
return next(index for index in indexes if index.name == name)
452426
except StopIteration:
@@ -455,7 +429,7 @@ def _retrieve_from_system_indexes(self, name: str) -> ContextGroundingIndex:
455429
async def _retrieve_from_system_indexes_async(
456430
self, name: str
457431
) -> ContextGroundingIndex:
458-
indexes = await self.retrieve_system_indexes_async(name=name)
432+
indexes = await self._retrieve_system_indexes_async(name=name)
459433
try:
460434
return next(index for index in indexes if index.name == name)
461435
except StopIteration:

packages/uipath-platform/tests/services/test_context_grounding_service.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -790,7 +790,7 @@ def test_retrieve_system_indexes(
790790
},
791791
)
792792

793-
indexes = service.retrieve_system_indexes()
793+
indexes = service._retrieve_system_indexes()
794794

795795
assert isinstance(indexes, list)
796796
assert len(indexes) == 2
@@ -806,7 +806,7 @@ def test_retrieve_system_indexes(
806806
assert HEADER_USER_AGENT in sent_requests[0].headers
807807
assert (
808808
sent_requests[0].headers[HEADER_USER_AGENT]
809-
== f"UiPath.Python.Sdk/UiPath.Python.Sdk.Activities.ContextGroundingService.retrieve_system_indexes/{version}"
809+
== f"UiPath.Python.Sdk/UiPath.Python.Sdk.Activities.ContextGroundingService._retrieve_system_indexes/{version}"
810810
)
811811

812812
def test_retrieve_system_indexes_with_name_filter(
@@ -831,7 +831,7 @@ def test_retrieve_system_indexes_with_name_filter(
831831
},
832832
)
833833

834-
indexes = service.retrieve_system_indexes(name="system-template-index")
834+
indexes = service._retrieve_system_indexes(name="system-template-index")
835835

836836
assert len(indexes) == 1
837837
assert indexes[0].name == "system-template-index"
@@ -864,7 +864,7 @@ async def test_retrieve_system_indexes_async(
864864
},
865865
)
866866

867-
indexes = await service.retrieve_system_indexes_async()
867+
indexes = await service._retrieve_system_indexes_async()
868868

869869
assert len(indexes) == 1
870870
assert indexes[0].id == "sys-index-1"
@@ -876,7 +876,7 @@ async def test_retrieve_system_indexes_async(
876876

877877
assert (
878878
sent_requests[0].headers[HEADER_USER_AGENT]
879-
== f"UiPath.Python.Sdk/UiPath.Python.Sdk.Activities.ContextGroundingService.retrieve_system_indexes_async/{version}"
879+
== f"UiPath.Python.Sdk/UiPath.Python.Sdk.Activities.ContextGroundingService._retrieve_system_indexes_async/{version}"
880880
)
881881

882882
def test_retrieve_system_indexes_escapes_single_quote_in_name(
@@ -901,7 +901,7 @@ def test_retrieve_system_indexes_escapes_single_quote_in_name(
901901
},
902902
)
903903

904-
indexes = service.retrieve_system_indexes(name="O'Brien")
904+
indexes = service._retrieve_system_indexes(name="O'Brien")
905905

906906
assert len(indexes) == 1
907907
assert indexes[0].name == "O'Brien"
@@ -947,7 +947,7 @@ def test_retrieve_system_indexes_empty(
947947
json={"value": []},
948948
)
949949

950-
indexes = service.retrieve_system_indexes()
950+
indexes = service._retrieve_system_indexes()
951951

952952
assert indexes == []
953953

0 commit comments

Comments
 (0)