format

Author: infwinston

fastchat/utils.py

@@ -175,7 +175,15 @@ def oai_moderation(text, custom_thresholds=None):
 
 def moderation_filter(text, model_list, do_moderation=False):
     # Apply moderation for below models
-    MODEL_KEYWORDS = ["claude", "gpt", "bard", "gemini", "mistral-large", "command-r", "dbrx"]
+    MODEL_KEYWORDS = [
+        "claude",
+        "gpt",
+        "bard",
+        "gemini",
+        "mistral-large",
+        "command-r",
+        "dbrx",
+    ]
 
     custom_thresholds = {"sexual": 0.3}
     # set a stricter threshold for claude