overwrite to_dict on ValidatorReference

Author: CalebCourier

guardrails/classes/validation/validator_reference.py

@@ -1,5 +1,8 @@
+from typing import Any, Dict
 from guardrails_api_client import ValidatorReference as IValidatorReference
 
+from guardrails.utils.serialization_utils import serialize
+
 
 # Docs only
 class ValidatorReference(IValidatorReference):
@@ -17,3 +20,15 @@ class ValidatorReference(IValidatorReference):
         args (Optional[List[Any]]): Positional arguments. Default None.
         kwargs (Optional[Dict[str, Any]]): Keyword arguments. Default None.
     """
+
+    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]
+
+        if self.kwargs:
+            ref_dict["kwargs"] = {k: serialize(v) for k, v in self.kwargs.items()}
+
+        return ref_dict

guardrails/guard.py

@@ -22,7 +22,6 @@
 
 from guardrails_api_client import (
     Guard as IGuard,
-    ValidatorReference,
     ValidatePayload,
     SimpleTypes,
     ValidationOutcome as IValidationOutcome,
@@ -36,6 +35,7 @@
 from guardrails.classes.rc import RC
 from guardrails.classes.validation.validation_result import ErrorSpan
 from guardrails.classes.validation.validation_summary import ValidationSummary
+from guardrails.classes.validation.validator_reference import ValidatorReference
 from guardrails.classes.validation_outcome import ValidationOutcome
 from guardrails.classes.execution import GuardExecutionOptions
 from guardrails.classes.generic import Stack