第一次提交全部代码

database/create_db.py → create_db.py

@@ -6,8 +6,8 @@ from langchain_chroma.vectorstores import Chroma
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 sys.path.append(os.path.dirname(os.path.dirname(__file__)))
 # 首先实现基本配置
-DEFAULT_DB_PATH = r"../knowledge_db"
-DEFAULT_PERSIST_PATH = '../vector_db/chroma'
+DEFAULT_DB_PATH = r"knowledge_db"
+DEFAULT_PERSIST_PATH = 'vector_db/chroma'
 
 # 目录下保存不同的文件
 # def get_files(dir_path):

database/chormadb.py

@@ -1,7 +1,7 @@
 import chromadb
 from embedding.embedding import get_embedding
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from database.create_db import load_txt_from_dir
+from create_db import load_txt_from_dir
 
 DEFAULT_DB_PATH = r"../knowledge_db"
 DEFAULT_PERSIST_PATH = '../vector_db/chroma'

qa_chain/get_vectordb.py

@@ -1,5 +1,5 @@
 import os
-from database.create_db import create_db, load_knowledge_db
+from create_db import create_db, load_knowledge_db
 from embedding.embedding import get_embedding
 
 # 定义默认路径