feat(guardrails): add missing rule description for UniversalRule (#519)

Co-authored-by: Valentina Bojan <valentina.bojan@uipath.com>

Author: valentinabojan

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.5.24"
+version = "0.5.25"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

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

@@ -191,9 +191,13 @@ async def _node(
                 reviewed_outputs = escalation_result.data.get("ReviewedOutputs")
                 reason = escalation_result.data.get("Reason")
                 if reviewed_inputs:
-                    metadata["escalation_data"]["reviewed_inputs"] = reviewed_inputs
+                    metadata["escalation_data"]["reviewed_inputs"] = (
+                        _parse_reviewed_data(reviewed_inputs)
+                    )
                 if reviewed_outputs:
-                    metadata["escalation_data"]["reviewed_outputs"] = reviewed_outputs
+                    metadata["escalation_data"]["reviewed_outputs"] = (
+                        _parse_reviewed_data(reviewed_outputs)
+                    )
                 if reason:
                     metadata["escalation_data"]["reason"] = reason
 
@@ -226,6 +230,23 @@ async def _node(
         return node_name, _node
 
 
+def _parse_reviewed_data(value: Any) -> Any:
+    if not value:
+        return value
+
+    if isinstance(value, str):
+        try:
+            # Try to parse as JSON first (handles JSON strings like "abcd")
+            parsed = json.loads(value)
+            return parsed
+        except json.JSONDecodeError:
+            # If not valid JSON, return as-is (plain string)
+            return value
+
+    # Already parsed (dict/list), return as-is
+    return value
+
+
 def _validate_message_count(
     state: AgentGuardrailsGraphState,
     execution_stage: ExecutionStage,

src/uipath_langchain/agent/guardrails/guardrails_factory.py

@@ -368,6 +368,9 @@ def _convert_agent_rule_to_deterministic(
     """
     if isinstance(agent_rule, UniversalRule):
         # UniversalRule is already compatible
+        agent_rule.rule_description = (
+            f"Always enforce the guardrail on {agent_rule.apply_to.value}"
+        )
         return agent_rule
 
     if isinstance(agent_rule, AgentWordRule):

tests/agent/guardrails/test_guardrails_factory.py

@@ -32,11 +32,13 @@
 )
 from uipath.core.guardrails import (
     AllFieldsSelector,
+    ApplyTo,
     BooleanRule,
     DeterministicGuardrail,
     FieldSource,
     GuardrailSelector,
     NumberRule,
+    UniversalRule,
     WordRule,
 )
 
@@ -818,6 +820,32 @@ def test_convert_boolean_rule(self) -> None:
         assert result.detects_violation(True) is True
         assert result.detects_violation(False) is False
 
+    def test_convert_universal_rule_sets_rule_description(self) -> None:
+        """UniversalRule should have rule_description set based on apply_to value."""
+        universal_rule = UniversalRule(
+            rule_type="always",
+            apply_to=ApplyTo.INPUT_AND_OUTPUT,
+        )
+        # Create a minimal guardrail for testing
+        guardrail = AgentCustomGuardrail.model_validate(
+            {
+                "$guardrailType": "custom",
+                "id": "test-id",
+                "name": "test-guardrail",
+                "description": "Test",
+                "enabledForEvals": True,
+                "selector": {"$selectorType": "all"},
+                "rules": [],
+                "action": {"$actionType": "block", "reason": "test"},
+            }
+        )
+        result = _convert_agent_rule_to_deterministic(universal_rule, guardrail, [])
+
+        assert isinstance(result, UniversalRule)
+        assert (
+            result.rule_description == "Always enforce the guardrail on inputAndOutput"
+        )
+
     def test_unsupported_rule_type_raises_value_error(self) -> None:
         # Create a mock rule that's not a supported type
         invalid_rule = cast(AgentWordRule, types.SimpleNamespace())