diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-13 07:41:29 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-13 07:41:29 +0800 |
| commit | 255b194bcc538b2557caa60c5b607b4c4bfc0abd (patch) | |
| tree | 2a2995f7a33df2d3024f489de4435a06e94dcef8 /src/config/mod.rs | |
| parent | 64982b4510e38153885bfd0a78c250110b3e03c5 (diff) | |
| download | aichat-255b194bcc538b2557caa60c5b607b4c4bfc0abd.tar.gz | |
feat: add `.starter` repl command (#594)
Diffstat (limited to 'src/config/mod.rs')
| -rw-r--r-- | src/config/mod.rs | 18 |
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