Add examples for set_initial_prompt(...)

This commit is contained in:
Chris Raethke 2023-11-27 11:53:19 +10:00
parent 8c8a5d78d9
commit 9aa2429606
2 changed files with 8 additions and 0 deletions

View file

@ -51,6 +51,7 @@ fn main() {
).expect("failed to open model"); ).expect("failed to open model");
let mut state = ctx.create_state().expect("failed to create key"); let mut state = ctx.create_state().expect("failed to create key");
let mut params = FullParams::new(SamplingStrategy::default()); let mut params = FullParams::new(SamplingStrategy::default());
params.set_initial_prompt("experience");
params.set_progress_callback_safe(|progress| println!("Progress callback: {}%", progress)); params.set_progress_callback_safe(|progress| println!("Progress callback: {}%", progress));
let st = std::time::Instant::now(); let st = std::time::Instant::now();

View file

@ -590,6 +590,13 @@ impl<'a, 'b> FullParams<'a, 'b> {
/// ///
/// # Panics /// # Panics
/// This method will panic if `initial_prompt` contains a null byte, as it cannot be converted into a `CString`. /// 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) { pub fn set_initial_prompt(&mut self, initial_prompt: &str) {
self.fp.initial_prompt = CString::new(initial_prompt) self.fp.initial_prompt = CString::new(initial_prompt)
.expect("Initial prompt contains null byte") .expect("Initial prompt contains null byte")