filter excel empty sheet (#8194)

2024-09-10 14:55:08 +08:00
parent 86f7f245e4
commit af92f19291
1 changed files with 4 additions and 1 deletions
--- a/api/core/rag/extractor/excel_extractor.py
+++ b/api/core/rag/extractor/excel_extractor.py
@@ -38,7 +38,10 @@ class ExcelExtractor(BaseExtractor):
            for sheet_name in wb.sheetnames:
                sheet = wb[sheet_name]
                data = sheet.values
-                cols = next(data)
+                try:
+                    cols = next(data)
+                except StopIteration:
+                    continue
                df = pd.DataFrame(data, columns=cols)

                df.dropna(how='all', inplace=True)