summaryrefslogtreecommitdiffstats
path: root/src/rag
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-13 19:41:54 +0800
committerGitHub <noreply@github.com>2024-06-13 19:41:54 +0800
commitf9847475b8eae99a70f16dc63bac67da2632474a (patch)
tree032c9eb02540cf8644a2df103d8a45dfc54557be /src/rag
parent255b194bcc538b2557caa60c5b607b4c4bfc0abd (diff)
downloadaichat-f9847475b8eae99a70f16dc63bac67da2632474a.tar.gz
feat: add rag and bot related cli options (#595)
Diffstat (limited to 'src/rag')
-rw-r--r--src/rag/mod.rs3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/rag/mod.rs b/src/rag/mod.rs
index f88e80e..c98bab7 100644
--- a/src/rag/mod.rs
+++ b/src/rag/mod.rs
@@ -50,6 +50,9 @@ impl Rag {
doc_paths: &[String],
abort_signal: AbortSignal,
) -> Result<Self> {
+ if !*IS_STDOUT_TERMINAL {
+ bail!("An interactive shell is required to initialize rag.")
+ }
debug!("init rag: {name}");
let model = select_embedding_model(config)?;
let chunk_size = set_chunk_size(&model)?;