fix dataset retrival in dataset mode (#3334)

2026-05-03 00:48:04 +08:00 · 2024-04-11 02:11:21 +08:00
parent 826c422ac4
commit 6164604462
2 changed files with 93 additions and 0 deletions
--- a/api/core/rag/extractor/csv_extractor.py
+++ b/api/core/rag/extractor/csv_extractor.py
@ -34,6 +34,7 @@ class CSVExtractor(BaseExtractor):

    def extract(self) -> list[Document]:
        """Load data into document objects."""
+        docs = []
        try:
            with open(self._file_path, newline="", encoding=self._encoding) as csvfile:
                docs = self._read_from_file(csvfile)