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."""
     """A parser for CSV data."""
 
 
     def __init__(
     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.llm_provider = llm_provider
         self.config = config
         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.XLSX: [parsers.XLSXParser],  # type: ignore
         #DocumentType.DOC: [parsers.DOCParser],  # type: ignore
         #DocumentType.DOC: [parsers.DOCParser],  # type: ignore
         DocumentType.PPT: [parsers.PPTParser],  # type: ignore
         DocumentType.PPT: [parsers.PPTParser],  # type: ignore
-        DocumentType.CSV: [parsers.CSVParser],  # type: ignore
+        DocumentType.CSV: [parsers.CSVParserAdvanced],  # type: ignore
     }
     }
 
 
     EXTRA_PARSERS = {
     EXTRA_PARSERS = {