Adds safety setting

2024-08-15 14:44:59 +01:00
parent 60c20958a4
commit cb954ea5db
3 changed files with 111 additions and 0 deletions
--- a/src/client.rs
+++ b/src/client.rs
@@ -159,6 +159,7 @@ impl<T: TokenProvider + Clone> GeminiClient<T> {
            generation_config: None,
            tools: None,
            system_instruction: None,
+            safety_settings: None,
        };

        let response = self.generate_content(&request, model).await?;
@@ -187,6 +188,7 @@ impl<T: TokenProvider + Clone> GeminiClient<T> {
            generation_config: generation_config.cloned(),
            tools: None,
            system_instruction: None,
+            safety_settings: None,
        };

        let response = self.generate_content(&request, "gemini-pro").await?;