feat: mypy for all type check (#10921)
This commit is contained in:
@@ -20,6 +20,8 @@ class OpenAIModeration(Moderation):
|
||||
def moderation_for_inputs(self, inputs: dict, query: str = "") -> ModerationInputsResult:
|
||||
flagged = False
|
||||
preset_response = ""
|
||||
if self.config is None:
|
||||
raise ValueError("The config is not set.")
|
||||
|
||||
if self.config["inputs_config"]["enabled"]:
|
||||
preset_response = self.config["inputs_config"]["preset_response"]
|
||||
@@ -35,6 +37,8 @@ class OpenAIModeration(Moderation):
|
||||
def moderation_for_outputs(self, text: str) -> ModerationOutputsResult:
|
||||
flagged = False
|
||||
preset_response = ""
|
||||
if self.config is None:
|
||||
raise ValueError("The config is not set.")
|
||||
|
||||
if self.config["outputs_config"]["enabled"]:
|
||||
flagged = self._is_violated({"text": text})
|
||||
|
||||
Reference in New Issue
Block a user