fix(guardrails): allow guardrail enums to be deserialized with upper/lowercase first letter

Author: Valentina Bojan

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-core"
-version = "0.3.0"
+version = "0.3.1"
 description = "UiPath Core abstractions"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath/core/guardrails/guardrails.py

@@ -1,9 +1,18 @@
 """Guardrails models for UiPath Platform."""
 
 from enum import Enum
-from typing import Annotated, Callable, Literal
+from typing import Annotated, Any, Callable, Literal
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, field_validator
+
+
+def _decapitalize_first_letter(s: str) -> str:
+    """Convert first letter to lowercase (e.g., 'SimpleText' -> 'simpleText')."""
+    if not s or len(s) == 0:
+        return s
+    if len(s) == 1:
+        return s.lower()
+    return s[0].lower() + s[1:]
 
 
 class GuardrailValidationResultType(str, Enum):
@@ -56,6 +65,12 @@ class FieldReference(BaseModel):
 
     model_config = ConfigDict(populate_by_name=True, extra="allow")
 
+    @field_validator("source", mode="before")
+    @classmethod
+    def normalize_type(cls, v: Any) -> Any:
+        """Normalize type by decapitalizing first letter."""
+        return _decapitalize_first_letter(v) if isinstance(v, str) else v
+
 
 class SelectorType(str, Enum):
     """Selector type enumeration."""
@@ -72,6 +87,17 @@ class AllFieldsSelector(BaseModel):
 
     model_config = ConfigDict(populate_by_name=True, extra="allow")
 
+    @field_validator("sources", mode="before")
+    @classmethod
+    def normalize_sources(cls, v: Any) -> Any:
+        """Normalize sources by decapitalizing first letter of each item."""
+        if isinstance(v, list):
+            return [
+                _decapitalize_first_letter(item) if isinstance(item, str) else item
+                for item in v
+            ]
+        return v
+
 
 class SpecificFieldsSelector(BaseModel):
     """Specific fields selector."""
@@ -128,6 +154,12 @@ class UniversalRule(BaseModel):
 
     model_config = ConfigDict(populate_by_name=True, extra="allow")
 
+    @field_validator("apply_to", mode="before")
+    @classmethod
+    def normalize_type(cls, v: Any) -> Any:
+        """Normalize type by decapitalizing first letter."""
+        return _decapitalize_first_letter(v) if isinstance(v, str) else v
+
 
 class NumberRule(BaseModel):
     """Number rule model."""

tests/guardrails/test_guardrails_models.py

@@ -0,0 +1,47 @@
+"""Tests for guardrails models normalization validators."""
+
+from typing import TYPE_CHECKING
+
+from uipath.core.guardrails import (
+    AllFieldsSelector,
+    FieldReference,
+    FieldSource,
+)
+
+if TYPE_CHECKING:
+    pass
+
+
+class TestGuardrailsModelsNormalization:
+    """Test guardrails models field normalization."""
+
+    def test_field_reference_normalizes_capitalized_source(
+        self,
+    ) -> None:
+        """Test that FieldReference normalizes capitalized source values to lowercase."""
+        # Create FieldReference with capitalized "Input" - should normalize to FieldSource.INPUT
+        field_ref = FieldReference(path="testField", source="Input")
+        assert field_ref.source == FieldSource.INPUT
+
+        # Create FieldReference with capitalized "Output" - should normalize to FieldSource.OUTPUT
+        field_ref = FieldReference(path="testField", source="Output")
+        assert field_ref.source == FieldSource.OUTPUT
+
+        # Create FieldReference with lowercase "input" - should work as-is
+        field_ref = FieldReference(path="testField", source="input")
+        assert field_ref.source == FieldSource.INPUT
+
+    def test_all_fields_selector_normalizes_capitalized_sources(
+        self,
+    ) -> None:
+        """Test that AllFieldsSelector normalizes capitalized source values in the list."""
+        # Create AllFieldsSelector with capitalized "Input" and "Output" - should normalize
+        selector = AllFieldsSelector(selector_type="all", sources=["Input", "Output"])
+        assert FieldSource.INPUT in selector.sources
+        assert FieldSource.OUTPUT in selector.sources
+        assert len(selector.sources) == 2
+
+        # Create AllFieldsSelector with mixed case - should normalize all
+        selector = AllFieldsSelector(selector_type="all", sources=["Input", "output"])
+        assert FieldSource.INPUT in selector.sources
+        assert FieldSource.OUTPUT in selector.sources