yetessam commited on
Commit
0fadd7f
·
verified ·
1 Parent(s): ba32940

Update tools/polite_guard.py

Browse files
Files changed (1) hide show
  1. tools/polite_guard.py +3 -3
tools/polite_guard.py CHANGED
@@ -22,9 +22,9 @@ class PoliteGuardTool(Tool):
22
  output_type = "any"
23
 
24
  def forward(self, input_text: Any) -> Any:
25
- self.label, self.score = self.ask_polite_guard(input_text)
26
- print(f"forward sets the following: {self.label } and {self.score}")
27
- return self.label
28
 
29
  def __init__(self, *args, **kwargs):
30
  self.is_initialized = False
 
22
  output_type = "any"
23
 
24
  def forward(self, input_text: Any) -> Any:
25
+ str_return_value = self.ask_polite_guard(input_text)
26
+ print(f"forward sets the following: {str_return_value}")
27
+ return str_return_value
28
 
29
  def __init__(self, *args, **kwargs):
30
  self.is_initialized = False