dep: bump pandas from 1.x to 2.x (#4820)

2024-06-04 13:24:28 +08:00 · 2024-06-04 13:24:28 +08:00 · 58db719a2c
commit 58db719a2c
parent 9abeb99b32
3 changed files with 64 additions and 4 deletions
--- a/api/core/rag/extractor/csv_extractor.py
+++ b/api/core/rag/extractor/csv_extractor.py
@ -57,7 +57,7 @@ class CSVExtractor(BaseExtractor):
        docs = []
        try:
            # load csv file into pandas dataframe
-            df = pd.read_csv(csvfile, error_bad_lines=False, **self.csv_args)
+            df = pd.read_csv(csvfile, on_bad_lines='skip', **self.csv_args)

            # check source column exists
            if self.source_column and self.source_column not in df.columns: