From a7a7aab7a052561f1abac352196d943e68d4bea1 Mon Sep 17 00:00:00 2001 From: takatost Date: Thu, 31 Aug 2023 15:42:28 +0800 Subject: [PATCH] fix: csv import error (#1063) --- api/core/data_loader/file_extractor.py | 2 +- api/core/data_loader/loader/{csv.py => csv_loader.py} | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) rename api/core/data_loader/loader/{csv.py => csv_loader.py} (98%) 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__)