AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 18:49:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

excel: get keys from every sheet (#2796)

Browse Source
This commit is contained in:
Weaxs 2024-03-12 01:59:25 -07:00 committed by GitHub
parent 6bd2730317
commit 20bd49285b
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
api/core/rag/extractor/excel_extractor.py
View File

@ -29,10 +29,10 @@ class ExcelExtractor(BaseExtractor):
def extract(self) -> list[Document]: def extract(self) -> list[Document]:
"""Load from file path.""" """Load from file path."""
data = [] data = []
keys = []
wb = load_workbook(filename=self._file_path, read_only=True) wb = load_workbook(filename=self._file_path, read_only=True)
# loop over all sheets # loop over all sheets
for sheet in wb: for sheet in wb:
keys = []
if 'A1:A1' == sheet.calculate_dimension(): if 'A1:A1' == sheet.calculate_dimension():
sheet.reset_dimensions() sheet.reset_dimensions()
for row in sheet.iter_rows(values_only=True): for row in sheet.iter_rows(values_only=True):