From 4fbbbd2d991b37ac04b77151ef862de9649bbfec Mon Sep 17 00:00:00 2001 From: sigoden Date: Fri, 28 Jun 2024 06:24:20 +0800 Subject: feat: `.file`/`--file` support URLs (#665) --- src/rag/mod.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/rag/mod.rs') 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> { .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) } -- cgit v1.2.3