diff --git a/pyproject.toml b/pyproject.toml
index 0906d948f..dd335d787 100644
--- a/pyproject.toml
+++ b/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"
diff --git a/src/uipath/agent/models/agent.py b/src/uipath/agent/models/agent.py
index 5222c0510..3bed821ac 100644
--- a/src/uipath/agent/models/agent.py
+++ b/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
 
diff --git a/uv.lock b/uv.lock
index a91189bb0..6702ecef7 100644
--- a/uv.lock
+++ b/uv.lock
@@ -2486,7 +2486,7 @@ wheels = [
 
 [[package]]
 name = "uipath"
-version = "2.5.1"
+version = "2.5.2"
 source = { editable = "." }
 dependencies = [
     { name = "applicationinsights" },
@@ -2555,7 +2555,7 @@ requires-dist = [
     { name = "tenacity", specifier = ">=9.0.0" },
     { name = "truststore", specifier = ">=0.10.1" },
     { name = "uipath-core", specifier = ">=0.1.6,<0.2.0" },
-    { name = "uipath-runtime", specifier = ">=0.5,<0.6.0" },
+    { name = "uipath-runtime", specifier = ">=0.5.0,<0.6.0" },
 ]
 
 [package.metadata.requires-dev]