fix: vs escalation tool configs (#556)

Author: radu-mocanu

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.5.41"
+version = "0.5.42"
 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.8.10,<2.9.0",
+    "uipath>=2.8.12,<2.9.0",
     "uipath-runtime>=0.7.1, <0.8.0",
     "langgraph>=1.0.0, <2.0.0",
     "langchain-core>=1.2.5, <2.0.0",

src/uipath_langchain/agent/tools/escalation_tool.py

@@ -15,10 +15,7 @@
     AgentEscalationResourceConfig,
     AssetRecipient,
     StandardRecipient,
-    TaskTitle,
-    TextBuilderTaskTitle,
 )
-from uipath.agent.utils.text_tokens import build_string_from_tokens
 from uipath.eval.mocks import mockable
 from uipath.platform import UiPath
 from uipath.platform.action_center.tasks import TaskRecipient, TaskRecipientType
@@ -36,7 +33,11 @@
 from ..exceptions import AgentTerminationException
 from ..react.types import AgentGraphState
 from .tool_node import ToolWrapperReturnType
-from .utils import sanitize_dict_for_serialization, sanitize_tool_name
+from .utils import (
+    resolve_task_title,
+    sanitize_dict_for_serialization,
+    sanitize_tool_name,
+)
 
 
 class EscalationAction(str, Enum):
@@ -86,19 +87,6 @@ async def resolve_asset(asset_name: str, folder_path: str) -> str | None:
         ) from e
 
 
-def _resolve_task_title(
-    task_title: TaskTitle | str | None, agent_input: dict[str, Any]
-) -> str:
-    """Resolve task title based on channel configuration."""
-    if isinstance(task_title, TextBuilderTaskTitle):
-        return build_string_from_tokens(task_title.tokens, agent_input)
-
-    if isinstance(task_title, str):
-        return task_title
-
-    return "Escalation Task"
-
-
 def _get_user_email(user: Any) -> str | None:
     """Extract email from user object/dict."""
     if user is None:
