diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index cc6c4b83..3a8a072b 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -977,15 +977,16 @@ def create_app(args): if not pm.is_installed("docling"): pm.install("docling") from docling.document_converter import DocumentConverter + converter = DocumentConverter() result = converter.convert(file_path) content = result.document.export_to_markdown() - case ".docx": if not pm.is_installed("docling"): pm.install("docling") from docling.document_converter import DocumentConverter + converter = DocumentConverter() result = converter.convert(file_path) content = result.document.export_to_markdown() @@ -994,6 +995,7 @@ def create_app(args): if not pm.is_installed("docling"): pm.install("docling") from docling.document_converter import DocumentConverter + converter = DocumentConverter() result = converter.convert(file_path) content = result.document.export_to_markdown() @@ -1001,6 +1003,7 @@ def create_app(args): if not pm.is_installed("docling"): pm.install("docling") from docling.document_converter import DocumentConverter + converter = DocumentConverter() result = converter.convert(file_path) content = result.document.export_to_markdown() @@ -1285,32 +1288,35 @@ def create_app(args): if not pm.is_installed("docling"): pm.install("docling") from docling.document_converter import DocumentConverter - converter = DocumentConverter() - result = converter.convert(file_path) - content = result.document.export_to_markdown() + converter = DocumentConverter() + result = converter.convert(file.filename) + content = result.document.export_to_markdown() case ".docx": if not pm.is_installed("docling"): pm.install("docling") from docling.document_converter import DocumentConverter + converter = DocumentConverter() - result = converter.convert(file_path) + result = converter.convert(file.filename) content = result.document.export_to_markdown() case ".pptx": if not pm.is_installed("docling"): pm.install("docling") from docling.document_converter import DocumentConverter + converter = DocumentConverter() - result = converter.convert(file_path) + result = converter.convert(file.filename) content = result.document.export_to_markdown() case ".xlsx": if not pm.is_installed("docling"): pm.install("docling") from docling.document_converter import DocumentConverter + converter = DocumentConverter() - result = converter.convert(file_path) + result = converter.convert(file.filename) content = result.document.export_to_markdown() case _: raise HTTPException( diff --git a/requirements.txt b/requirements.txt index 9f9660d9..c5ccac23 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,6 +2,9 @@ accelerate aiofiles aiohttp configparser + +# File manipulation libraries +docling graspologic # database packages @@ -11,9 +14,6 @@ networkx numpy pipmaster pydantic - -# File manipulation libraries -docling python-dotenv setuptools