override to_dict on ValidatorReference to serialize args and kwargs

Author: CalebCourier

guardrails/classes/schema/processed_schema.py

@@ -1,8 +1,8 @@
 from dataclasses import dataclass, field
 from typing import Any, Dict, List
-from guardrails_api_client import ValidatorReference
 from guardrails.classes.execution.guard_execution_options import GuardExecutionOptions
 from guardrails.classes.output_type import OutputTypes
+from guardrails.classes.validation.validator_reference import ValidatorReference
 from guardrails.types.validator import ValidatorMap
 
 

guardrails/classes/validation/validator_reference.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict
 from guardrails_api_client import ValidatorReference as IValidatorReference
 
-from guardrails.utils.serialization_utils import serialize
+from guardrails.utils.serialization_utils import to_dict
 
 
 # Docs only
@@ -21,14 +21,25 @@ class ValidatorReference(IValidatorReference):
         kwargs (Optional[Dict[str, Any]]): Keyword arguments. Default None.
     """
 
+    @classmethod
+    def from_interface(cls, interface: IValidatorReference) -> "ValidatorReference":
+        """Create a ValidatorReference from an interface."""
+        return cls(
+            id=interface.id,
+            on=interface.on,
+            on_fail=interface.on_fail,  # type: ignore
+            args=interface.args,
+            kwargs=interface.kwargs,
+        )
+
     def to_dict(self) -> Dict[str, Any]:
         ref_dict = super().to_dict()
 
         # serialize args and kwargs
         if self.args:
-            ref_dict["args"] = [serialize(a) for a in self.args]
+            ref_dict["args"] = [to_dict(a) for a in self.args]
 
         if self.kwargs:
-            ref_dict["kwargs"] = {k: serialize(v) for k, v in self.kwargs.items()}
+            ref_dict["kwargs"] = {k: to_dict(v) for k, v in self.kwargs.items()}
 
         return ref_dict

guardrails/guard.py

@@ -166,7 +166,7 @@ def __init__(
             id=id,
             name=name,
             description=description,
-            validators=validators,
+            validators=[],
             output_schema=model_schema,
             history=history,  # type: ignore - pyright doesn't understand pydantic overrides
         )
@@ -180,6 +180,9 @@ def __init__(
         # self.output_schema: Optional[ModelSchema] = None
         # self.history = history
 
+        ### Overrides ###
+        self.validators = validators
+
         ### Legacy ##
         self._num_reasks = None
         self._rail: Optional[str] = None
@@ -208,7 +211,10 @@ def __init__(
                 if loaded_guard:
                     self.id = loaded_guard.id
                     self.description = loaded_guard.description
-                    self.validators = loaded_guard.validators or []
+                    self.validators = [  # type: ignore
+                        ValidatorReference.from_interface(v)
+                        for v in loaded_guard.validators or []
+                    ]
 
                     loaded_output_schema = (
                         ModelSchema.from_dict(  # trims out extra keys
@@ -1376,7 +1382,7 @@ def to_dict(self) -> Dict[str, Any]:
             id=self.id,
             name=self.name,
             description=self.description,
-            validators=self.validators,
+            validators=self.validators,  # type: ignore
             output_schema=self.output_schema,
             history=[c.to_interface() for c in self.history],  # type: ignore
         )
@@ -1416,7 +1422,10 @@ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional["Guard"]:
             id=i_guard.id,
             name=i_guard.name,
             description=i_guard.description,
-            validators=i_guard.validators,
+            validators=[
+                ValidatorReference.from_interface(i_val)
+                for i_val in i_guard.validators or []
+            ],
             output_schema=output_schema,
         )
 

guardrails/schema/primitive_schema.py

@@ -3,10 +3,10 @@
 from guardrails_api_client.models.model_schema import ModelSchema
 from guardrails_api_client.models.simple_types import SimpleTypes
 from guardrails_api_client.models.validation_type import ValidationType
-from guardrails_api_client.models.validator_reference import ValidatorReference
 
 from guardrails.classes.output_type import OutputTypes
 from guardrails.classes.schema.processed_schema import ProcessedSchema
+from guardrails.classes.validation.validator_reference import ValidatorReference
 from guardrails.validator_base import Validator
 
 

guardrails/schema/pydantic_schema.py

@@ -13,9 +13,9 @@
 
 from pydantic import AliasChoices, AliasGenerator, AliasPath, BaseModel
 from pydantic.fields import FieldInfo
-from guardrails_api_client import ValidatorReference
 from guardrails.classes.output_type import OutputTypes
 from guardrails.classes.schema.processed_schema import ProcessedSchema
+from guardrails.classes.validation.validator_reference import ValidatorReference
 from guardrails.logger import logger
 from guardrails.types import (
     ModelOrListOfModels,

guardrails/schema/rail_schema.py

@@ -7,10 +7,11 @@
 from lxml import etree as ET
 from lxml.etree import _Element, Element, SubElement, XMLParser
 from xml.etree.ElementTree import canonicalize
-from guardrails_api_client import ModelSchema, SimpleTypes, ValidatorReference
+from guardrails_api_client import ModelSchema, SimpleTypes
 from guardrails.classes.execution.guard_execution_options import GuardExecutionOptions
 from guardrails.classes.output_type import OutputTypes
 from guardrails.classes.schema.processed_schema import ProcessedSchema
+from guardrails.classes.validation.validator_reference import ValidatorReference
 from guardrails.logger import logger
 from guardrails.types import RailTypes
 from guardrails.types.validator import ValidatorMap

guardrails/utils/serialization_utils.py

@@ -2,10 +2,29 @@
 import json
 from typing import Any, Optional
 import warnings
+from dataclasses import asdict, is_dataclass
+from pydantic import BaseModel
 
 from guardrails.classes.generic.default_json_encoder import DefaultJSONEncoder
 
 
+# This is the same logic as the DefaultJSONEncoder but without stringifying everything
+def to_dict(o):
+    if hasattr(o, "to_dict"):
+        return o.to_dict()
+    elif isinstance(o, BaseModel):
+        return o.model_dump()
+    elif is_dataclass(o):
+        return asdict(o)
+    elif isinstance(o, set):
+        return list(o)
+    elif isinstance(o, datetime):
+        return o.isoformat()
+    elif hasattr(o, "__dict__"):
+        return o.__dict__
+    return o
+
+
 # TODO: What other common cases we should consider?
 def serialize(val: Any) -> Optional[str]:
     try:

tests/integration_tests/schema/test_primitive_schema.py

@@ -1,6 +1,6 @@
 import json
 
-from guardrails_api_client.models.validator_reference import ValidatorReference
+from guardrails.classes.validation.validator_reference import ValidatorReference
 from guardrails.classes.schema.processed_schema import ProcessedSchema
 from guardrails.schema.primitive_schema import primitive_to_schema
 from guardrails.classes.output_type import OutputTypes

tests/integration_tests/schema/test_pydantic_schema.py

@@ -1,6 +1,6 @@
 import json
 
-from guardrails_api_client.models.validator_reference import ValidatorReference
+from guardrails.classes.validation.validator_reference import ValidatorReference
 from guardrails.classes.schema.processed_schema import ProcessedSchema
 from guardrails.schema.pydantic_schema import pydantic_model_to_schema
 from guardrails.classes.output_type import OutputTypes

tests/integration_tests/schema/test_rail_schema.py

@@ -3,7 +3,7 @@
 
 from xml.etree.ElementTree import canonicalize
 
-from guardrails_api_client.models.validator_reference import ValidatorReference
+from guardrails.classes.validation.validator_reference import ValidatorReference
 from guardrails.classes.schema.processed_schema import ProcessedSchema
 from guardrails.schema.rail_schema import (
     rail_file_to_schema,