pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

add docker compose (#8)

Browse Source 
* add docker compose

* add docker compose
This commit is contained in:
KevinHuSh
2023-12-15 19:38:32 +08:00
committed by GitHub
parent f4456af464
commit 738c322508
6 changed files with 153 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
python/nlp/huchunk.py
View File

@ -291,6 +291,12 @@ class PdfChunker(HuChunker):
class DocxChunker(HuChunker):
@dataclass
class Fields:
text_chunks: List = None
table_chunks: List = None
def __init__(self, doc_parser):
self.doc = doc_parser
super().__init__()
@ -336,6 +342,12 @@ class DocxChunker(HuChunker):
class ExcelChunker(HuChunker):
@dataclass
class Fields:
text_chunks: List = None
table_chunks: List = None
def __init__(self, excel_parser):
self.excel = excel_parser
super().__init__()
@ -354,10 +366,10 @@ if __name__ == "__main__":
from parser import PdfParser
ckr = PdfChunker(PdfParser())
if sys.argv[1].split(".")[-1].lower().find("doc") >= 0:
from .parser import DocxParser
from parser import DocxParser
ckr = DocxChunker(DocxParser())
if sys.argv[1].split(".")[-1].lower().find("xlsx") >= 0:
from .parser import ExcelParser
from parser import ExcelParser
ckr = ExcelChunker(ExcelParser())
# ckr.html(sys.argv[1])