From 9e15a3409e7080f0cf06b6dcfc0077142e3e8972 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 4 Mar 2024 10:51:30 +0800
Subject: feat: reduce gemini blocking of unsafe content with safetySettings
 (#334)

---
 src/client/vertexai.rs | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/client/vertexai.rs')

diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs
index 50e4b21..94becdb 100644
--- a/src/client/vertexai.rs
+++ b/src/client/vertexai.rs
@@ -259,6 +259,12 @@ pub(crate) fn build_body(data: SendData, _model: String) -> Result<Value> {
 
     let mut body = json!({
         "contents": contents,
+        "safetySettings":[
+            {"category":"HARM_CATEGORY_HARASSMENT","threshold":"BLOCK_ONLY_HIGH"},
+            {"category":"HARM_CATEGORY_HATE_SPEECH","threshold":"BLOCK_ONLY_HIGH"},
+            {"category":"HARM_CATEGORY_SEXUALLY_EXPLICIT","threshold":"BLOCK_ONLY_HIGH"},
+            {"category":"HARM_CATEGORY_DANGEROUS_CONTENT","threshold":"BLOCK_ONLY_HIGH"}
+        ]
     });
 
     if let Some(temperature) = temperature {
-- 
cgit v1.2.3