ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-02-01 08:48:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update lollms_file_system.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-06-18 09:09:23 +02:00 committed by GitHub
parent fa66706061
commit 9a47685a8c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/server/endpoints/lollms_file_system.py
View File

@ -134,7 +134,7 @@ def select_rag_database() -> Optional[Dict[str, Path]]:
vdb = VectorDatabase(Path(folder_path)/"db_name.sqlite", v)
# Get all files in the folder
folder = Path(folder_path)
file_types = ['*.txt', '*.pdf', '*.docx', '*.pptx', '*.msg']
file_types = [f"*{f}" for f in TextDocumentsLoader.get_supported_file_types]
files = []
for file_type in file_types:
files.extend(folder.glob(file_type))