From 08593d9381a0313f058b364a93b3dfb8ef1b7710 Mon Sep 17 00:00:00 2001
From: thedragonsinn <98635854+thedragonsinn@users.noreply.github.com>
Date: Sun, 9 Feb 2025 10:05:54 +0530
Subject: [PATCH] `ai`: update safety settings

---
 app/plugins/ai/models.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/app/plugins/ai/models.py b/app/plugins/ai/models.py
index 0f0c3c8..7b8ee74 100644
--- a/app/plugins/ai/models.py
+++ b/app/plugins/ai/models.py
@@ -37,10 +37,12 @@ class Settings:
         max_output_tokens=4000,
 
         safety_settings=[
-            SafetySetting(category="HARM_CATEGORY_HARASSMENT", threshold="BLOCK_NONE"),
+            SafetySetting(category="HARM_CATEGORY_UNSPECIFIED", threshold="BLOCK_NONE"),
             SafetySetting(category="HARM_CATEGORY_HATE_SPEECH", threshold="BLOCK_NONE"),
-            SafetySetting(category="HARM_CATEGORY_SEXUALLY_EXPLICIT", threshold="BLOCK_NONE"),
             SafetySetting(category="HARM_CATEGORY_DANGEROUS_CONTENT", threshold="BLOCK_NONE"),
+            SafetySetting(category="HARM_CATEGORY_HARASSMENT", threshold="BLOCK_NONE"),
+            SafetySetting(category="HARM_CATEGORY_SEXUALLY_EXPLICIT", threshold="BLOCK_NONE"),
+            SafetySetting(category="HARM_CATEGORY_CIVIC_INTEGRITY", threshold="BLOCK_NONE"),
         ],
         # fmt:on