diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-27 12:30:09 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-27 12:30:09 +0800 |
| commit | f82524fd154a1bf4a7277db868147afdb0cd507a (patch) | |
| tree | 2fd588377d63b47b4ae1a051a0e363a9737f3f8f /src/rag/mod.rs | |
| parent | 33a0049a81103c1f6c5639b9cec43cb14a9ef367 (diff) | |
| download | aichat-f82524fd154a1bf4a7277db868147afdb0cd507a.tar.gz | |
feat: implement native rag url loader (#660)
Diffstat (limited to 'src/rag/mod.rs')
| -rw-r--r-- | src/rag/mod.rs | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/rag/mod.rs b/src/rag/mod.rs index f01fc0c..9e37e6f 100644 --- a/src/rag/mod.rs +++ b/src/rag/mod.rs @@ -237,7 +237,7 @@ impl Rag { if let Some(path) = path.strip_suffix("**") { new_paths.push((path.to_string(), RECURSIVE_URL_LOADER.into())); } else { - new_paths.push((path.to_string(), "url".into())) + new_paths.push((path.to_string(), URL_LOADER.into())) } } else { let path = Path::new(path); @@ -275,6 +275,7 @@ impl Rag { for (index, (path, loader_name)) in new_paths.into_iter().enumerate() { println!("Loading {path} [{}/{new_paths_len}]", index + 1); let documents = load(&loaders, &path, &loader_name) + .await .with_context(|| format!("Failed to load '{path}'"))?; let separator = get_separators(&loader_name); let splitter = RecursiveCharacterTextSplitter::new( |
