From ccd9de7792df4b1a23288297fd07779aa4757782 Mon Sep 17 00:00:00 2001 From: Debanjum Date: Thu, 27 Mar 2025 13:48:45 +0530 Subject: [PATCH] Improve safety settings for Gemini chat models - Align remaining harm categories to only refuse in high harm scenarios as well - Handle response for new "negligible" harm probability as well --- src/khoj/processor/conversation/google/utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/khoj/processor/conversation/google/utils.py b/src/khoj/processor/conversation/google/utils.py index b224e2a0..a5b8b74b 100644 --- a/src/khoj/processor/conversation/google/utils.py +++ b/src/khoj/processor/conversation/google/utils.py @@ -52,6 +52,10 @@ SAFETY_SETTINGS = [ category=gtypes.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT, threshold=gtypes.HarmBlockThreshold.BLOCK_ONLY_HIGH, ), + gtypes.SafetySetting( + category=gtypes.HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY, + threshold=gtypes.HarmBlockThreshold.BLOCK_ONLY_HIGH, + ), ] @@ -246,6 +250,7 @@ def generate_safety_response(safety_ratings: list[gtypes.SafetyRating]): # Add a bit of variety to the discomfort level based on the safety rating probability discomfort_level = { gtypes.HarmProbability.HARM_PROBABILITY_UNSPECIFIED: " ", + gtypes.HarmProbability.NEGLIGIBLE: "a little ", gtypes.HarmProbability.LOW: "a bit ", gtypes.HarmProbability.MEDIUM: "moderately ", gtypes.HarmProbability.HIGH: random.choice(["very ", "quite ", "fairly "]),