AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-16 10:25:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

filter excel empty sheet (#8194)

Browse Source
This commit is contained in:
Jyong 2024-09-10 14:55:08 +08:00 committed by GitHub
parent 86f7f245e4
commit af92f19291
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/rag/extractor/excel_extractor.py
View File

@ -38,7 +38,10 @@ class ExcelExtractor(BaseExtractor):
for sheet_name in wb.sheetnames: for sheet_name in wb.sheetnames:
sheet = wb[sheet_name] sheet = wb[sheet_name]
data = sheet.values data = sheet.values
cols = next(data) try:
cols = next(data)
except StopIteration:
continue
df = pd.DataFrame(data, columns=cols) df = pd.DataFrame(data, columns=cols)
df.dropna(how='all', inplace=True) df.dropna(how='all', inplace=True)