From d21ff78727fe5641667db4c51fc08186e70cad1e Mon Sep 17 00:00:00 2001
From: h <h@kotikot.com>
Date: Tue, 12 Aug 2025 20:10:06 +0300
Subject: [PATCH] fix(solaris): proper building google safety settings

---
 src/dependencies/providers/model.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/dependencies/providers/model.py b/src/dependencies/providers/model.py
index 239b4bc..2857247 100644
--- a/src/dependencies/providers/model.py
+++ b/src/dependencies/providers/model.py
@@ -12,12 +12,22 @@ from pydantic_ai.providers.google import GoogleProvider
 
 from utils.db.models import DynamicConfig, RespondSession
 from utils.env import env
+from utils.logging import console
 
 from ..types.model import RespondModel, ReviewModel
 
 GOOGLE_SAFETY_SETTINGS = [
-    SafetySettingDict(category=category, threshold=HarmBlockThreshold.OFF)
-    for category in HarmCategory
+    {
+        "category": HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+        "threshold": HarmBlockThreshold.OFF,
+    }
+    for category in [
+        HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+        HarmCategory.HARM_CATEGORY_HARASSMENT,
+        HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY,
+    ]
 ]