Selaa lähdekoodia

Fix/input-value-type-in-moderation (#10893)

-LAN- 5 kuukautta sitten
vanhempi
commit
1be8365684
1 muutettua tiedostoa jossa 5 lisäystä ja 2 poistoa
  1. 5 2
      api/core/moderation/keywords/keywords.py

+ 5 - 2
api/core/moderation/keywords/keywords.py

@@ -1,3 +1,6 @@
+from collections.abc import Sequence
+from typing import Any
+
 from core.moderation.base import Moderation, ModerationAction, ModerationInputsResult, ModerationOutputsResult
 
 
@@ -62,5 +65,5 @@ class KeywordsModeration(Moderation):
     def _is_violated(self, inputs: dict, keywords_list: list) -> bool:
         return any(self._check_keywords_in_value(keywords_list, value) for value in inputs.values())
 
-    def _check_keywords_in_value(self, keywords_list, value) -> bool:
-        return any(keyword.lower() in value.lower() for keyword in keywords_list)
+    def _check_keywords_in_value(self, keywords_list: Sequence[str], value: Any) -> bool:
+        return any(keyword.lower() in str(value).lower() for keyword in keywords_list)