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