From 9aa24296063d38b059d5d9183717d8c2e4aa0724 Mon Sep 17 00:00:00 2001 From: Chris Raethke Date: Mon, 27 Nov 2023 11:53:19 +1000 Subject: [PATCH] Add examples for set_initial_prompt(...) --- examples/full_usage/src/main.rs | 1 + src/whisper_params.rs | 7 +++++++ 2 files changed, 8 insertions(+) diff --git a/examples/full_usage/src/main.rs b/examples/full_usage/src/main.rs index 5826a5b..eee6658 100644 --- a/examples/full_usage/src/main.rs +++ b/examples/full_usage/src/main.rs @@ -51,6 +51,7 @@ fn main() { ).expect("failed to open model"); let mut state = ctx.create_state().expect("failed to create key"); let mut params = FullParams::new(SamplingStrategy::default()); + params.set_initial_prompt("experience"); params.set_progress_callback_safe(|progress| println!("Progress callback: {}%", progress)); let st = std::time::Instant::now(); diff --git a/src/whisper_params.rs b/src/whisper_params.rs index 795cada..37281cf 100644 --- a/src/whisper_params.rs +++ b/src/whisper_params.rs @@ -590,6 +590,13 @@ impl<'a, 'b> FullParams<'a, 'b> { /// /// # Panics /// This method will panic if `initial_prompt` contains a null byte, as it cannot be converted into a `CString`. + /// + /// # Examples + /// ``` + /// let mut params = FullParams::new(SamplingStrategy::default()); + /// params.set_initial_prompt("Hello, world!"); + /// // ... further usage of params ... + /// ``` pub fn set_initial_prompt(&mut self, initial_prompt: &str) { self.fp.initial_prompt = CString::new(initial_prompt) .expect("Initial prompt contains null byte")