diff --git a/src/knowledge/base.py b/src/knowledge/base.py index cfde47f2..2eb643a4 100644 --- a/src/knowledge/base.py +++ b/src/knowledge/base.py @@ -136,7 +136,12 @@ class KnowledgeBase(ABC): """ from src.utils import hashstr - db_id = f"kb_{hashstr(database_name, with_salt=True)}" + from src.utils import hashstr + + # 从 kwargs 中获取 is_private 配置 + is_private = kwargs.get('is_private', False) + prefix = "kb_private_" if is_private else "kb_" + db_id = f"{prefix}{hashstr(database_name, with_salt=True)}" # 创建数据库记录 # 确保 Pydantic 模型被转换为字典,以便 JSON 序列化 diff --git a/src/knowledge/implementations/chroma.py b/src/knowledge/implementations/chroma.py index 976b6060..c7c4bd99 100644 --- a/src/knowledge/implementations/chroma.py +++ b/src/knowledge/implementations/chroma.py @@ -20,7 +20,7 @@ from src.utils.datetime_utils import utc_isoformat class ChromaKB(KnowledgeBase): - """基于 ChromaDB 的向量知识库实现""" + """基于 ChromaDB 的向量库""" def __init__(self, work_dir: str, **kwargs): """ diff --git a/src/knowledge/implementations/milvus.py b/src/knowledge/implementations/milvus.py index f999c43a..6b4da162 100644 --- a/src/knowledge/implementations/milvus.py +++ b/src/knowledge/implementations/milvus.py @@ -21,7 +21,7 @@ MILVUS_AVAILABLE = True class MilvusKB(KnowledgeBase): - """基于 Milvus 的生产级向量知识库实现""" + """基于 Milvus 的生产级向量库""" def __init__(self, work_dir: str, **kwargs): """ diff --git a/web/src/views/DataBaseView.vue b/web/src/views/DataBaseView.vue index e49060e0..7b4e8b65 100644 --- a/web/src/views/DataBaseView.vue +++ b/web/src/views/DataBaseView.vue @@ -76,6 +76,20 @@ placeholder="新建知识库描述" :auto-size="{ minRows: 5, maxRows: 10 }" /> + +

隐私设置

+
+ + 设置为私有知识库 + + + +