Browse Source

filter excel empty sheet (#8194)

Jyong 7 tháng trước cách đây
mục cha
commit
af92f19291

+ 4 - 1
api/core/rag/extractor/excel_extractor.py

@@ -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)