Skip to content

Commit 15307b8

Browse files
authored
Update prepdocs.py
updating the csv parser code and importing the CsvParser class
1 parent bc1b5e3 commit 15307b8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

app/backend/prepdocs.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
IntegratedVectorizerStrategy,
2121
)
2222
from prepdocslib.jsonparser import JsonParser
23+
from prepdocslib.csvparser import CsvParser
2324
from prepdocslib.listfilestrategy import (
2425
ADLSGen2ListFileStrategy,
2526
ListFileStrategy,
@@ -183,6 +184,7 @@ def setup_file_processors(
183184
".docx": FileProcessor(doc_int_parser, sentence_text_splitter),
184185
".pptx": FileProcessor(doc_int_parser, sentence_text_splitter),
185186
".xlsx": FileProcessor(doc_int_parser, sentence_text_splitter),
187+
".csv": FileProcessor(CsvParser(), sentence_text_splitter),
186188
".png": FileProcessor(doc_int_parser, sentence_text_splitter),
187189
".jpg": FileProcessor(doc_int_parser, sentence_text_splitter),
188190
".jpeg": FileProcessor(doc_int_parser, sentence_text_splitter),

0 commit comments

Comments
 (0)