From 5865f6a0e146e7e0e65d34414582e364291c401f Mon Sep 17 00:00:00 2001 From: Fedor Date: Mon, 4 Nov 2024 19:19:45 +0300 Subject: [PATCH] Modified safety settings for google_ai --- crates/language_model/src/request.rs | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/crates/language_model/src/request.rs b/crates/language_model/src/request.rs index 06dde1862ab37..76657657819ee 100644 --- a/crates/language_model/src/request.rs +++ b/crates/language_model/src/request.rs @@ -294,7 +294,10 @@ impl LanguageModelRequest { top_p: None, top_k: None, }), - safety_settings: None, + safety_settings: Some(vec![google_ai::SafetySetting { + category: google_ai::HarmCategory::DangerousContent, + threshold: google_ai::HarmBlockThreshold::BlockNone, + }]), } }