diff options
| author | sigoden <sigoden@gmail.com> | 2024-07-12 17:58:04 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-07-12 17:58:04 +0800 |
| commit | cfcf624e3be72364b4c3123e4f48be559f1edb40 (patch) | |
| tree | 98960e02299b5ef348c6bc7dab73b95945a26b7e /src/main.rs | |
| parent | 5ca8975bbc20c94de0066a53a9b562e4a196c66e (diff) | |
| download | aichat-cfcf624e3be72364b4c3123e4f48be559f1edb40.tar.gz | |
refactor: `.starter` rendered in markdown format (#708)
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/src/main.rs b/src/main.rs index 2583b73..13276a5 100644 --- a/src/main.rs +++ b/src/main.rs @@ -20,7 +20,7 @@ use crate::config::{ EXPLAIN_SHELL_ROLE, SHELL_ROLE, TEMP_SESSION_NAME, }; use crate::function::{eval_tool_calls, need_send_tool_results}; -use crate::render::{render_error, MarkdownRender}; +use crate::render::render_error; use crate::repl::Repl; use crate::utils::{ create_abort_signal, create_spinner, detect_shell, extract_block, run_command, AbortSignal, @@ -187,9 +187,7 @@ async fn start_directive( text.clone() }; if *IS_STDOUT_TERMINAL { - let render_options = config.read().render_options()?; - let mut markdown_render = MarkdownRender::init(render_options)?; - println!("{}", markdown_render.render(&text).trim()); + println!("{}", config.read().markdown_render(&text)?); } else { println!("{}", text); } @@ -242,10 +240,8 @@ async fn shell_execute(config: &GlobalConfig, shell: &Shell, mut input: Input) - config .write() .after_chat_completion(&input, &eval_str, &[])?; - let render_options = config.read().render_options()?; - let mut markdown_render = MarkdownRender::init(render_options)?; if config.read().dry_run { - println!("{}", markdown_render.render(&eval_str).trim()); + println!("{}", config.read().markdown_render(&eval_str)?); return Ok(()); } if *IS_STDOUT_TERMINAL { |
