added extra safety filter option

app/api_helpers.py

@@ -122,7 +122,12 @@ def create_generation_config(request: OpenAIRequest) -> Dict[str, Any]:
             types.SafetySetting(category="HARM_CATEGORY_DANGEROUS_CONTENT", threshold=safety_threshold),
             types.SafetySetting(category="HARM_CATEGORY_SEXUALLY_EXPLICIT", threshold=safety_threshold),
             types.SafetySetting(category="HARM_CATEGORY_HARASSMENT", threshold=safety_threshold),
-            types.SafetySetting(category="HARM_CATEGORY_CIVIC_INTEGRITY", threshold=safety_threshold)
+            types.SafetySetting(category="HARM_CATEGORY_CIVIC_INTEGRITY", threshold=safety_threshold),
+            types.SafetySetting(category="HARM_CATEGORY_UNSPECIFIED", threshold=safety_threshold),
+            types.SafetySetting(category="HARM_CATEGORY_IMAGE_HATE", threshold=safety_threshold),
+            types.SafetySetting(category="HARM_CATEGORY_IMAGE_DANGEROUS_CONTENT", threshold=safety_threshold),
+            types.SafetySetting(category="HARM_CATEGORY_IMAGE_HARASSMENT", threshold=safety_threshold),
+            types.SafetySetting(category="HARM_CATEGORY_IMAGE_SEXUALLY_EXPLICIT", threshold=safety_threshold),
     ]
     # config["thinking_config"] = {"include_thoughts": True}
 

app/openai_handler.py

@@ -148,7 +148,12 @@ class OpenAIDirectHandler:
             {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": safety_threshold},
             {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": safety_threshold},
             {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": safety_threshold},
-            {"category": 'HARM_CATEGORY_CIVIC_INTEGRITY', "threshold": safety_threshold}
+            {"category": 'HARM_CATEGORY_CIVIC_INTEGRITY', "threshold": safety_threshold},
+            {"category": "HARM_CATEGORY_UNSPECIFIED", "threshold": safety_threshold},
+            {"category": "HARM_CATEGORY_IMAGE_HATE", "threshold": safety_threshold},
+            {"category": "HARM_CATEGORY_IMAGE_DANGEROUS_CONTENT", "threshold": safety_threshold},
+            {"category": "HARM_CATEGORY_IMAGE_HARASSMENT", "threshold": safety_threshold},
+            {"category": 'HARM_CATEGORY_IMAGE_SEXUALLY_EXPLICIT', "threshold": safety_threshold}
         ]
 
     def create_openai_client(self, project_id: str, gcp_token: str, location: str = "global") -> openai.AsyncOpenAI:

app/requirements.txt

@@ -3,7 +3,7 @@ uvicorn==0.27.1
 google-auth==2.38.0
 google-cloud-aiplatform==1.86.0
 pydantic==2.6.1
-google-genai==1.17.0
+google-genai==1.24.0
 httpx[socks]>=0.25.0
 openai
 google-auth-oauthlib