.

confident-ai · Jan 10, 2025 · f359df1 · f359df1
1 parent c8abb7a
commit f359df1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/deepeval/guardrails/base_guard.py b/deepeval/guardrails/base_guard.py
@@ -10,6 +10,7 @@ class BaseDecorativeGuard(ABC):
     success: Optional[bool] = None
     evaluation_model: Optional[str] = None
     error: Optional[str] = None
+    latency: Optional[float] = None
     guard_type: GuardType
 
     @property

diff --git a/deepeval/guardrails/guardrails.py b/deepeval/guardrails/guardrails.py
@@ -56,7 +56,7 @@ def guard_input(self, input: str):
                     "Access denied: You need Enterprise access on Confident AI to use deepeval's guardrails."
                 )
 
-    def guard_response(self, input: str, response: str):
+    def guard_output(self, input: str, response: str):
         if len(self.guards) == 0:
             raise TypeError(
                 "Guardrails cannot guard LLM responses when no guards are provided."