From cce4fec108feb7c963c2394c4daf5bb2d0274b1b Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 3 Jul 2024 16:51:30 +0800 Subject: refactor: update config.example.yaml and readme, fix typos (#682) --- src/rag/loader.rs | 2 +- src/rag/mod.rs | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) (limited to 'src') diff --git a/src/rag/loader.rs b/src/rag/loader.rs index 37fa803..7fa1fd6 100644 --- a/src/rag/loader.rs +++ b/src/rag/loader.rs @@ -7,7 +7,7 @@ use std::{collections::HashMap, path::Path}; pub const EXTENSION_METADATA: &str = "__extension__"; pub const PATH_METADATA: &str = "__path__"; -pub async fn load_recrusive_url( +pub async fn load_recursive_url( loaders: &HashMap, path: &str, ) -> Result> { diff --git a/src/rag/mod.rs b/src/rag/mod.rs index e2f6ddd..3e7eda5 100644 --- a/src/rag/mod.rs +++ b/src/rag/mod.rs @@ -276,7 +276,7 @@ impl Rag { println!("Load {path} [{}/{paths_len}]", index + 1); if Self::is_url_path(path) { if let Some(path) = path.strip_suffix("**") { - files.extend(load_recrusive_url(&loaders, path).await?); + files.extend(load_recursive_url(&loaders, path).await?); } else { files.push(load_url(&loaders, path).await?); } @@ -326,11 +326,11 @@ impl Rag { "\npath: {path}\n{metadata}\n\n" )); let document = RagDocument::new(contents); - let splitted_documents = splitter.split_documents(&[document], &split_options); + let split_documents = splitter.split_documents(&[document], &split_options); rag_files.push(RagFile { hash: hash.clone(), path, - documents: splitted_documents, + documents: split_documents, }); } -- cgit v1.2.3