Feat/modify pii masking endpoint (#891)

Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

Author: yashwagle1

pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "uipath-langchain"
-version = "0.11.12"
+version = "0.11.13"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
     "uipath>=2.10.74, <2.11.0",
     "uipath-core>=0.5.17, <0.6.0",
-    "uipath-platform>=0.1.59, <0.2.0",
+    "uipath-platform>=0.1.61, <0.2.0",
     "uipath-runtime>=0.11.0, <0.12.0",
     "langgraph>=1.1.8, <2.0.0",
     "langchain-core>=1.2.11, <2.0.0",

src/uipath_langchain/agent/tools/internal_tools/pii_masker.py

@@ -10,7 +10,7 @@
 
 from uipath.core.feature_flags import FeatureFlags
 from uipath.platform import UiPath
-from uipath.platform.semantic_proxy import (
+from uipath.platform.pii_detection import (
     PiiDetectionRequest,
     PiiDetectionResponse,
     PiiDocument,
@@ -94,7 +94,7 @@ async def apply(
             ],
             entity_thresholds=self._entity_thresholds_from_policy() or None,
         )
-        self._result = await self._client.semantic_proxy.detect_pii_async(request)
+        self._result = await self._client.pii_detection.detect_pii_async(request)
         logger.info(
             "PII detection completed: %d document entities, %d file entities",
             sum(len(d.pii_entities) for d in self._result.response),

tests/agent/tools/internal_tools/test_pii_masker.py

@@ -5,7 +5,7 @@
 
 import pytest
 from uipath.core.feature_flags import FeatureFlags
-from uipath.platform.semantic_proxy import (
+from uipath.platform.pii_detection import (
     PiiDetectionResponse,
     PiiDocumentResult,
     PiiEntity,
@@ -53,8 +53,8 @@ def _make_client(
     upload_result: uuid.UUID | None = None,
 ) -> Mock:
     client = Mock()
-    client.semantic_proxy = Mock()
-    client.semantic_proxy.detect_pii_async = AsyncMock(return_value=response)
+    client.pii_detection = Mock()
+    client.pii_detection.detect_pii_async = AsyncMock(return_value=response)
     client.jobs = Mock()
     client.jobs.create_attachment_async = AsyncMock(
         return_value=upload_result or uuid.uuid4()
@@ -241,7 +241,7 @@ async def test_masks_prompt_and_annotates_files(self, httpx_mock):
         assert masked_files[0].masked_attachment_id == str(uploaded_uuid)
         assert masked_files[0].attachment_id == "orig-uuid"
 
-        request = client.semantic_proxy.detect_pii_async.call_args[0][0]
+        request = client.pii_detection.detect_pii_async.call_args[0][0]
         assert request.files[0].file_name == "doc.pdf"
         assert request.files[0].file_type == "pdf"
 
@@ -315,7 +315,7 @@ async def test_passes_entity_thresholds_from_policy(self):
 
         await PiiMasker(client, policy).apply("original", [])
 
-        request = client.semantic_proxy.detect_pii_async.call_args[0][0]
+        request = client.pii_detection.detect_pii_async.call_args[0][0]
         assert request.entity_thresholds == [
             PiiEntityThreshold(category="Email", confidence_threshold=0.6)
         ]