summaryrefslogtreecommitdiffstats
path: root/src/main.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-07-12 17:58:04 +0800
committerGitHub <noreply@github.com>2024-07-12 17:58:04 +0800
commitcfcf624e3be72364b4c3123e4f48be559f1edb40 (patch)
tree98960e02299b5ef348c6bc7dab73b95945a26b7e /src/main.rs
parent5ca8975bbc20c94de0066a53a9b562e4a196c66e (diff)
downloadaichat-cfcf624e3be72364b4c3123e4f48be559f1edb40.tar.gz
refactor: `.starter` rendered in markdown format (#708)
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs10
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 {