feat: add asset recipient support for guardrails

Author: dianagrecu-uipath

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

@@ -8,6 +8,7 @@
 from langchain_core.messages import AIMessage, AnyMessage, BaseMessage, ToolMessage
 from langgraph.types import Command, interrupt
 from uipath._utils import UiPathUrl
+from uipath.agent.models.agent import AgentEscalationRecipient
 from uipath.platform.common import CreateEscalation, UiPathConfig
 from uipath.platform.guardrails import (
     BaseGuardrail,
@@ -17,6 +18,7 @@
 
 from ...exceptions import AgentTerminationException
 from ...react.types import AgentGuardrailsGraphState
+from ...tools.escalation_tool import resolve_recipient_value
 from ..types import ExecutionStage
 from ..utils import _extract_tool_args_from_message, get_message_content
 from .base_action import GuardrailAction, GuardrailActionNode
@@ -35,20 +37,20 @@ def __init__(
         app_name: str,
         app_folder_path: str,
         version: int,
-        assignee: str,
+        recipient: AgentEscalationRecipient,
     ):
         """Initialize EscalateAction with escalation app configuration.
 
         Args:
             app_name: Name of the escalation app.
             app_folder_path: Folder path where the escalation app is located.
             version: Version of the escalation app.
-            assignee: User or role assigned to handle the escalation.
+            recipient: Recipient object (StandardRecipient or AssetRecipient).
         """
         self.app_name = app_name
         self.app_folder_path = app_folder_path
         self.version = version
-        self.assignee = assignee
+        self.recipient = recipient
 
     def action_node(
         self,
@@ -74,6 +76,9 @@ def action_node(
         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)
+
             # Validate message count based on execution stage
             _validate_message_count(state, execution_stage)
 
@@ -134,7 +139,7 @@ async def _node(
                     app_folder_path=self.app_folder_path,
                     title="Agents Guardrail Task",
                     data=data,
-                    assignee=self.assignee,
+                    assignee=assignee,
                 )
             )
 

src/uipath_langchain/agent/guardrails/guardrails_factory.py

@@ -20,7 +20,6 @@
     AgentUnknownGuardrail,
     AgentWordOperator,
     AgentWordRule,
-    StandardRecipient,
 )
 from uipath.core.guardrails import (
     AllFieldsSelector,
@@ -432,18 +431,17 @@ def build_guardrails_with_actions(
                 )
             )
         elif isinstance(action, AgentGuardrailEscalateAction):
-            if isinstance(action.recipient, StandardRecipient):
-                result.append(
-                    (
-                        converted_guardrail,
-                        EscalateAction(
-                            app_name=action.app.name,
-                            app_folder_path=action.app.folder_name,
-                            version=action.app.version,
-                            assignee=action.recipient.value,
-                        ),
-                    )
+            result.append(
+                (
+                    converted_guardrail,
+                    EscalateAction(
+                        app_name=action.app.name,
+                        app_folder_path=action.app.folder_name,
+                        version=action.app.version,
+                        recipient=action.recipient,
+                    ),
                 )
+            )
         elif isinstance(action, AgentGuardrailFilterAction):
             result.append((converted_guardrail, FilterAction(fields=action.fields)))
     return result