ContentAgent

Sleeping

yetessam commited on Feb 14

Commit

9dce3ef

verified ·

1 Parent(s): 8643744

Update tools/polite_guard.py

Files changed (1) hide show

tools/polite_guard.py CHANGED Viewed

@@ -15,7 +15,7 @@ class PoliteGuardTool(Tool):
     def forward(self, input_text: Any) -> Any:
         self.label, self.score = self.ask_polite_guard(input_text)
     def __init__(self, *args, **kwargs):
         self.is_initialized = False

     def forward(self, input_text: Any) -> Any:
         self.label, self.score = self.ask_polite_guard(input_text)
+        return self.label
     def __init__(self, *args, **kwargs):
         self.is_initialized = False