diff options
| -rw-r--r-- | src/client/vertexai.rs | 6 |
1 files changed, 6 insertions, 0 deletions
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 { |