From 2539e24fe9eadf69cb20b01d2303a99fa2e8a363 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 8 Mar 2023 17:13:11 +0800 Subject: chore: improve code by getting prompt from config (#39) --- src/main.rs | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'src/main.rs') diff --git a/src/main.rs b/src/main.rs index edb9907..8c24945 100644 --- a/src/main.rs +++ b/src/main.rs @@ -71,10 +71,9 @@ fn start_directive( input: &str, no_stream: bool, ) -> Result<()> { - let prompt = config.borrow().get_prompt(); let highlight = config.borrow().highlight && stdout().is_terminal(); let output = if no_stream { - let output = client.send_message(input, prompt)?; + let output = client.send_message(input)?; if highlight { let mut markdown_render = MarkdownRender::new(); println!("{}\n", markdown_render.render(&output).trim_end()); @@ -90,7 +89,7 @@ fn start_directive( abort_clone.set_ctrlc(); }) .expect("Error setting Ctrl-C handler"); - let output = render_stream(input, None, &client, highlight, false, abort, wg.clone())?; + let output = render_stream(input, &client, highlight, false, abort, wg.clone())?; wg.wait(); output }; -- cgit v1.2.3