summaryrefslogtreecommitdiffstats
path: root/src/rag/mod.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-28 06:24:20 +0800
committerGitHub <noreply@github.com>2024-06-28 06:24:20 +0800
commit4fbbbd2d991b37ac04b77151ef862de9649bbfec (patch)
tree7e4343fb19b8d105b39ad6137d23944d8bffce60 /src/rag/mod.rs
parent10bd71297db11c163f95625080d956469a1d8689 (diff)
downloadaichat-4fbbbd2d991b37ac04b77151ef862de9649bbfec.tar.gz
feat: `.file`/`--file` support URLs (#665)
Diffstat (limited to 'src/rag/mod.rs')
-rw-r--r--src/rag/mod.rs4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/rag/mod.rs b/src/rag/mod.rs
index 536092f..ae208d2 100644
--- a/src/rag/mod.rs
+++ b/src/rag/mod.rs
@@ -59,7 +59,7 @@ impl Rag {
paths = add_document_paths()?;
};
debug!("doc paths: {paths:?}");
- let loaders = config.read().rag_document_loaders.clone();
+ let loaders = config.read().document_loaders.clone();
let spinner = create_spinner("Starting").await;
tokio::select! {
ret = rag.add_paths(loaders, &paths, Some(spinner.clone())) => {
@@ -641,7 +641,7 @@ fn add_document_paths() -> Result<Vec<String>> {
.with_validator(required!("This field is required"))
.with_help_message("e.g. file;dir/;dir/**/*.md;url;sites/**")
.prompt()?;
- let paths = text.split(';').map(|v| v.to_string()).collect();
+ let paths = text.split(';').map(|v| v.trim().to_string()).collect();
Ok(paths)
}