github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-14 22:43:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

style:fmt

Browse Source
This commit is contained in:
aries_ckt 2023-07-25 17:22:11 +08:00
parent e7dcbe3dab
commit 795be61d4a
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/embedding_engine/docx_loader.py
View File

@ -22,5 +22,7 @@ class DocxLoader(BaseLoader):
para = doc.paragraphs[i]
text = para.text
content.append(text)
docs.append(Document(page_content=''.join(content), metadata={"source": self.file_path}))
docs.append(
Document(page_content="".join(content), metadata={"source": self.file_path})
)
return docs

6
pilot/embedding_engine/ppt_loader.py
View File

@ -20,5 +20,9 @@ class PPTLoader(BaseLoader):
for slide in pr.slides:
for shape in slide.shapes:
if hasattr(shape, "text") and shape.text is not "":
docs.append(Document(page_content=shape.text, metadata={"source": slide.slide_id}))
docs.append(
Document(
page_content=shape.text, metadata={"source": slide.slide_id}
)
)
return docs