summaryrefslogtreecommitdiffstats
path: root/src/rag
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-07-10 10:48:02 +0800
committerGitHub <noreply@github.com>2024-07-10 10:48:02 +0800
commit48219d2808bafc88fa4fa4d6997aa9dc92e6f686 (patch)
tree65fefbd6e1c883a8ea55568bd24ac67c678b4a7c /src/rag
parent0afe5fa24b8990991635c7023f0fb89403f55d90 (diff)
downloadaichat-48219d2808bafc88fa4fa4d6997aa9dc92e6f686.tar.gz
refactor: fetch image as data url (#694)
Diffstat (limited to 'src/rag')
-rw-r--r--src/rag/loader.rs4
1 files changed, 2 insertions, 2 deletions
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<String, String>,
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<String, String>,
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);