瀏覽代碼

fix: csv import error (#1063)

takatost 1 年之前
父節點
當前提交
a7a7aab7a0
共有 2 個文件被更改,包括 3 次插入3 次删除
  1. 1 1
      api/core/data_loader/file_extractor.py
  2. 2 2
      api/core/data_loader/loader/csv_loader.py

+ 1 - 1
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

+ 2 - 2
api/core/data_loader/loader/csv.py → 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__)