참고소스 수정본

참고/guardrails-main/guardrails/actions/refrain.py

@@ -0,0 +1,43 @@
+from typing import Any, Dict, List, Union
+from guardrails.classes.output_type import OutputTypes
+from guardrails.logger import logger
+
+
+class Refrain:
+    pass
+
+
+def check_for_refrain(value: Union[List, Dict]) -> bool:
+    if isinstance(value, Refrain):
+        return True
+    elif isinstance(value, list):
+        for item in value:
+            if check_for_refrain(item):
+                return True
+    elif isinstance(value, dict):
+        for key, child in value.items():
+            if check_for_refrain(child):
+                return True
+
+    return False
+
+
+# Could be a generic instead of Any
+def apply_refrain(value: Any, output_type: OutputTypes) -> Any:
+    """Recursively check for any values that are instances of Refrain.
+
+    If found, return an empty value of the appropriate type.
+    """
+    refrain_value = {}
+    if output_type == OutputTypes.STRING:
+        refrain_value = ""
+    elif output_type == OutputTypes.LIST:
+        refrain_value = []
+
+    if check_for_refrain(value):
+        # If the data contains a `Refain` value, we return an empty
+        # value.
+        logger.debug("Refrain detected.")
+        value = refrain_value
+
+    return value