From 5d763fc10c4e3d40268ead473ccb0000b4923bd1 Mon Sep 17 00:00:00 2001 From: sigoden Date: Sat, 20 Apr 2024 09:21:52 +0800 Subject: feat: add `.prompt` repl command (#420) --- src/repl/mod.rs | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'src/repl/mod.rs') diff --git a/src/repl/mod.rs b/src/repl/mod.rs index 3f30317..564bdf9 100644 --- a/src/repl/mod.rs +++ b/src/repl/mod.rs @@ -25,10 +25,15 @@ use std::{env, process}; const MENU_NAME: &str = "completion_menu"; lazy_static! { - static ref REPL_COMMANDS: [ReplCommand; 15] = [ + static ref REPL_COMMANDS: [ReplCommand; 16] = [ ReplCommand::new(".help", "Print this help message", State::all()), ReplCommand::new(".info", "Print system info", State::all()), ReplCommand::new(".model", "Switch LLM model", State::all()), + ReplCommand::new( + ".prompt", + "Use a temp role with this prompt", + State::able_change_role() + ), ReplCommand::new(".role", "Use a role", State::able_change_role()), ReplCommand::new(".info role", "Show the role info", State::in_role(),), ReplCommand::new(".exit role", "Leave current role", State::in_role(),), @@ -169,6 +174,12 @@ impl Repl { } None => println!("Usage: .model "), }, + ".prompt" => match args { + Some(text) => { + self.config.write().set_prompt(text)?; + } + None => println!("Usage: .prompt ..."), + }, ".role" => match args { Some(args) => match args.split_once(|c| c == '\n' || c == ' ') { Some((name, text)) => { @@ -181,7 +192,7 @@ impl Repl { self.config.write().set_role(args)?; } }, - None => println!(r#"Usage: .role [text...]"#), + None => println!(r#"Usage: .role [text]..."#), }, ".session" => { self.config.write().start_session(args)?; -- cgit v1.2.3