Add option to log guardrail sequence run info.

Author: qiuosier

ads/llm/chain.py

@@ -24,6 +24,7 @@
 logger = logging.getLogger(__name__)
 SPEC_CHAIN_TYPE = "_type"
 SPEC_CHAIN = "chain"
+LOG_ADS_GUARDRAIL_INFO = "LOG_ADS_GUARDRAIL_INFO"
 
 
 class GuardrailSequence(RunnableSequence):
@@ -46,6 +47,11 @@ class GuardrailSequence(RunnableSequence):
     When this is ``True``, the ``BlockedByGuardrail`` exception from the guardrail will be raised.
     """
 
+    log_info: bool = False
+    """Indicate whether to print the run info at the end of each invocation.
+    This option can also be turned on if the environment variable LOG_ADS_GUARDRAIL_INFO is set to "1".
+    """
+
     @property
     def steps(self) -> List[Runnable[Any, Any]]:
         """Steps in the sequence."""
@@ -158,6 +164,9 @@ def run(self, input: Any, num_generations: int = 1, **kwargs) -> GuardrailIO:
                 raise ex
             obj.data = [ex.message]
             obj.info.append(ex.info)
+        if self.log_info or os.environ.get(LOG_ADS_GUARDRAIL_INFO) == "1":
+            # LOG_ADS_GUARDRAIL_INFO is set to "1" in score.py by default.
+            print(obj.dict())
         return obj
 
     def _save_to_file(self, chain_dict, filename, overwrite=False):

ads/llm/guardrails/huggingface.py

@@ -17,6 +17,7 @@ class HuggingFaceEvaluation(Guardrail):
 
     @root_validator(skip_on_failure=True)
     def load_model(cls, values):
+        """Loads the model from Huggingface."""
         if values.get("path"):
             path = values["path"]
         else: