Temperature config update for Gemini redacter

abdolence · Aug 7, 2024 · a8cd479 · a8cd479
1 parent 8f70a3d
commit a8cd479
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/redacters/gemini_llm.rs b/src/redacters/gemini_llm.rs
@@ -127,6 +127,8 @@ impl<'a> GeminiLlmRedacter<'a> {
                         generation_config: Some(
                             gcloud_sdk::google::ai::generativelanguage::v1beta::GenerationConfig {
                                 candidate_count: Some(1),
+                                temperature: Some(0.2),
+                                stop_sequences: vec![generate_random_text_separator.clone()],
                                 ..std::default::Default::default()
                             },
                         ),