diff options
Diffstat (limited to 'internal/weblog_utilities.py')
| -rw-r--r-- | internal/weblog_utilities.py | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/internal/weblog_utilities.py b/internal/weblog_utilities.py index c307929c..ae354116 100644 --- a/internal/weblog_utilities.py +++ b/internal/weblog_utilities.py @@ -180,11 +180,16 @@ def check_comment_spam(post, comment): """ safety_settings = [ - {"category": "HARM_CATEGORY_DANGEROUS", "threshold": "BLOCK_NONE"}, - {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"}, - {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"}, - {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"}, - {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}, + genai.types.SafetySetting( + category=cat, + threshold=genai.types.HarmBlockThreshold.BLOCK_NONE, + ) + for cat in ( + genai.types.HarmCategory.HARM_CATEGORY_HARASSMENT, + genai.types.HarmCategory.HARM_CATEGORY_HATE_SPEECH, + genai.types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT, + genai.types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT, + ) ] response = client.models.generate_content( @@ -192,6 +197,6 @@ def check_comment_spam(post, comment): contents=prompt, config=genai.types.GenerateContentConfig(safety_settings=safety_settings), ) - result = response.text.strip() + result = (response.text or "").strip() return result.upper() == "Y" # Return True if spam, False otherwise |