feat: add compatibility for guardrails entitlements phase 1

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.4.0"
+version = "0.4.1"
 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/guardrail_nodes.py

@@ -100,10 +100,28 @@ def _create_validation_command(
     Returns:
         Command to update state and route to appropriate node.
     """
-    if not result.validation_passed:
+    from uipath.core.guardrails import GuardrailValidationResultType
+
+    # Handle new format: result has a 'result' attribute with GuardrailValidationResultType
+    if hasattr(result, "result"):
+        if result.result != GuardrailValidationResultType.PASSED:
+            return Command(
+                goto=failure_node, update={"guardrail_validation_result": result.reason}
+            )
+    # Handle old format: backwards compatibility for tests using validation_passed
+    elif hasattr(result, "validation_passed"):
+        if not result.validation_passed:
+            return Command(
+                goto=failure_node, update={"guardrail_validation_result": result.reason}
+            )
+    else:
+        # Fallback: assume failure if format is unknown
         return Command(
-            goto=failure_node, update={"guardrail_validation_result": result.reason}
+            goto=failure_node,
+            update={"guardrail_validation_result": "Unknown validation result format"},
         )
+
+    # Update guardail trace with skip/reason
     return Command(goto=success_node, update={"guardrail_validation_result": None})
 
 

tests/cli/conftest.py

@@ -1,7 +1,10 @@
 from unittest.mock import patch
 
 import pytest
-from uipath.core.guardrails import GuardrailValidationResult
+from uipath.core.guardrails import (
+    GuardrailValidationResult,
+    GuardrailValidationResultType,
+)
 
 
 @pytest.fixture
@@ -19,7 +22,10 @@ def mock_guardrails_service():
 
     def mock_evaluate_guardrail(text, guardrail):
         """Mock guardrail evaluation - always passes validation."""
-        return GuardrailValidationResult(validation_passed=True, reason="")
+        return GuardrailValidationResult(
+            result=GuardrailValidationResultType.PASSED,
+            reason="",
+        )
 
     with patch(
         "uipath.platform.guardrails.GuardrailsService.evaluate_guardrail",

tests/cli/test_agent_with_guardrails.py

@@ -23,7 +23,10 @@
 
 import pytest
 from langchain_core.messages import AIMessage
-from uipath.core.guardrails import GuardrailValidationResult
+from uipath.core.guardrails import (
+    GuardrailValidationResult,
+    GuardrailValidationResultType,
+)
 from uipath.runtime import (
     UiPathExecuteOptions,
     UiPathRuntimeContext,
@@ -288,12 +291,13 @@ def mock_evaluate_guardrail(text, guardrail):
                         and ".com" in text
                     ):
                         return GuardrailValidationResult(
-                            validation_passed=False,  # PII detected - should trigger block!
+                            result=GuardrailValidationResultType.VALIDATION_FAILED,
                             reason="PII detected in text",
                         )
                     else:
                         return GuardrailValidationResult(
-                            validation_passed=True, reason=""
+                            result=GuardrailValidationResultType.PASSED,
+                            reason="",
                         )
 
                 with (
@@ -422,12 +426,15 @@ def mock_evaluate_guardrail(text, guardrail):
                     # Prompt injection guardrail should detect and block
                     if guardrail.name == "Prompt injection guardrail":
                         return GuardrailValidationResult(
-                            validation_passed=False,
+                            result=GuardrailValidationResultType.VALIDATION_FAILED,
                             reason="Prompt injection detected",
                         )
 
                     # All other guardrails pass
-                    return GuardrailValidationResult(validation_passed=True, reason="")
+                    return GuardrailValidationResult(
+                        result=GuardrailValidationResultType.PASSED,
+                        reason="",
+                    )
 
                 # Mock LLM - should NOT be called if guardrail blocks at LLM level
                 async def mock_llm_invoke(*args, **kwargs):
@@ -867,12 +874,15 @@ def mock_evaluate_guardrail(text, guardrail):
                         and ".com" in text
                     ):
                         return GuardrailValidationResult(
-                            validation_passed=False,
+                            result=GuardrailValidationResultType.VALIDATION_FAILED,
                             reason="PII detected in tool input",
                         )
 
                     # All other guardrails pass
-                    return GuardrailValidationResult(validation_passed=True, reason="")
+                    return GuardrailValidationResult(
+                        result=GuardrailValidationResultType.PASSED,
+                        reason="",
+                    )
 
                 # Mock LLM responses - tool call contains an email address
                 async def mock_llm_invoke(*args, **kwargs):
@@ -1025,12 +1035,15 @@ def mock_evaluate_guardrail(text, guardrail):
                         and ".com" in text
                     ):
                         return GuardrailValidationResult(
-                            validation_passed=False,
+                            result=GuardrailValidationResultType.VALIDATION_FAILED,
                             reason="PII detected in LLM output",
                         )
 
                     # All other guardrails pass
-                    return GuardrailValidationResult(validation_passed=True, reason="")
+                    return GuardrailValidationResult(
+                        result=GuardrailValidationResultType.PASSED,
+                        reason="",
+                    )
 
                 # Mock LLM responses - LLM output contains PII in tool call args
                 async def mock_llm_invoke(*args, **kwargs):
@@ -1228,12 +1241,15 @@ def mock_evaluate_guardrail(text, guardrail):
                         and ".com" in text
                     ):
                         return GuardrailValidationResult(
-                            validation_passed=False,
+                            result=GuardrailValidationResultType.VALIDATION_FAILED,
                             reason="PII detected in LLM output",
                         )
 
                     # All other guardrails pass
-                    return GuardrailValidationResult(validation_passed=True, reason="")
+                    return GuardrailValidationResult(
+                        result=GuardrailValidationResultType.PASSED,
+                        reason="",
+                    )
 
                 # Mock LLM responses - LLM output contains PII in tool call args
                 async def mock_llm_invoke(*args, **kwargs):