소스 검색

excel: get keys from every sheet (#2796)

Weaxs 1 년 전
부모
커밋
20bd49285b
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      api/core/rag/extractor/excel_extractor.py

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

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