diff --git a/api/core/data_loader/file_extractor.py b/api/core/data_loader/file_extractor.py index 4b811f51c6..a603bee749 100644 --- a/api/core/data_loader/file_extractor.py +++ b/api/core/data_loader/file_extractor.py @@ -6,7 +6,7 @@ import requests from langchain.document_loaders import TextLoader, Docx2txtLoader from langchain.schema import Document -from core.data_loader.loader.csv import CSVLoader +from core.data_loader.loader.csv_loader import CSVLoader from core.data_loader.loader.excel import ExcelLoader from core.data_loader.loader.html import HTMLLoader from core.data_loader.loader.markdown import MarkdownLoader diff --git a/api/core/data_loader/loader/csv.py b/api/core/data_loader/loader/csv_loader.py similarity index 98% rename from api/core/data_loader/loader/csv.py rename to api/core/data_loader/loader/csv_loader.py index ff57ef61e7..0789121cf1 100644 --- a/api/core/data_loader/loader/csv.py +++ b/api/core/data_loader/loader/csv_loader.py @@ -1,10 +1,10 @@ import logging +import csv from typing import Optional, Dict, List from langchain.document_loaders import CSVLoader as LCCSVLoader from langchain.document_loaders.helpers import detect_file_encodings - -from models.dataset import Document +from langchain.schema import Document logger = logging.getLogger(__name__)