update class method of KnowledgeBase and KnowledgeFile
This commit is contained in:
parent
14a5de6a29
commit
cd0f2a2c41
|
|
@ -16,8 +16,6 @@ SUPPORTED_VS_TYPES = ["faiss", "milvus"]
|
||||||
DB_ROOT = os.path.join(KB_ROOT_PATH, "info.db")
|
DB_ROOT = os.path.join(KB_ROOT_PATH, "info.db")
|
||||||
|
|
||||||
|
|
||||||
# TODO: 知识库信息入库
|
|
||||||
|
|
||||||
def list_kbs_from_db():
|
def list_kbs_from_db():
|
||||||
conn = sqlite3.connect(DB_ROOT)
|
conn = sqlite3.connect(DB_ROOT)
|
||||||
c = conn.cursor()
|
c = conn.cursor()
|
||||||
|
|
|
||||||
|
|
@ -10,7 +10,6 @@ class KnowledgeFile:
|
||||||
knowledge_base_name: str
|
knowledge_base_name: str
|
||||||
):
|
):
|
||||||
self.kb = KnowledgeBase.load(knowledge_base_name)
|
self.kb = KnowledgeBase.load(knowledge_base_name)
|
||||||
self.knowledge_base_type = "faiss"
|
|
||||||
self.filename = filename
|
self.filename = filename
|
||||||
self.ext = os.path.splitext(filename)[-1]
|
self.ext = os.path.splitext(filename)[-1]
|
||||||
self.filepath = get_file_path(knowledge_base_name, filename)
|
self.filepath = get_file_path(knowledge_base_name, filename)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue