feat: support customizing gemini safeSettings (#375)

1 files changed, 4 insertions, 1 deletions

diff --git a/src/client/gemini.rs b/src/client/gemini.rs
index 60c1e3b..39371fd 100644
--- a/src/client/gemini.rs
+++ b/src/client/gemini.rs
@@ -23,6 +23,7 @@ const TOKENS_COUNT_FACTORS: TokensCountFactors = (5, 2);
 pub struct GeminiConfig {
     pub name: Option<String>,
     pub api_key: Option<String>,
+    pub block_threshold: Option<String>,
     pub extra: Option<ExtraConfig>,
 }
 
@@ -73,7 +74,9 @@ impl GeminiClient {
             false => "generateContent",
         };
 
-        let body = build_body(data, self.model.name.clone())?;
+        let block_threshold = self.config.block_threshold.clone();
+
+        let body = build_body(data, self.model.name.clone(), block_threshold)?;
 
         let model = self.model.name.clone();