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/config/mod.rs | |
| parent | 33a0049a81103c1f6c5639b9cec43cb14a9ef367 (diff) | |
| download | aichat-f82524fd154a1bf4a7277db868147afdb0cd507a.tar.gz | |
feat: implement native rag url loader (#660)
Diffstat (limited to 'src/config/mod.rs')
| -rw-r--r-- | src/config/mod.rs | 16 |
1 files changed, 6 insertions, 10 deletions
diff --git a/src/config/mod.rs b/src/config/mod.rs index 69609f4..ff337c0 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -1434,16 +1434,12 @@ impl Config { } fn setup_rag_document_loaders(&mut self) { - [ - ("pdf", "pdftotext $1 -"), - ("docx", "pandoc --to plain $1"), - ("url", "curl -fsSL $1"), - ] - .into_iter() - .for_each(|(k, v)| { - let (k, v) = (k.to_string(), v.to_string()); - self.rag_document_loaders.entry(k).or_insert(v); - }); + [("pdf", "pdftotext $1 -"), ("docx", "pandoc --to plain $1")] + .into_iter() + .for_each(|(k, v)| { + let (k, v) = (k.to_string(), v.to_string()); + self.rag_document_loaders.entry(k).or_insert(v); + }); } } |
