summaryrefslogtreecommitdiffstats
path: root/src/config/mod.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-13 07:41:29 +0800
committerGitHub <noreply@github.com>2024-06-13 07:41:29 +0800
commit255b194bcc538b2557caa60c5b607b4c4bfc0abd (patch)
tree2a2995f7a33df2d3024f489de4435a06e94dcef8 /src/config/mod.rs
parent64982b4510e38153885bfd0a78c250110b3e03c5 (diff)
downloadaichat-255b194bcc538b2557caa60c5b607b4c4bfc0abd.tar.gz
feat: add `.starter` repl command (#594)
Diffstat (limited to 'src/config/mod.rs')
-rw-r--r--src/config/mod.rs18
1 files changed, 17 insertions, 1 deletions
diff --git a/src/config/mod.rs b/src/config/mod.rs
index 7e5d6cf..b1eb41f 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -918,7 +918,15 @@ impl Config {
if let Some(bot) = &self.bot {
bot.export()
} else {
- bail!("No rag")
+ bail!("No bot")
+ }
+ }
+
+ pub fn bot_banner(&self) -> Result<String> {
+ if let Some(bot) = &self.bot {
+ Ok(bot.banner())
+ } else {
+ bail!("No bot")
}
}
@@ -1023,6 +1031,14 @@ impl Config {
.collect(),
".rag" => self.list_rags().into_iter().map(|v| (v, None)).collect(),
".bot" => list_bots().into_iter().map(|v| (v, None)).collect(),
+ ".starter" => match &self.bot {
+ Some(bot) => bot
+ .converstaion_staters()
+ .iter()
+ .map(|v| (v.clone(), None))
+ .collect(),
+ None => vec![],
+ },
".set" => vec![
"max_output_tokens",
"temperature",