Update gen/gemini_chat.py

gen/gemini_chat.py

@@ -83,19 +83,19 @@ def _default_safety_settings():
     return [
         {
             "category": "HARM_CATEGORY_HARASSMENT",
-            "threshold": "NEGLIGIBLE"
+            "threshold": "BLOCK_NONE"
         },
         {
             "category": "HARM_CATEGORY_HATE_SPEECH",
-            "threshold": "NEGLIGIBLE"
+            "threshold": "BLOCK_NONE"
         },
         {
             "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-            "threshold": "NEGLIGIBLE"
+            "threshold": "BLOCK_NONE"
         },
         {
             "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-            "threshold": "NEGLIGIBLE"
+            "threshold": "BLOCK_NONE"
         },
     ]
 
@@ -113,7 +113,7 @@ async def gen_text(
 ):
     model = genai.GenerativeModel(model_name="gemini-1.0-pro",
                                 generation_config=gen_config,
-                                safety_settings=None)#safety_settings)
+                                safety_settings=safety_settings)
     
     user_prompt = prompts[-1]
     prompts = prompts[:-1]
@@ -124,7 +124,6 @@ async def gen_text(
     )
 
     async for resp in resps:
-        print(resp.candidate.safety_ratings)
         async for word in _word_generator(resp.text):
             yield word + " "