jack 1 month ago
parent
commit
efd352f5a6

+ 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 = {