From 02c68a0ca582a6b56db94e6e4a592af70404a0e6 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Sun, 26 Oct 2025 15:08:43 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E9=9A=90=E7=A7=81?=
 =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E5=8A=9F=E8=83=BD=EF=BC=8C=E6=94=AF=E6=8C=81?=
 =?UTF-8?q?=E5=88=9B=E5=BB=BA=E7=A7=81=E6=9C=89=E7=9F=A5=E8=AF=86=E5=BA=93?=
 =?UTF-8?q?=E5=B9=B6=E5=9C=A8=E7=95=8C=E9=9D=A2=E4=B8=AD=E6=98=BE=E7=A4=BA?=
 =?UTF-8?q?=E9=94=81=E5=AE=9A=E5=9B=BE=E6=A0=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/knowledge/base.py                   |  7 +++-
 src/knowledge/implementations/chroma.py |  2 +-
 src/knowledge/implementations/milvus.py |  2 +-
 web/src/views/DataBaseView.vue          | 50 +++++++++++++++++++++++--
 4 files changed, 54 insertions(+), 7 deletions(-)

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 }"
       />
+
+      <h3 style="margin-top: 20px;">隐私设置</h3>
+      <div class="privacy-config">
+        <a-switch
+          v-model:checked="newDatabase.is_private"
+          checked-children="私有"
+          un-checked-children="公开"
+          size="default"
+        />
+        <span style="margin-left: 12px;">设置为私有知识库</span>
+        <a-tooltip title="在部分智能体的设计中，可以根据隐私标志来决定启用什么模型和策略。例如，对于私有知识库，可以选择更严格的数据处理和访问控制策略，以保护敏感信息的安全性和隐私性。">
+          <InfoCircleOutlined style="margin-left: 8px; color: var(--gray-500); cursor: help;" />
+        </a-tooltip>
+      </div>
       <template #footer>
         <a-button key="back" @click="cancelCreateDatabase">取消</a-button>
         <a-button key="submit" type="primary" :loading="state.creating" @click="createDatabase">创建</a-button>
@@ -104,6 +118,12 @@
         :key="database.db_id"
         class="database dbcard"
         @click="navigateToDatabase(database.db_id)">
+        <!-- 私有知识库锁定图标 -->
+        <LockOutlined
+          v-if="database.metadata?.is_private"
+          class="private-lock-icon"
+          title="私有知识库"
+        />
         <div class="top">
           <div class="icon">
             <component :is="getKbTypeIcon(database.kb_type || 'lightrag')" />
@@ -132,7 +152,7 @@
           >
             {{ getKbTypeLabel(database.kb_type || 'lightrag') }}
           </a-tag>
-        </div>
+          </div>
 
         <!-- <button @click="deleteDatabase(database.collection_name)">删除</button> -->
       </div>
@@ -146,6 +166,7 @@ import { useRouter, useRoute } from 'vue-router';
 import { useConfigStore } from '@/stores/config';
 import { message } from 'ant-design-vue'
 import { BookPlus, Database, Zap, FileDigit,  Waypoints, Building2 } from 'lucide-vue-next';
+import { LockOutlined, InfoCircleOutlined } from '@ant-design/icons-vue';
 import { databaseApi, typeApi } from '@/apis/knowledge_api';
 import HeaderComponent from '@/components/HeaderComponent.vue';
 import ModelSelectorComponent from '@/components/ModelSelectorComponent.vue';
@@ -189,6 +210,7 @@ const emptyEmbedInfo = {
   description: '',
   embed_model_name: configStore.config?.embed_model,
   kb_type: 'chroma', // 默认为 Milvus
+  is_private: false, // 默认为公开知识库
   // Vector 知识库特有配置
   storage: '', // 存储方式配置
   // LightRAG 特有配置
@@ -391,7 +413,9 @@ const createDatabase = () => {
     description: newDatabase.description?.trim() || '',
     embed_model_name: newDatabase.embed_model_name || configStore.config.embed_model,
     kb_type: newDatabase.kb_type,
-    additional_params: {}
+    additional_params: {
+      is_private: newDatabase.is_private || false
+    }
   }
 
   // 添加类型特有的配置
@@ -450,6 +474,12 @@ onMounted(() => {
     margin: 12px 0;
   }
 
+  .privacy-config {
+    display: flex;
+    align-items: center;
+    margin-bottom: 12px;
+  }
+
   .kb-type-cards {
     display: grid;
     grid-template-columns: repeat(3, 1fr);
@@ -473,7 +503,6 @@ onMounted(() => {
 
       &:hover {
         border-color: var(--main-color);
-        transform: translateY(-1px);
       }
 
       // 为不同知识库类型设置不同的悬停颜色
@@ -577,7 +606,7 @@ onMounted(() => {
         color: var(--gray-600);
         line-height: 1.5;
         margin-bottom: 12px;
-        min-height: 40px;
+        // min-height: 40px;
       }
 
       .card-features {
@@ -687,6 +716,19 @@ onMounted(() => {
   cursor: pointer;
   display: flex;
   flex-direction: column;
+  position: relative; // 为绝对定位的锁定图标提供参考
+
+  .private-lock-icon {
+    position: absolute;
+    top: 24px;
+    right: 16px;
+    color: var(--gray-700);
+    background: var(--gray-100);
+    font-size: 12px;
+    border-radius: 6px;
+    padding: 5px;
+    z-index: 1;
+  }
 
   .top {
     display: flex;