summaryrefslogtreecommitdiffstats
path: root/src/main.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2023-03-11 07:37:47 +0800
committerGitHub <noreply@github.com>2023-03-11 07:37:47 +0800
commitd48cd64162416c906451a0f81b71676acd185610 (patch)
treea394b2fcb9d47ecbbb95fbe27c68df3dfad546d4 /src/main.rs
parent52de7ec33a2ef598b85c02336fb27c194fcdf609 (diff)
downloadaichat-d48cd64162416c906451a0f81b71676acd185610.tar.gz
feat: provide `--prompt` for adding a prompt from cli (#62)
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/main.rs b/src/main.rs
index b198d14..5bcdf11 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -48,6 +48,9 @@ fn main() -> Result<()> {
if cli.no_highlight {
config.lock().highlight = false;
}
+ if let Some(prompt) = &cli.prompt {
+ config.lock().add_prompt(prompt)?;
+ }
let no_stream = cli.no_stream;
let client = ChatGptClient::init(config.clone())?;
if atty::isnt(atty::Stream::Stdin) {