fix: add AgentAllFieldsSelector dto

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.5.1"
+version = "2.5.2"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath/agent/models/agent.py

@@ -16,7 +16,7 @@
 from uipath.core.guardrails import (
     BaseGuardrail,
     FieldReference,
-    FieldSelector,
+    SpecificFieldsSelector,
     UniversalRule,
 )
 
@@ -655,13 +655,27 @@ class AgentWordRule(BaseModel):
     """Word rule model."""
 
     rule_type: Literal["word"] = Field(alias="$ruleType")
-    field_selector: FieldSelector = Field(alias="fieldSelector")
+    field_selector: AgentFieldSelector = Field(alias="fieldSelector")
     operator: AgentWordOperator
     value: str | None = None
 
     model_config = ConfigDict(populate_by_name=True, extra="allow")
 
 
+class AgentAllFieldsSelector(BaseModel):
+    """All fields selector."""
+
+    selector_type: Literal["all"] = Field(alias="$selectorType")
+
+    model_config = ConfigDict(populate_by_name=True, extra="allow")
+
+
+AgentFieldSelector = Annotated[
+    AgentAllFieldsSelector | SpecificFieldsSelector,
+    Field(discriminator="selector_type"),
+]
+
+
 class AgentNumberOperator(str, Enum):
     """Number operator enumeration."""
 
@@ -677,7 +691,7 @@ class AgentNumberRule(BaseModel):
     """Number rule model."""
 
     rule_type: Literal["number"] = Field(alias="$ruleType")
-    field_selector: FieldSelector = Field(alias="fieldSelector")
+    field_selector: AgentFieldSelector = Field(alias="fieldSelector")
     operator: AgentNumberOperator
     value: float
 
@@ -694,7 +708,7 @@ class AgentBooleanRule(BaseModel):
     """Boolean rule model."""
 
     rule_type: Literal["boolean"] = Field(alias="$ruleType")
-    field_selector: FieldSelector = Field(alias="fieldSelector")
+    field_selector: AgentFieldSelector = Field(alias="fieldSelector")
     operator: AgentBooleanOperator
     value: bool