feat: add ArgumentEmail and ArgumentGroupName escalation recipient types

Author: dushyant-uipath

packages/uipath/src/uipath/agent/models/agent.py

@@ -135,6 +135,8 @@ class AgentEscalationRecipientType(str, CaseInsensitiveEnum):
     ASSET_USER_EMAIL = "AssetUserEmail"
     GROUP_NAME = "GroupName"
     ASSET_GROUP_NAME = "AssetGroupName"
+    ARGUMENT_EMAIL = "ArgumentEmail"
+    ARGUMENT_GROUP_NAME = "ArgumentGroupName"
 
 
 class AgentContextRetrievalMode(str, CaseInsensitiveEnum):
@@ -481,6 +483,8 @@ class AgentA2aResourceConfig(BaseAgentResourceConfig):
     5: AgentEscalationRecipientType.GROUP_NAME,
     "staticgroupname": AgentEscalationRecipientType.GROUP_NAME,
     6: AgentEscalationRecipientType.ASSET_GROUP_NAME,
+    7: AgentEscalationRecipientType.ARGUMENT_EMAIL,
+    8: AgentEscalationRecipientType.ARGUMENT_GROUP_NAME,
 }
 
 
@@ -536,8 +540,37 @@ class AssetRecipient(BaseEscalationRecipient):
     folder_path: str = Field(..., alias="folderPath")
 
 
+class ArgumentEmailRecipient(BaseEscalationRecipient):
+    """Argument email recipient resolved from a named input argument.
+
+    The argument_path supports dot-notation for nested input fields (e.g. "user.email").
+    """
+
+    type: Literal[AgentEscalationRecipientType.ARGUMENT_EMAIL,] = Field(
+        ..., alias="type"
+    )
+    argument_path: str = Field(..., alias="argumentName")
+
+
+class ArgumentGroupNameRecipient(BaseEscalationRecipient):
+    """Argument group name recipient resolved from a named input argument.
+
+    The argument_path supports dot-notation for nested input fields (e.g. "team.groupName").
+    """
+
+    type: Literal[AgentEscalationRecipientType.ARGUMENT_GROUP_NAME,] = Field(
+        ..., alias="type"
+    )
+    argument_path: str = Field(..., alias="argumentName")
+
+
 AgentEscalationRecipient = Annotated[
-    Union[StandardRecipient, AssetRecipient],
+    Union[
+        StandardRecipient,
+        AssetRecipient,
+        ArgumentEmailRecipient,
+        ArgumentGroupNameRecipient,
+    ],
     Field(discriminator="type"),
     BeforeValidator(_normalize_recipient_type),
 ]

packages/uipath/tests/agent/models/test_agent.py

@@ -1,7 +1,7 @@
 from typing import Any
 
 import pytest
-from pydantic import TypeAdapter
+from pydantic import TypeAdapter, ValidationError
 
 from uipath.agent.models.agent import (
     AgentA2aResourceConfig,
@@ -43,6 +43,8 @@
     AgentUnknownToolResourceConfig,
     AgentWordOperator,
     AgentWordRule,
+    ArgumentEmailRecipient,
+    ArgumentGroupNameRecipient,
     AssetRecipient,
     BatchTransformFileExtension,
     BatchTransformWebSearchGrounding,
@@ -3789,3 +3791,49 @@ def test_a2a_resource_case_insensitive(self):
         ]
         assert len(a2a_resources) == 1
         assert isinstance(a2a_resources[0], AgentA2aResourceConfig)
+
+
+class TestArgumentRecipientDeserialization:
+    def test_argument_email_recipient_by_type_int(self):
+        payload = {"type": 7, "argumentName": "assigneeEmail"}
+        recipient: AgentEscalationRecipient = TypeAdapter(
+            AgentEscalationRecipient
+        ).validate_python(payload)
+        assert isinstance(recipient, ArgumentEmailRecipient)
+        assert recipient.argument_path == "assigneeEmail"
+        assert recipient.type == AgentEscalationRecipientType.ARGUMENT_EMAIL
+
+    def test_argument_group_name_recipient_by_type_int(self):
+        payload = {"type": 8, "argumentName": "assigneeGroup"}
+        recipient: AgentEscalationRecipient = TypeAdapter(
+            AgentEscalationRecipient
+        ).validate_python(payload)
+        assert isinstance(recipient, ArgumentGroupNameRecipient)
+        assert recipient.argument_path == "assigneeGroup"
+        assert recipient.type == AgentEscalationRecipientType.ARGUMENT_GROUP_NAME
+
+    def test_argument_email_recipient_by_type_string(self):
+        payload = {"type": "ArgumentEmail", "argumentName": "emailArg"}
+        recipient: AgentEscalationRecipient = TypeAdapter(
+            AgentEscalationRecipient
+        ).validate_python(payload)
+        assert isinstance(recipient, ArgumentEmailRecipient)
+        assert recipient.argument_path == "emailArg"
+
+    def test_argument_group_name_recipient_by_type_string(self):
+        payload = {"type": "ArgumentGroupName", "argumentName": "groupArg"}
+        recipient: AgentEscalationRecipient = TypeAdapter(
+            AgentEscalationRecipient
+        ).validate_python(payload)
+        assert isinstance(recipient, ArgumentGroupNameRecipient)
+        assert recipient.argument_path == "groupArg"
+
+    def test_argument_email_recipient_missing_argument_name_raises(self):
+        payload = {"type": 7}
+        with pytest.raises(ValidationError):
+            TypeAdapter(AgentEscalationRecipient).validate_python(payload)
+
+    def test_argument_group_name_recipient_missing_argument_name_raises(self):
+        payload = {"type": 8}
+        with pytest.raises(ValidationError):
+            TypeAdapter(AgentEscalationRecipient).validate_python(payload)