jack 1 月之前
父節點
當前提交
efd352f5a6
共有 2 個文件被更改,包括 3 次插入2 次删除
  1. 2 1
      py/core/parsers/structured/csv_parser.py
  2. 1 1
      py/core/providers/ingestion/unstructured/base.py

+ 2 - 1
py/core/parsers/structured/csv_parser.py

@@ -43,8 +43,9 @@ class CSVParserAdvanced(AsyncParser[str | bytes]):
     """A parser for CSV data."""
 
     def __init__(
-        self, config: IngestionConfig, llm_provider: CompletionProvider
+        self, config: IngestionConfig, database_provider: DatabaseProvider, llm_provider: CompletionProvider
     ):
+        self.database_provider = database_provider
         self.llm_provider = llm_provider
         self.config = config
 

+ 1 - 1
py/core/providers/ingestion/unstructured/base.py

@@ -98,7 +98,7 @@ class UnstructuredIngestionProvider(IngestionProvider):
         DocumentType.XLSX: [parsers.XLSXParser],  # type: ignore
         #DocumentType.DOC: [parsers.DOCParser],  # type: ignore
         DocumentType.PPT: [parsers.PPTParser],  # type: ignore
-        DocumentType.CSV: [parsers.CSVParser],  # type: ignore
+        DocumentType.CSV: [parsers.CSVParserAdvanced],  # type: ignore
     }
 
     EXTRA_PARSERS = {