@@ -253,8 +241,10 @@ async def escalation_wrapper(
         if tool.metadata is None:
             raise RuntimeError("Tool metadata is required for task_title resolution")
 
-        tool.metadata["task_title"] = _resolve_task_title(
-            channel.task_title, sanitize_dict_for_serialization(dict(state))
+        tool.metadata["task_title"] = resolve_task_title(
+            channel.task_title,
+            sanitize_dict_for_serialization(dict(state)),
+            default_title="Escalation Task",
         )
 
         tool.metadata["_call_id"] = call.get("id")

src/uipath_langchain/agent/tools/ixp_escalation_tool.py

@@ -1,13 +1,17 @@
 """Ixp escalation tool."""
 
+from typing import Any
+
 from langchain.tools import BaseTool
 from langchain_core.messages import ToolCall
 from langchain_core.tools import StructuredTool
+from langgraph.func import task
 from langgraph.types import interrupt
 from pydantic import BaseModel
 from uipath.agent.models.agent import AgentIxpVsEscalationResourceConfig
 from uipath.eval.mocks import mockable
-from uipath.platform.common import DocumentExtractionValidation
+from uipath.platform import UiPath
+from uipath.platform.common import WaitDocumentExtractionValidation
 from uipath.platform.documents import (
     ActionPriority,
     ExtractionResponseIXP,
@@ -21,7 +25,11 @@
 )
 
 from .structured_tool_with_output_type import StructuredToolWithOutputType
-from .utils import sanitize_tool_name
+from .utils import (
+    resolve_task_title,
+    sanitize_dict_for_serialization,
+    sanitize_tool_name,
+)
 
 
 class StructuredToolWithWrapper(StructuredToolWithOutputType, ToolWrapperMixin):
@@ -31,19 +39,14 @@ class StructuredToolWithWrapper(StructuredToolWithOutputType, ToolWrapperMixin):
 def create_ixp_escalation_tool(
     resource: AgentIxpVsEscalationResourceConfig,
 ) -> StructuredTool:
-    """Uses interrupt() to suspend graph execution until data is extracted (handled by runtime)."""
+    """Uses interrupt() to suspend graph execution until extraction is validated."""
     tool_name: str = sanitize_tool_name(resource.name)
     storage_bucket_name: str = resource.vs_escalation_properties.storage_bucket_name
     storage_bucket_folder_path: str = (
         resource.vs_escalation_properties.storage_bucket_folder_path
     )
-    action_priority: ActionPriority = (
-        resource.vs_escalation_properties.action_priority or ActionPriority.MEDIUM
-    )
-    action_title: str = (
-        resource.vs_escalation_properties.action_title or "VS Escalation Task"
-    )
-
+    channel = resource.channels[0]
+    action_priority = ActionPriority.from_str(channel.priority)
     ixp_tool_name: str = resource.vs_escalation_properties.ixp_tool_id
 
     class OutputSchema(BaseModel):
@@ -59,14 +62,26 @@ class OutputSchema(BaseModel):
     async def ixp_escalation_tool(
         extraction_result: ExtractionResponseIXP,
     ) -> OutputSchema:
-        response = interrupt(
-            DocumentExtractionValidation(
+        task_title = "VS Escalation Task"
+        if tool.metadata is not None:
+            task_title = tool.metadata.get("task_title") or task_title
+
+        @task
+        async def start_extraction_validation() -> Any:
+            client = UiPath()
+            return await client.documents.start_ixp_extraction_validation_async(
                 extraction_response=extraction_result,
-                action_title=action_title,
+                action_title=task_title,
                 storage_bucket_name=storage_bucket_name,
-                action_folder=storage_bucket_folder_path,
+                storage_bucket_directory_path=storage_bucket_folder_path,
                 action_priority=action_priority,
             )
+
+        validation_response = await start_extraction_validation()
+        response = interrupt(
+            WaitDocumentExtractionValidation(
+                extraction_validation=validation_response,
+            )
         )
         return OutputSchema(data=response["dataProjection"])
 
@@ -75,6 +90,15 @@ async def ixp_escalation_tool_wrapper(
         call: ToolCall,
         state: AgentGraphState,
     ) -> ToolWrapperReturnType:
+        if tool.metadata is None:
+            raise RuntimeError("Tool metadata is required for task_title resolution")
+
+        tool.metadata["task_title"] = resolve_task_title(
+            channel.task_title,
+            sanitize_dict_for_serialization(dict(state)),
+            default_title="VS Escalation Task",
+        )
+
         extraction_result = state.inner_state.tools_storage.get(ixp_tool_name)
         if not extraction_result:
             raise RuntimeError(
@@ -90,6 +114,13 @@ async def ixp_escalation_tool_wrapper(
         args_schema={},
         coroutine=ixp_escalation_tool,
         output_type=OutputSchema,
+        metadata={
+            "tool_type": "vs_escalation",
+            "display_name": channel.properties.app_name,
+            "channel_type": channel.type,
+            "ixp_tool_id": ixp_tool_name,
+            "storage_bucket_name": storage_bucket_name,
+        },
     )
     tool.set_tool_wrappers(awrapper=ixp_escalation_tool_wrapper)
 

src/uipath_langchain/agent/tools/utils.py

@@ -3,6 +3,9 @@
 import re
 from typing import Any
 
+from uipath.agent.models.agent import TaskTitle, TextBuilderTaskTitle
+from uipath.agent.utils.text_tokens import build_string_from_tokens
+
 
 def sanitize_tool_name(name: str) -> str:
     """Sanitize tool name for LLM compatibility (alphanumeric, underscore, hyphen only, max 64 chars)."""
@@ -40,3 +43,18 @@ def sanitize_dict_for_serialization(args: dict[str, Any]) -> dict[str, Any]:
         else:
             converted_args[key] = value
     return converted_args
+
+
+def resolve_task_title(
+    task_title: TaskTitle | str | None,
+    agent_input: dict[str, Any],
+    default_title: str = "Escalation Task",
+) -> str:
+    """Resolve task title based on channel configuration."""
+    if isinstance(task_title, TextBuilderTaskTitle):
+        return build_string_from_tokens(task_title.tokens, agent_input)
+
+    if isinstance(task_title, str):
+        return task_title
+
+    return default_title