filter excel empty sheet (langgenius#8194)

heydevs-io · Oct 23, 2024 · e76b943 · e76b943
1 parent 2d7223d
commit e76b943
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/api/core/rag/extractor/excel_extractor.py b/api/core/rag/extractor/excel_extractor.py
@@ -38,7 +38,10 @@ def extract(self) -> list[Document]:
             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)