gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added extra extentions

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-02-20 10:59:07 +01:00
committed by GitHub
parent fbe7e9994d
commit 38b62d58ad
1 changed files with 38 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
lightrag/api/lightrag_server.py
View File

@@ -579,7 +579,39 @@ class DocumentManager:
".docx",
".pptx",
".xlsx",
),
".rtf", # Rich Text Format
".odt", # OpenDocument Text
".tex", # LaTeX
".epub", # Electronic Publication
".html", # HyperText Markup Language
".htm", # HyperText Markup Language
".csv", # Comma-Separated Values
".json", # JavaScript Object Notation
".xml", # eXtensible Markup Language
".yaml", # YAML Ain't Markup Language
".yml", # YAML
".log", # Log files
".conf", # Configuration files
".ini", # Initialization files
".properties",# Java properties files
".sql", # SQL scripts
".bat", # Batch files
".sh", # Shell scripts
".c", # C source code
".cpp", # C++ source code
".py", # Python source code
".java", # Java source code
".js", # JavaScript source code
".ts", # TypeScript source code
".swift", # Swift source code
".go", # Go source code
".rb", # Ruby source code
".php", # PHP source code
".css", # Cascading Style Sheets
".scss", # Sassy CSS
".less", # LESS CSS
)
):
self.input_dir = Path(input_dir)
self.supported_extensions = supported_extensions
@@ -1148,8 +1180,12 @@ def create_app(args):
# Process based on file type
match ext:
case ".txt" | ".md":
case ".txt" | ".md" | ".html" | ".htm" | ".tex" | ".json" | ".xml" | ".yaml" | ".yml" | \
".rtf" | ".odt" | ".epub" | ".csv" | ".log" | ".conf" | ".ini" | ".properties" | \
".sql" | ".bat" | ".sh" | ".c" | ".cpp" | ".py" | ".java" | ".js" | ".ts" | \
".swift" | ".go" | ".rb" | ".php" | ".css" | ".scss" | ".less":
content = file.decode("utf-8")
case ".pdf":
if not pm.is_installed("pypdf2"):
pm.install("pypdf2")