feat: add task recipient

Author: dianapirvulescu

pyproject.toml

@@ -5,7 +5,7 @@ description = "Python SDK that enables developers to build and deploy LangGraph
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
-    "uipath>=2.5.22,<2.6.0",
+    "uipath>=2.5.23,<2.6.0",
     "uipath-runtime>=0.5.1,<0.6.0",
     "langgraph>=1.0.0, <2.0.0",
     "langchain-core>=1.2.5, <2.0.0",

src/uipath_langchain/agent/guardrails/actions/escalate_action.py

@@ -9,7 +9,6 @@
 from langgraph.types import Command, interrupt
 from uipath._utils import UiPathUrl
 from uipath.agent.models.agent import AgentEscalationRecipient
-from uipath.platform.action_center.tasks import TaskRecipient
 from uipath.platform.common import CreateEscalation, UiPathConfig
 from uipath.platform.guardrails import (
     BaseGuardrail,
@@ -79,7 +78,7 @@ async def _node(
             state: AgentGuardrailsGraphState,
         ) -> Dict[str, Any] | Command[Any]:
             # Resolve recipient value (handles both StandardRecipient and AssetRecipient)
-            assignee = await resolve_recipient_value(self.recipient)
+            task_recipient = await resolve_recipient_value(self.recipient)
 
             # Validate message count based on execution stage
             _validate_message_count(state, execution_stage)
@@ -141,7 +140,7 @@ async def _node(
                     app_folder_path=self.app_folder_path,
                     title="Agents Guardrail Task",
                     data=data,
-                    recipient=self.recipient,
+                    recipient=task_recipient,
                 )
             )
 

src/uipath_langchain/agent/guardrails/guardrails_factory.py

@@ -32,8 +32,6 @@
     UniversalRule,
     WordRule,
 )
-
-from uipath.platform.action_center.tasks import TaskRecipient
 from uipath.platform.guardrails import BaseGuardrail, GuardrailScope
 
 from uipath_langchain.agent.guardrails.actions import (

src/uipath_langchain/agent/tools/escalation_tool.py

@@ -9,12 +9,14 @@
 from uipath.agent.models.agent import (
     AgentEscalationChannel,
     AgentEscalationRecipient,
+    AgentEscalationRecipientType,
     AgentEscalationResourceConfig,
     AssetRecipient,
     StandardRecipient,
 )
 from uipath.eval.mocks import mockable
 from uipath.platform import UiPath
+from uipath.platform.action_center.tasks import TaskRecipient, TaskRecipientType
 from uipath.platform.common import CreateEscalation
 from uipath.runtime.errors import UiPathErrorCode
 
@@ -32,13 +34,24 @@ class EscalationAction(str, Enum):
     END = "end"
 
 
-async def resolve_recipient_value(recipient: AgentEscalationRecipient) -> str | None:
+async def resolve_recipient_value(
+    recipient: AgentEscalationRecipient,
+) -> TaskRecipient | None:
     """Resolve recipient value based on recipient type."""
     if isinstance(recipient, AssetRecipient):
-        return await resolve_asset(recipient.asset_name, recipient.folder_path)
+        value = await resolve_asset(recipient.asset_name, recipient.folder_path)
+        type = None
+        if recipient.type == AgentEscalationRecipientType.ASSET_USER_EMAIL:
+            type = TaskRecipientType.EMAIL
+        elif recipient.type == AgentEscalationRecipientType.ASSET_GROUP_NAME:
+            type = TaskRecipientType.GROUP_NAME
+        return TaskRecipient(value=value, type=type)
 
     if isinstance(recipient, StandardRecipient):
-        return recipient.value
+        type = TaskRecipientType(recipient.type)
+        if recipient.type == AgentEscalationRecipientType.USER_EMAIL:
+            type = TaskRecipientType.EMAIL
+        return TaskRecipient(value=recipient.value, type=type)
 
     return None
 
@@ -86,21 +99,21 @@ async def create_escalation_tool(
     async def escalation_tool_fn(**kwargs: Any) -> dict[str, Any]:
         task_title = channel.task_title or "Escalation Task"
 
-        assignee: str | None = (
+        recipient: TaskRecipient | None = (
             await resolve_recipient_value(channel.recipients[0])
             if channel.recipients
             else None
         )
 
-        # Assignee requires runtime resolution, store in metadata after resolving
+        # Recipient requires runtime resolution, store in metadata after resolving
         if tool.metadata is not None:
-            tool.metadata["assignee"] = assignee
+            tool.metadata["recipient"] = recipient
 
         result = interrupt(
             CreateEscalation(
                 title=task_title,
                 data=kwargs,
-                assignee=assignee,
+                recipient=recipient,
                 app_name=channel.properties.app_name,
                 app_folder_path=channel.properties.folder_name,
                 app_version=channel.properties.app_version,