From 48219d2808bafc88fa4fa4d6997aa9dc92e6f686 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 10 Jul 2024 10:48:02 +0800 Subject: refactor: fetch image as data url (#694) --- src/rag/loader.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/rag') diff --git a/src/rag/loader.rs b/src/rag/loader.rs index 0b0d935..c764d78 100644 --- a/src/rag/loader.rs +++ b/src/rag/loader.rs @@ -60,7 +60,7 @@ pub async fn load_file( loaders: &HashMap, path: &str, ) -> Result<(String, RagMetadata)> { - let extension = path_extension(path).unwrap_or_else(|| DEFAULT_EXTENSION.into()); + let extension = get_patch_extension(path).unwrap_or_else(|| DEFAULT_EXTENSION.into()); match loaders.get(&extension) { Some(loader_command) => load_with_command(path, &extension, loader_command), None => load_plain(path, &extension).await, @@ -71,7 +71,7 @@ pub async fn load_url( loaders: &HashMap, path: &str, ) -> Result<(String, RagMetadata)> { - let (contents, extension) = fetch(loaders, path).await?; + let (contents, extension) = fetch(loaders, path, false).await?; let mut metadata: RagMetadata = Default::default(); metadata.insert(PATH_METADATA.into(), path.into()); metadata.insert(EXTENSION_METADATA.into(), extension); -- cgit v1.2.3