From b5c951ed9869c262c76db39cb9f1944c8221fdb5 Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Mon, 6 May 2024 15:56:00 +0800
Subject: [PATCH 1/9] =?UTF-8?q?feat:=20=E6=AE=B5=E8=90=BD=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E6=89=B9=E9=87=8F=E9=80=89=E6=8B=A9=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/styles/app.scss                        | 30 ++++++++
 ui/src/views/document/index.vue               | 22 +++---
 ui/src/views/paragraph/index.vue              | 70 ++++++++++++++++++-
 .../problem/component/RelateProblemDialog.vue | 27 +------
 4 files changed, 110 insertions(+), 39 deletions(-)

diff --git a/ui/src/styles/app.scss b/ui/src/styles/app.scss
index 69b6440b48e..3743c0e7103 100644
--- a/ui/src/styles/app.scss
+++ b/ui/src/styles/app.scss
@@ -573,3 +573,33 @@ h4 {
     color: var(--app-text-color);
   }
 }
+
+// card 选中样式
+.selected {
+  border: 1px solid var(--el-color-primary) !important;
+  &:before {
+    content: '';
+    position: absolute;
+    right: 0;
+    top: 0;
+    border: 14px solid var(--el-color-primary);
+    border-bottom-color: transparent;
+    border-left-color: transparent;
+  }
+
+  &:after {
+    content: '';
+    width: 3px;
+    height: 6px;
+    position: absolute;
+    right: 5px;
+    top: 2px;
+    border: 2px solid #fff;
+    border-top-color: transparent;
+    border-left-color: transparent;
+    transform: rotate(35deg);
+  }
+  &:hover {
+    border: 1px solid var(--el-color-primary);
+  }
+}
diff --git a/ui/src/views/document/index.vue b/ui/src/views/document/index.vue
index 8e5a5a4bc51..9b7cffdd1e2 100644
--- a/ui/src/views/document/index.vue
+++ b/ui/src/views/document/index.vue
@@ -20,15 +20,15 @@
               v-if="datasetDetail.type === '1'"
               >同步文档</el-button
             >
-            <el-button @click="openDatasetDialog()" :disabled="multipleSelection.length === 0"
-              >迁移</el-button
-            >
-            <el-button @click="openBatchEditDocument" :disabled="multipleSelection.length === 0"
-              >设置</el-button
-            >
-            <el-button @click="deleteMulDocument" :disabled="multipleSelection.length === 0"
-              >删除</el-button
-            >
+            <el-button @click="openDatasetDialog()" :disabled="multipleSelection.length === 0">
+              迁移
+            </el-button>
+            <el-button @click="openBatchEditDocument" :disabled="multipleSelection.length === 0">
+              设置
+            </el-button>
+            <el-button @click="deleteMulDocument" :disabled="multipleSelection.length === 0">
+              删除
+            </el-button>
           </div>
 
           <el-input
@@ -327,9 +327,7 @@ const handleSelectionChange = (val: any[]) => {
 
 function openBatchEditDocument() {
   const arr: string[] = multipleSelection.value.map((v) => v.id)
-  if (batchEditDocumentDialogRef) {
-    batchEditDocumentDialogRef?.value?.open(arr)
-  }
+  batchEditDocumentDialogRef?.value?.open(arr)
 }
 
 /**
diff --git a/ui/src/views/paragraph/index.vue b/ui/src/views/paragraph/index.vue
index 8471f759518..55b4d0602e3 100644
--- a/ui/src/views/paragraph/index.vue
+++ b/ui/src/views/paragraph/index.vue
@@ -8,7 +8,16 @@
         >）</el-text
       >
       <div class="document-detail__header">
-        <el-button @click="addParagraph" type="primary" :disabled="loading"> 添加分段 </el-button>
+        <el-button @click="isBatch = true" v-if="isBatch === false"> 批量选择 </el-button>
+        <el-button @click="isBatch = false" v-if="isBatch === true"> 取消选择 </el-button>
+        <el-button
+          @click="addParagraph"
+          type="primary"
+          :disabled="loading"
+          v-if="isBatch === false"
+        >
+          添加分段
+        </el-button>
       </div>
     </template>
     <div
@@ -57,7 +66,28 @@
                 :key="index"
                 class="p-8"
               >
+                <!-- 批量操作card -->
+                <CardBox
+                  v-if="isBatch === true"
+                  shadow="hover"
+                  :title="item.title || '-'"
+                  :description="item.content"
+                  class="document-card cursor"
+                  :class="multipleSelection.includes(item.id) ? 'selected' : ''"
+                  :showIcon="false"
+                  @click="selectHandle(item.id)"
+                >
+                  <div class="active-button" @click.stop></div>
+
+                  <template #footer>
+                    <div class="footer-content flex-between">
+                      <span> {{ numberFormat(item?.content.length) || 0 }} 个 字符 </span>
+                    </div>
+                  </template>
+                </CardBox>
+                <!-- 非批量操作card -->
                 <CardBox
+                  v-else
                   shadow="hover"
                   :title="item.title || '-'"
                   :description="item.content"
@@ -90,6 +120,13 @@
           </InfiniteScroll>
         </div>
       </el-scrollbar>
+
+      <div class="mul-operation border-t w-full" v-if="isBatch === true">
+        <el-button :disabled="multipleSelection.length === 0"> 迁移 </el-button>
+
+        <el-button :disabled="multipleSelection.length === 0"> 删除 </el-button>
+        <span class="ml-8"> 已选 {{ multipleSelection.length }} 项 </span>
+      </div>
     </div>
     <ParagraphDialog ref="ParagraphDialogRef" :title="title" @refresh="refresh" />
   </LayoutContainer>
@@ -118,12 +155,24 @@ const title = ref('')
 const search = ref('')
 const searchType = ref('title')
 
+// 批量操作
+const isBatch = ref(false)
+const multipleSelection = ref<any[]>([])
+
 const paginationConfig = reactive({
   current_page: 1,
   page_size: 20,
   total: 0
 })
 
+function selectHandle(id: string) {
+  if (multipleSelection.value.includes(id)) {
+    multipleSelection.value.splice(multipleSelection.value.indexOf(id), 1)
+  } else {
+    multipleSelection.value.push(id)
+  }
+}
+
 function searchHandle() {
   paginationConfig.current_page = 1
   paragraphDetail.value = []
@@ -219,6 +268,12 @@ onMounted(() => {
     height: 210px;
     background: var(--app-layout-bg-color);
     border: 1px solid var(--app-layout-bg-color);
+    &.selected {
+      background: #ffffff;
+      &:hover {
+        background: #ffffff;
+      }
+    }
     &:hover {
       background: #ffffff;
       border: 1px solid var(--el-border-color);
@@ -243,5 +298,18 @@ onMounted(() => {
       top: 16px;
     }
   }
+
+  &__main {
+    position: relative;
+    box-sizing: border-box;
+    .mul-operation {
+      position: absolute;
+      bottom: 0;
+      left: 0;
+      padding: 16px 24px;
+      box-sizing: border-box;
+      background: #ffffff;
+    }
+  }
 }
 </style>
diff --git a/ui/src/views/problem/component/RelateProblemDialog.vue b/ui/src/views/problem/component/RelateProblemDialog.vue
index 0641f85327b..7306a827ff0 100644
--- a/ui/src/views/problem/component/RelateProblemDialog.vue
+++ b/ui/src/views/problem/component/RelateProblemDialog.vue
@@ -80,7 +80,7 @@
                   :title="item.title || '-'"
                   :description="item.content"
                   class="paragraph-card cursor mb-16"
-                  :class="isAssociation(item.id) ? 'active' : ''"
+                  :class="isAssociation(item.id) ? 'selected' : ''"
                   :showIcon="false"
                   @click="associationClick(item)"
                 >
@@ -244,31 +244,6 @@ defineExpose({ open })
 <style lang="scss" scope>
 .paragraph-card {
   position: relative;
-  &.active {
-    border: 1px solid var(--el-color-primary);
-    &:before {
-      content: '';
-      position: absolute;
-      right: 0;
-      top: 0;
-      border: 14px solid var(--el-color-primary);
-      border-bottom-color: transparent;
-      border-left-color: transparent;
-    }
-
-    &:after {
-      content: '';
-      width: 3px;
-      height: 6px;
-      position: absolute;
-      right: 5px;
-      top: 2px;
-      border: 2px solid #fff;
-      border-top-color: transparent;
-      border-left-color: transparent;
-      transform: rotate(35deg);
-    }
-  }
 }
 .paragraph-badge {
   .el-badge__content {

From 48281e3b582a979b6e5be217cb0e4abf1cf9f8e1 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Tue, 7 May 2024 17:13:54 +0800
Subject: [PATCH 2/9] =?UTF-8?q?feat:=20=E6=AE=B5=E8=90=BD=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E6=89=B9=E9=87=8F=E9=80=89=E6=8B=A9=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/common/event/listener_manage.py          |  15 ++
 .../serializers/paragraph_serializers.py      | 162 +++++++++++++++++-
 apps/dataset/urls.py                          |   4 +
 apps/dataset/views/paragraph.py               |  47 ++++-
 apps/embedding/vector/base_vector.py          |   6 +-
 apps/embedding/vector/pg_vector.py            |   3 +
 6 files changed, 233 insertions(+), 4 deletions(-)

diff --git a/apps/common/event/listener_manage.py b/apps/common/event/listener_manage.py
index b266bcef0fb..3204e662bb7 100644
--- a/apps/common/event/listener_manage.py
+++ b/apps/common/event/listener_manage.py
@@ -56,6 +56,12 @@ def __init__(self, source_id_list: List[str], target_dataset_id: str):
         self.target_dataset_id = target_dataset_id
 
 
+class UpdateEmbeddingDocumentIdArgs:
+    def __init__(self, source_id_list: List[str], target_document_id: str):
+        self.source_id_list = source_id_list
+        self.target_document_id = target_document_id
+
+
 class ListenerManagement:
     embedding_by_problem_signal = signal("embedding_by_problem")
     embedding_by_paragraph_signal = signal("embedding_by_paragraph")
@@ -216,10 +222,19 @@ def update_embedding_dataset_id(args: UpdateEmbeddingDatasetIdArgs):
         VectorStore.get_embedding_vector().update_by_source_ids(args.source_id_list,
                                                                 {'dataset_id': args.target_dataset_id})
 
+    @staticmethod
+    def update_embedding_document_id(args: UpdateEmbeddingDocumentIdArgs):
+        VectorStore.get_embedding_vector().update_by_source_ids(args.source_id_list,
+                                                                {'document_id': args.target_document_id})
+
     @staticmethod
     def delete_embedding_by_source_ids(source_ids: List[str]):
         VectorStore.get_embedding_vector().delete_by_source_ids(source_ids, SourceType.PROBLEM)
 
+    @staticmethod
+    def delete_embedding_by_paragraph_ids(paragraph_ids: List[str]):
+        VectorStore.get_embedding_vector().delete_by_paragraph_ids(paragraph_ids)
+
     @staticmethod
     def delete_embedding_by_dataset_id_list(source_ids: List[str]):
         VectorStore.get_embedding_vector().delete_by_dataset_id_list(source_ids)
diff --git a/apps/dataset/serializers/paragraph_serializers.py b/apps/dataset/serializers/paragraph_serializers.py
index 0fc20744e75..2981e4d304a 100644
--- a/apps/dataset/serializers/paragraph_serializers.py
+++ b/apps/dataset/serializers/paragraph_serializers.py
@@ -15,13 +15,13 @@
 from rest_framework import serializers
 
 from common.db.search import page_search
-from common.event.listener_manage import ListenerManagement
+from common.event.listener_manage import ListenerManagement, UpdateEmbeddingDocumentIdArgs
 from common.exception.app_exception import AppApiException
 from common.mixins.api_mixin import ApiMixin
 from common.util.common import post
 from common.util.field_message import ErrMessage
 from dataset.models import Paragraph, Problem, Document, ProblemParagraphMapping
-from dataset.serializers.common_serializers import update_document_char_length
+from dataset.serializers.common_serializers import update_document_char_length, BatchSerializer
 from dataset.serializers.problem_serializers import ProblemInstanceSerializer, ProblemSerializer, ProblemSerializers
 from embedding.models import SourceType
 
@@ -272,6 +272,164 @@ def get_request_params_api():
                                       description='问题id')
                     ]
 
+    class Batch(serializers.Serializer):
+        dataset_id = serializers.UUIDField(required=True, error_messages=ErrMessage.uuid("知识库id"))
+        document_id = serializers.UUIDField(required=True, error_messages=ErrMessage.uuid("文档id"))
+
+        @transaction.atomic
+        def batch_delete(self, instance: Dict, with_valid=True):
+            if with_valid:
+                BatchSerializer(data=instance).is_valid(model=Paragraph, raise_exception=True)
+                self.is_valid(raise_exception=True)
+            paragraph_id_list = instance.get("id_list")
+            QuerySet(Paragraph).filter(id__in=paragraph_id_list).delete()
+            QuerySet(ProblemParagraphMapping).filter(paragraph_id__in=paragraph_id_list).delete()
+            # 删除向量库
+            ListenerManagement.delete_embedding_by_paragraph_ids(paragraph_id_list)
+            return True
+
+    class Migrate(ApiMixin, serializers.Serializer):
+        dataset_id = serializers.UUIDField(required=True, error_messages=ErrMessage.uuid("知识库id"))
+        document_id = serializers.UUIDField(required=True, error_messages=ErrMessage.uuid("文档id"))
+        target_dataset_id = serializers.UUIDField(required=True, error_messages=ErrMessage.uuid("目标知识库id"))
+        target_document_id = serializers.UUIDField(required=True, error_messages=ErrMessage.uuid("目标文档id"))
+        paragraph_id_list = serializers.ListField(required=True, error_messages=ErrMessage.char("段落列表"),
+                                                  child=serializers.UUIDField(required=True,
+                                                                              error_messages=ErrMessage.uuid("段落id")))
+
+        def is_valid(self, *, raise_exception=False):
+            super().is_valid(raise_exception=True)
+            document_list = QuerySet(Document).filter(
+                id__in=[self.data.get('document_id'), self.data.get('target_document_id')])
+            document_id = self.data.get('document_id')
+            target_document_id = self.data.get('target_document_id')
+            if document_id == target_document_id:
+                raise AppApiException(5000, "需要迁移的文档和目标文档一致")
+            if len([document for document in document_list if str(document.id) == self.data.get('document_id')]) < 1:
+                raise AppApiException(5000, f"文档id不存在【{self.data.get('document_id')}】")
+            if len([document for document in document_list if
+                    str(document.id) == self.data.get('target_document_id')]) < 1:
+                raise AppApiException(5000, f"目标文档id不存在【{self.data.get('target_document_id')}】")
+
+        @transaction.atomic
+        def migrate(self, with_valid=True):
+            if with_valid:
+                self.is_valid(raise_exception=True)
+            dataset_id = self.data.get('dataset_id')
+            target_dataset_id = self.data.get('target_dataset_id')
+            document_id = self.data.get('document_id')
+            target_document_id = self.data.get('target_document_id')
+            paragraph_id_list = self.data.get('paragraph_id_list')
+            paragraph_list = QuerySet(Paragraph).filter(dataset_id=dataset_id, document_id=document_id,
+                                                        id__in=paragraph_id_list)
+            problem_paragraph_mapping_list = QuerySet(ProblemParagraphMapping).filter(paragraph__in=paragraph_list)
+            # 同数据集迁移
+            if target_dataset_id == dataset_id:
+                paragraph_list.update(document_id=target_document_id)
+                if len(problem_paragraph_mapping_list):
+                    problem_paragraph_mapping_list = [
+                        self.update_problem_paragraph_mapping(target_document_id,
+                                                              problem_paragraph_mapping) for problem_paragraph_mapping
+                        in
+                        problem_paragraph_mapping_list]
+                    # 修改mapping
+                    QuerySet(ProblemParagraphMapping).bulk_update(problem_paragraph_mapping_list,
+                                                                  ['document_id'])
+                ListenerManagement.update_embedding_document_id(UpdateEmbeddingDocumentIdArgs(
+                    [paragraph.id for paragraph in paragraph_list],
+                    target_document_id))
+            # 不同数据集迁移
+            else:
+                problem_list = QuerySet(Problem).filter(
+                    id__in=[problem_paragraph_mapping.problem_id for problem_paragraph_mapping in
+                            problem_paragraph_mapping_list])
+                # 目标数据集问题
+                target_problem_list = list(
+                    QuerySet(Problem).filter(content__in=[problem.content for problem in problem_list],
+                                             dataset_id=target_dataset_id))
+
+                target_handle_problem_list = [
+                    self.get_target_dataset_problem(target_dataset_id, target_document_id, problem_paragraph_mapping,
+                                                    problem_list, target_problem_list) for
+                    problem_paragraph_mapping
+                    in
+                    problem_paragraph_mapping_list]
+
+                create_problem_list = [problem for problem, is_create in target_handle_problem_list if
+                                       is_create is not None and is_create]
+                # 插入问题
+                QuerySet(Problem).bulk_create(create_problem_list)
+                # 修改mapping
+                QuerySet(ProblemParagraphMapping).bulk_update(problem_paragraph_mapping_list,
+                                                              ['problem_id', 'dataset_id', 'document_id'])
+                paragraph_list.update(dataset_id=target_dataset_id, document_id=target_document_id)
+                ListenerManagement.update_embedding_document_id(UpdateEmbeddingDocumentIdArgs(
+                    [*[problem_paragraph_mapping.id for problem_paragraph_mapping in problem_paragraph_mapping_list]
+                        , *[paragraph.id for paragraph in paragraph_list]],
+                    target_document_id))
+
+        @staticmethod
+        def update_problem_paragraph_mapping(target_document_id: str, problem_paragraph_mapping):
+            problem_paragraph_mapping.document_id = target_document_id
+            return problem_paragraph_mapping
+
+        @staticmethod
+        def get_target_dataset_problem(target_dataset_id: str,
+                                       target_document_id: str,
+                                       problem_paragraph_mapping,
+                                       source_problem_list,
+                                       target_problem_list):
+            source_problem_list = [source_problem for source_problem in source_problem_list if
+                                   source_problem.id == problem_paragraph_mapping.problem_id]
+            problem_paragraph_mapping.dataset_id = target_dataset_id
+            problem_paragraph_mapping.document_id = target_document_id
+            if len(source_problem_list) > 0:
+                problem_content = source_problem_list[-1].content
+                problem_list = [problem for problem in target_problem_list if problem.content == problem_content]
+                if len(problem_list) > 0:
+                    problem = problem_list[-1]
+                    problem_paragraph_mapping.problem_id = problem.id
+                    return problem, False
+                else:
+                    problem = Problem(id=uuid.uuid1(), dataset_id=target_dataset_id, content=problem_content)
+                    target_problem_list.append(problem)
+                    problem_paragraph_mapping.problem_id = problem.id
+                    return problem, True
+            return None
+
+        @staticmethod
+        def get_request_params_api():
+            return [openapi.Parameter(name='dataset_id',
+                                      in_=openapi.IN_PATH,
+                                      type=openapi.TYPE_STRING,
+                                      required=True,
+                                      description='文档id'),
+                    openapi.Parameter(name='document_id',
+                                      in_=openapi.IN_PATH,
+                                      type=openapi.TYPE_STRING,
+                                      required=True,
+                                      description='文档id'),
+                    openapi.Parameter(name='target_dataset_id',
+                                      in_=openapi.IN_PATH,
+                                      type=openapi.TYPE_STRING,
+                                      required=True,
+                                      description='目标知识库id'),
+                    openapi.Parameter(name='target_document_id',
+                                      in_=openapi.IN_PATH,
+                                      type=openapi.TYPE_STRING,
+                                      required=True,
+                                      description='目标知识库id')
+                    ]
+
+        @staticmethod
+        def get_request_body_api():
+            return openapi.Schema(
+                type=openapi.TYPE_ARRAY,
+                items=openapi.Schema(type=openapi.TYPE_STRING),
+                title='段落id列表',
+                description="段落id列表"
+            )
+
     class Operate(ApiMixin, serializers.Serializer):
         # 段落id
         paragraph_id = serializers.UUIDField(required=True, error_messages=ErrMessage.char(
diff --git a/apps/dataset/urls.py b/apps/dataset/urls.py
index be68ccdc4c6..237a81f5976 100644
--- a/apps/dataset/urls.py
+++ b/apps/dataset/urls.py
@@ -25,6 +25,10 @@
     path('dataset/<str:dataset_id>/document/migrate/<str:target_dataset_id>', views.Document.Migrate.as_view()),
     path('dataset/<str:dataset_id>/document/<str:document_id>/refresh', views.Document.Refresh.as_view()),
     path('dataset/<str:dataset_id>/document/<str:document_id>/paragraph', views.Paragraph.as_view()),
+    path(
+        'dataset/<str:dataset_id>/document/<str:document_id>/paragraph/migrate/dataset/<str:target_dataset_id>/document/<str:target_document_id>',
+        views.Paragraph.BatchMigrate.as_view()),
+    path('dataset/<str:dataset_id>/document/<str:document_id>/paragraph/_batch', views.Paragraph.Batch.as_view()),
     path('dataset/<str:dataset_id>/document/<str:document_id>/paragraph/<int:current_page>/<int:page_size>',
          views.Paragraph.Page.as_view(), name='paragraph_page'),
     path('dataset/<str:dataset_id>/document/<str:document_id>/paragraph/<paragraph_id>',
diff --git a/apps/dataset/views/paragraph.py b/apps/dataset/views/paragraph.py
index 3ce6f1114f9..af968b8ab27 100644
--- a/apps/dataset/views/paragraph.py
+++ b/apps/dataset/views/paragraph.py
@@ -12,9 +12,10 @@
 from rest_framework.views import Request
 
 from common.auth import TokenAuth, has_permissions
-from common.constants.permission_constants import Permission, Group, Operate
+from common.constants.permission_constants import Permission, Group, Operate, CompareConstants
 from common.response import result
 from common.util.common import query_params_to_single_dict
+from dataset.serializers.common_serializers import BatchSerializer
 from dataset.serializers.paragraph_serializers import ParagraphSerializers
 
 
@@ -168,6 +169,50 @@ def delete(self, request: Request, dataset_id: str, document_id: str, paragraph_
             o.is_valid(raise_exception=True)
             return result.success(o.delete())
 
+    class Batch(APIView):
+        authentication_classes = [TokenAuth]
+
+        @action(methods=['DELETE'], detail=False)
+        @swagger_auto_schema(operation_summary="批量删除段落",
+                             operation_id="批量删除段落",
+                             request_body=
+                             BatchSerializer.get_request_body_api(),
+                             manual_parameters=ParagraphSerializers.Create.get_request_params_api(),
+                             responses=result.get_default_response(),
+                             tags=["知识库/文档/段落"])
+        @has_permissions(
+            lambda r, k: Permission(group=Group.DATASET, operate=Operate.MANAGE,
+                                    dynamic_tag=k.get('dataset_id')))
+        def delete(self, request: Request, dataset_id: str, document_id: str):
+            return result.success(ParagraphSerializers.Batch(
+                data={"dataset_id": dataset_id, 'document_id': document_id}).batch_delete(request.data))
+
+    class BatchMigrate(APIView):
+        authentication_classes = [TokenAuth]
+
+        @action(methods=['PUT'], detail=False)
+        @swagger_auto_schema(operation_summary="批量迁移段落",
+                             operation_id="批量迁移段落",
+                             manual_parameters=ParagraphSerializers.Migrate.get_request_params_api(),
+                             request_body=ParagraphSerializers.Migrate.get_request_body_api(),
+                             responses=result.get_default_response(),
+                             tags=["知识库/文档/段落"]
+                             )
+        @has_permissions(
+            lambda r, k: Permission(group=Group.DATASET, operate=Operate.MANAGE,
+                                    dynamic_tag=k.get('dataset_id')),
+            lambda r, k: Permission(group=Group.DATASET, operate=Operate.MANAGE,
+                                    dynamic_tag=k.get('target_dataset_id')),
+            compare=CompareConstants.AND
+        )
+        def put(self, request: Request, dataset_id: str, target_dataset_id: str, document_id: str, target_document_id):
+            return result.success(
+                ParagraphSerializers.Migrate(
+                    data={'dataset_id': dataset_id, 'target_dataset_id': target_dataset_id,
+                          'document_id': document_id,
+                          'target_document_id': target_document_id,
+                          'paragraph_id_list': request.data}).migrate())
+
     class Page(APIView):
         authentication_classes = [TokenAuth]
 
diff --git a/apps/embedding/vector/base_vector.py b/apps/embedding/vector/base_vector.py
index 496150ddab2..4c42319b97c 100644
--- a/apps/embedding/vector/base_vector.py
+++ b/apps/embedding/vector/base_vector.py
@@ -113,7 +113,7 @@ def search(self, query_text, dataset_id_list: list[str], exclude_document_id_lis
         return result[0]
 
     @abstractmethod
-    def query(self, query_text:str,query_embedding: List[float],  dataset_id_list: list[str],
+    def query(self, query_text: str, query_embedding: List[float], dataset_id_list: list[str],
               exclude_document_id_list: list[str],
               exclude_paragraph_list: list[str], is_active: bool, top_n: int, similarity: float,
               search_mode: SearchMode):
@@ -173,3 +173,7 @@ def delete_by_source_ids(self, source_ids: List[str], source_type: str):
     @abstractmethod
     def delete_by_paragraph_id(self, paragraph_id: str):
         pass
+
+    @abstractmethod
+    def delete_by_paragraph_ids(self, paragraph_ids: List[str]):
+        pass
diff --git a/apps/embedding/vector/pg_vector.py b/apps/embedding/vector/pg_vector.py
index d5e5d125eff..8665c01ef0d 100644
--- a/apps/embedding/vector/pg_vector.py
+++ b/apps/embedding/vector/pg_vector.py
@@ -139,6 +139,9 @@ def delete_by_source_id(self, source_id: str, source_type: str):
     def delete_by_paragraph_id(self, paragraph_id: str):
         QuerySet(Embedding).filter(paragraph_id=paragraph_id).delete()
 
+    def delete_by_paragraph_ids(self, paragraph_ids: List[str]):
+        QuerySet(Embedding).filter(paragraph_id__in=paragraph_ids).delete()
+
 
 class ISearch(ABC):
     @abstractmethod

From cbff463a1d23a8808ebb28bcc8a0fa53a6eef8a5 Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Tue, 7 May 2024 18:01:22 +0800
Subject: [PATCH 3/9] =?UTF-8?q?feat:=20=E6=89=B9=E9=87=8F=E5=88=A0?=
 =?UTF-8?q?=E9=99=A4=E6=AE=B5=E8=90=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/api/paragraph.ts          | 21 ++++++++++++++++++++-
 ui/src/views/paragraph/index.vue | 23 +++++++++++++++++++++--
 2 files changed, 41 insertions(+), 3 deletions(-)

diff --git a/ui/src/api/paragraph.ts b/ui/src/api/paragraph.ts
index fc1169a57f4..e5be29c2f65 100644
--- a/ui/src/api/paragraph.ts
+++ b/ui/src/api/paragraph.ts
@@ -48,6 +48,24 @@ const delParagraph: (
   )
 }
 
+/**
+ * 批量删除段落
+ * @param 参数 dataset_id, document_id
+ */
+const delMulParagraph: (
+  dataset_id: string,
+  document_id: string,
+  data: any,
+  loading?: Ref<boolean>
+) => Promise<Result<boolean>> = (dataset_id, document_id, data, loading) => {
+  return del(
+    `${prefix}/${dataset_id}/document/${document_id}/paragraph/_batch`,
+    undefined,
+    { id_list: data },
+    loading
+  )
+}
+
 /**
  * 创建段落
  * @param 参数 
@@ -189,5 +207,6 @@ export default {
   getProblem,
   postProblem,
   disassociationProblem,
-  associationProblem
+  associationProblem,
+  delMulParagraph
 }
diff --git a/ui/src/views/paragraph/index.vue b/ui/src/views/paragraph/index.vue
index 55b4d0602e3..ea1a5998330 100644
--- a/ui/src/views/paragraph/index.vue
+++ b/ui/src/views/paragraph/index.vue
@@ -9,7 +9,7 @@
       >
       <div class="document-detail__header">
         <el-button @click="isBatch = true" v-if="isBatch === false"> 批量选择 </el-button>
-        <el-button @click="isBatch = false" v-if="isBatch === true"> 取消选择 </el-button>
+        <el-button @click="cancelSelectedHandle" v-if="isBatch === true"> 取消选择 </el-button>
         <el-button
           @click="addParagraph"
           type="primary"
@@ -124,7 +124,9 @@
       <div class="mul-operation border-t w-full" v-if="isBatch === true">
         <el-button :disabled="multipleSelection.length === 0"> 迁移 </el-button>
 
-        <el-button :disabled="multipleSelection.length === 0"> 删除 </el-button>
+        <el-button :disabled="multipleSelection.length === 0" @click="deleteMulParagraph">
+          删除
+        </el-button>
         <span class="ml-8"> 已选 {{ multipleSelection.length }} 项 </span>
       </div>
     </div>
@@ -165,6 +167,23 @@ const paginationConfig = reactive({
   total: 0
 })
 
+function deleteMulParagraph() {
+  paragraphApi
+    .delMulParagraph(id, documentId, multipleSelection.value, changeStateloading)
+    .then(() => {
+      paragraphDetail.value = paragraphDetail.value.filter(
+        (v) => !multipleSelection.value.includes(v.id)
+      )
+      multipleSelection.value = []
+      MsgSuccess('批量删除成功')
+    })
+}
+
+function cancelSelectedHandle() {
+  isBatch.value = false
+  multipleSelection.value = []
+}
+
 function selectHandle(id: string) {
   if (multipleSelection.value.includes(id)) {
     multipleSelection.value.splice(multipleSelection.value.indexOf(id), 1)

From c105b5e0faf4f8e6dc9b6f81a22d6ca413ffd39e Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Tue, 7 May 2024 18:59:44 +0800
Subject: [PATCH 4/9] =?UTF-8?q?feat:=20=E6=89=B9=E9=87=8F=E8=BF=81?=
 =?UTF-8?q?=E7=A7=BB=E6=AE=B5=E8=90=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/api/paragraph.ts                       |  30 +++-
 .../component/SelectDocumentDialog.vue        | 162 ++++++++++++++++++
 ui/src/views/paragraph/index.vue              |  43 ++++-
 3 files changed, 226 insertions(+), 9 deletions(-)
 create mode 100644 ui/src/views/paragraph/component/SelectDocumentDialog.vue

diff --git a/ui/src/api/paragraph.ts b/ui/src/api/paragraph.ts
index e5be29c2f65..675fa6efab9 100644
--- a/ui/src/api/paragraph.ts
+++ b/ui/src/api/paragraph.ts
@@ -122,6 +122,33 @@ const putParagraph: (
   )
 }
 
+/**
+ * 批量迁移段落
+ * @param 参数 dataset_id,target_dataset_id,
+ */
+const putMigrateMulParagraph: (
+  dataset_id: string,
+  document_id: string,
+  target_dataset_id: string,
+  target_document_id: string,
+  data: any,
+  loading?: Ref<boolean>
+) => Promise<Result<boolean>> = (
+  dataset_id,
+  document_id,
+  target_dataset_id,
+  target_document_id,
+  data,
+  loading
+) => {
+  return put(
+    `${prefix}/${dataset_id}/document/${document_id}/paragraph/migrate/dataset/${target_dataset_id}/document/${target_document_id}`,
+    data,
+    undefined,
+    loading
+  )
+}
+
 /**
  * 问题列表
  * @param 参数 dataset_id，document_id，paragraph_id
@@ -208,5 +235,6 @@ export default {
   postProblem,
   disassociationProblem,
   associationProblem,
-  delMulParagraph
+  delMulParagraph,
+  putMigrateMulParagraph
 }
diff --git a/ui/src/views/paragraph/component/SelectDocumentDialog.vue b/ui/src/views/paragraph/component/SelectDocumentDialog.vue
new file mode 100644
index 00000000000..55a34763486
--- /dev/null
+++ b/ui/src/views/paragraph/component/SelectDocumentDialog.vue
@@ -0,0 +1,162 @@
+<template>
+  <el-dialog title="选择知识库/文档" v-model="dialogVisible" width="500">
+    <el-form
+      ref="formRef"
+      :model="form"
+      label-position="top"
+      require-asterisk-position="right"
+      :rules="rules"
+      @submit.prevent
+    >
+      <el-form-item label="选择知识库" prop="dataset_id">
+        <el-select
+          v-model="form.dataset_id"
+          filterable
+          placeholder="请选择知识库"
+          :loading="optionLoading"
+          @change="changeDataset"
+        >
+          <el-option v-for="item in datasetList" :key="item.id" :label="item.name" :value="item.id">
+            <span class="flex align-center">
+              <AppAvatar
+                v-if="!item.dataset_id && item.type === '1'"
+                class="mr-12 avatar-purple"
+                shape="square"
+                :size="24"
+              >
+                <img src="@/assets/icon_web.svg" style="width: 58%" alt="" />
+              </AppAvatar>
+              <AppAvatar
+                v-else-if="!item.dataset_id && item.type === '0'"
+                class="mr-12"
+                shape="square"
+                :size="24"
+              >
+                <img src="@/assets/icon_document.svg" style="width: 58%" alt="" />
+              </AppAvatar>
+              {{ item.name }}
+            </span>
+          </el-option>
+        </el-select>
+      </el-form-item>
+      <el-form-item label="保存至文档" prop="document_id">
+        <el-select
+          v-model="form.document_id"
+          filterable
+          placeholder="请选择文档"
+          :loading="optionLoading"
+        >
+          <el-option
+            v-for="item in documentList"
+            :key="item.id"
+            :label="item.name"
+            :value="item.id"
+          >
+            {{ item.name }}
+          </el-option>
+        </el-select>
+      </el-form-item>
+    </el-form>
+    <template #footer>
+      <span class="dialog-footer">
+        <el-button @click.prevent="dialogVisible = false"> 取消 </el-button>
+        <el-button type="primary" @click="submitForm(formRef)" :loading="loading"> 迁移 </el-button>
+      </span>
+    </template>
+  </el-dialog>
+</template>
+<script setup lang="ts">
+import { ref, watch, reactive } from 'vue'
+import { useRoute } from 'vue-router'
+import type { FormInstance, FormRules } from 'element-plus'
+import paragraphApi from '@/api/paragraph'
+import useStore from '@/stores'
+
+const { dataset, document } = useStore()
+
+const route = useRoute()
+const {
+  params: { id, documentId }
+} = route as any
+
+const emit = defineEmits(['refresh'])
+const formRef = ref()
+
+const dialogVisible = ref<boolean>(false)
+const loading = ref(false)
+
+const form = ref<any>({
+  dataset_id: '',
+  document_id: ''
+})
+
+const rules = reactive<FormRules>({
+  dataset_id: [{ required: true, message: '请选择知识库', trigger: 'change' }],
+  document_id: [{ required: true, message: '请选择文档', trigger: 'change' }]
+})
+
+const datasetList = ref<any[]>([])
+const documentList = ref<any[]>([])
+const optionLoading = ref(false)
+const paragraphList = ref<string[]>([])
+
+watch(dialogVisible, (bool) => {
+  if (!bool) {
+    form.value = {
+      dataset_id: '',
+      document_id: ''
+    }
+    datasetList.value = []
+    documentList.value = []
+    paragraphList.value = []
+    formRef.value?.clearValidate()
+  }
+})
+
+function changeDataset(id: string) {
+  form.value.document_id = ''
+  getDocument(id)
+}
+
+function getDocument(id: string) {
+  document.asyncGetAllDocument(id, loading).then((res: any) => {
+    documentList.value = res.data?.filter((v: any) => v.id !== documentId)
+  })
+}
+
+function getDataset() {
+  dataset.asyncGetAllDataset(loading).then((res: any) => {
+    datasetList.value = res.data?.filter((v: any) => v.id !== id)
+  })
+}
+
+const open = (list: any) => {
+  paragraphList.value = list
+  getDataset()
+  formRef.value?.clearValidate()
+  dialogVisible.value = true
+}
+const submitForm = async (formEl: FormInstance | undefined) => {
+  if (!formEl) return
+  await formEl.validate((valid, fields) => {
+    if (valid) {
+      paragraphApi
+        .putMigrateMulParagraph(
+          id,
+          documentId,
+          form.value.dataset_id,
+          form.value.document_id,
+          paragraphList.value,
+          loading
+        )
+        .then(() => {
+          emit('refresh')
+          dialogVisible.value = false
+        })
+    }
+  })
+}
+
+defineExpose({ open })
+</script>
+<style lang="scss" scope></style>
diff --git a/ui/src/views/paragraph/index.vue b/ui/src/views/paragraph/index.vue
index ea1a5998330..66c70f74def 100644
--- a/ui/src/views/paragraph/index.vue
+++ b/ui/src/views/paragraph/index.vue
@@ -122,7 +122,9 @@
       </el-scrollbar>
 
       <div class="mul-operation border-t w-full" v-if="isBatch === true">
-        <el-button :disabled="multipleSelection.length === 0"> 迁移 </el-button>
+        <el-button :disabled="multipleSelection.length === 0" @click="openSelectDocumentDialog">
+          迁移
+        </el-button>
 
         <el-button :disabled="multipleSelection.length === 0" @click="deleteMulParagraph">
           删除
@@ -131,6 +133,7 @@
       </div>
     </div>
     <ParagraphDialog ref="ParagraphDialogRef" :title="title" @refresh="refresh" />
+    <SelectDocumentDialog ref="SelectDocumentDialogRef" @refresh="refreshMigrateParagraph" />
   </LayoutContainer>
 </template>
 <script setup lang="ts">
@@ -139,6 +142,7 @@ import { useRoute } from 'vue-router'
 import documentApi from '@/api/document'
 import paragraphApi from '@/api/paragraph'
 import ParagraphDialog from './component/ParagraphDialog.vue'
+import SelectDocumentDialog from './component/SelectDocumentDialog.vue'
 import { numberFormat } from '@/utils/utils'
 import { MsgSuccess, MsgConfirm } from '@/utils/message'
 import useStore from '@/stores'
@@ -148,6 +152,7 @@ const {
   params: { id, documentId }
 } = route as any
 
+const SelectDocumentDialogRef = ref()
 const ParagraphDialogRef = ref()
 const loading = ref(false)
 const changeStateloading = ref(false)
@@ -167,16 +172,38 @@ const paginationConfig = reactive({
   total: 0
 })
 
+function refreshMigrateParagraph() {
+  paragraphDetail.value = paragraphDetail.value.filter(
+    (v) => !multipleSelection.value.includes(v.id)
+  )
+  multipleSelection.value = []
+  MsgSuccess('迁移删除成功')
+}
+
+function openSelectDocumentDialog() {
+  SelectDocumentDialogRef.value.open(multipleSelection.value)
+}
 function deleteMulParagraph() {
-  paragraphApi
-    .delMulParagraph(id, documentId, multipleSelection.value, changeStateloading)
+  MsgConfirm(
+    `是否批量删除 ${multipleSelection.value.length} 个分段?`,
+    `删除后无法恢复，请谨慎操作。`,
+    {
+      confirmButtonText: '删除',
+      confirmButtonClass: 'danger'
+    }
+  )
     .then(() => {
-      paragraphDetail.value = paragraphDetail.value.filter(
-        (v) => !multipleSelection.value.includes(v.id)
-      )
-      multipleSelection.value = []
-      MsgSuccess('批量删除成功')
+      paragraphApi
+        .delMulParagraph(id, documentId, multipleSelection.value, changeStateloading)
+        .then(() => {
+          paragraphDetail.value = paragraphDetail.value.filter(
+            (v) => !multipleSelection.value.includes(v.id)
+          )
+          multipleSelection.value = []
+          MsgSuccess('批量删除成功')
+        })
     })
+    .catch(() => {})
 }
 
 function cancelSelectedHandle() {

From b48c23949ff37b6b4a747ad07b379ae18669d12b Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Tue, 7 May 2024 19:35:32 +0800
Subject: [PATCH 5/9] =?UTF-8?q?feat:=20=E6=AE=B5=E8=90=BD=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E6=89=B9=E9=87=8F=E9=80=89=E6=8B=A9=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/dataset/serializers/paragraph_serializers.py      | 10 +++++++---
 .../views/paragraph/component/SelectDocumentDialog.vue |  2 +-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/apps/dataset/serializers/paragraph_serializers.py b/apps/dataset/serializers/paragraph_serializers.py
index 2981e4d304a..84b5fd7ed0d 100644
--- a/apps/dataset/serializers/paragraph_serializers.py
+++ b/apps/dataset/serializers/paragraph_serializers.py
@@ -15,7 +15,7 @@
 from rest_framework import serializers
 
 from common.db.search import page_search
-from common.event.listener_manage import ListenerManagement, UpdateEmbeddingDocumentIdArgs
+from common.event.listener_manage import ListenerManagement, UpdateEmbeddingDocumentIdArgs, UpdateEmbeddingDatasetIdArgs
 from common.exception.app_exception import AppApiException
 from common.mixins.api_mixin import ApiMixin
 from common.util.common import post
@@ -325,7 +325,6 @@ def migrate(self, with_valid=True):
             problem_paragraph_mapping_list = QuerySet(ProblemParagraphMapping).filter(paragraph__in=paragraph_list)
             # 同数据集迁移
             if target_dataset_id == dataset_id:
-                paragraph_list.update(document_id=target_document_id)
                 if len(problem_paragraph_mapping_list):
                     problem_paragraph_mapping_list = [
                         self.update_problem_paragraph_mapping(target_document_id,
@@ -338,6 +337,7 @@ def migrate(self, with_valid=True):
                 ListenerManagement.update_embedding_document_id(UpdateEmbeddingDocumentIdArgs(
                     [paragraph.id for paragraph in paragraph_list],
                     target_document_id))
+                paragraph_list.update(document_id=target_document_id)
             # 不同数据集迁移
             else:
                 problem_list = QuerySet(Problem).filter(
@@ -362,11 +362,15 @@ def migrate(self, with_valid=True):
                 # 修改mapping
                 QuerySet(ProblemParagraphMapping).bulk_update(problem_paragraph_mapping_list,
                                                               ['problem_id', 'dataset_id', 'document_id'])
-                paragraph_list.update(dataset_id=target_dataset_id, document_id=target_document_id)
                 ListenerManagement.update_embedding_document_id(UpdateEmbeddingDocumentIdArgs(
                     [*[problem_paragraph_mapping.id for problem_paragraph_mapping in problem_paragraph_mapping_list]
                         , *[paragraph.id for paragraph in paragraph_list]],
                     target_document_id))
+                ListenerManagement.update_embedding_dataset_id(UpdateEmbeddingDatasetIdArgs(
+                    [*[problem_paragraph_mapping.id for problem_paragraph_mapping in problem_paragraph_mapping_list]
+                        , *[paragraph.id for paragraph in paragraph_list]],
+                    target_dataset_id))
+                paragraph_list.update(dataset_id=target_dataset_id, document_id=target_document_id)
 
         @staticmethod
         def update_problem_paragraph_mapping(target_document_id: str, problem_paragraph_mapping):
diff --git a/ui/src/views/paragraph/component/SelectDocumentDialog.vue b/ui/src/views/paragraph/component/SelectDocumentDialog.vue
index 55a34763486..caebe338ffd 100644
--- a/ui/src/views/paragraph/component/SelectDocumentDialog.vue
+++ b/ui/src/views/paragraph/component/SelectDocumentDialog.vue
@@ -126,7 +126,7 @@ function getDocument(id: string) {
 
 function getDataset() {
   dataset.asyncGetAllDataset(loading).then((res: any) => {
-    datasetList.value = res.data?.filter((v: any) => v.id !== id)
+    datasetList.value = res.data
   })
 }
 

From 3558a54e249fc954acf8feb1069e77de36855536 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Tue, 7 May 2024 19:52:55 +0800
Subject: [PATCH 6/9] =?UTF-8?q?feat:=20=E6=AE=B5=E8=90=BD=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E6=89=B9=E9=87=8F=E9=80=89=E6=8B=A9=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/common/event/listener_manage.py           | 18 ++++++++++--------
 .../serializers/document_serializers.py        |  6 ++++--
 .../serializers/paragraph_serializers.py       | 15 +++++++--------
 apps/embedding/vector/base_vector.py           |  4 ++++
 apps/embedding/vector/pg_vector.py             |  3 +++
 5 files changed, 28 insertions(+), 18 deletions(-)

diff --git a/apps/common/event/listener_manage.py b/apps/common/event/listener_manage.py
index 3204e662bb7..415d20562ec 100644
--- a/apps/common/event/listener_manage.py
+++ b/apps/common/event/listener_manage.py
@@ -51,15 +51,16 @@ def __init__(self, problem_id: str, problem_content: str):
 
 
 class UpdateEmbeddingDatasetIdArgs:
-    def __init__(self, source_id_list: List[str], target_dataset_id: str):
-        self.source_id_list = source_id_list
+    def __init__(self, paragraph_id_list: List[str], target_dataset_id: str):
+        self.paragraph_id_list = paragraph_id_list
         self.target_dataset_id = target_dataset_id
 
 
 class UpdateEmbeddingDocumentIdArgs:
-    def __init__(self, source_id_list: List[str], target_document_id: str):
-        self.source_id_list = source_id_list
+    def __init__(self, paragraph_id_list: List[str], target_document_id: str, target_dataset_id: str):
+        self.paragraph_id_list = paragraph_id_list
         self.target_document_id = target_document_id
+        self.target_dataset_id = target_dataset_id
 
 
 class ListenerManagement:
@@ -219,13 +220,14 @@ def update_problem(args: UpdateProblemArgs):
 
     @staticmethod
     def update_embedding_dataset_id(args: UpdateEmbeddingDatasetIdArgs):
-        VectorStore.get_embedding_vector().update_by_source_ids(args.source_id_list,
-                                                                {'dataset_id': args.target_dataset_id})
+        VectorStore.get_embedding_vector().update_by_paragraph_ids(args.paragraph_id_list,
+                                                                   {'dataset_id': args.target_dataset_id})
 
     @staticmethod
     def update_embedding_document_id(args: UpdateEmbeddingDocumentIdArgs):
-        VectorStore.get_embedding_vector().update_by_source_ids(args.source_id_list,
-                                                                {'document_id': args.target_document_id})
+        VectorStore.get_embedding_vector().update_by_paragraph_ids(args.paragraph_id_list,
+                                                                   {'document_id': args.target_document_id,
+                                                                    'dataset_id': args.target_dataset_id})
 
     @staticmethod
     def delete_embedding_by_source_ids(source_ids: List[str]):
diff --git a/apps/dataset/serializers/document_serializers.py b/apps/dataset/serializers/document_serializers.py
index 919172d139d..3da185d77b3 100644
--- a/apps/dataset/serializers/document_serializers.py
+++ b/apps/dataset/serializers/document_serializers.py
@@ -166,10 +166,12 @@ def migrate(self, with_valid=True):
                                      meta={})
             else:
                 document_list.update(dataset_id=target_dataset_id)
-            paragraph_list.update(dataset_id=target_dataset_id)
+            # 修改向量信息
             ListenerManagement.update_embedding_dataset_id(UpdateEmbeddingDatasetIdArgs(
-                [problem_paragraph_mapping.id for problem_paragraph_mapping in problem_paragraph_mapping_list],
+                [paragraph.id for paragraph in paragraph_list],
                 target_dataset_id))
+            # 修改段落信息
+            paragraph_list.update(dataset_id=target_dataset_id)
 
         @staticmethod
         def get_target_dataset_problem(target_dataset_id: str,
diff --git a/apps/dataset/serializers/paragraph_serializers.py b/apps/dataset/serializers/paragraph_serializers.py
index 84b5fd7ed0d..589710ce09d 100644
--- a/apps/dataset/serializers/paragraph_serializers.py
+++ b/apps/dataset/serializers/paragraph_serializers.py
@@ -334,9 +334,11 @@ def migrate(self, with_valid=True):
                     # 修改mapping
                     QuerySet(ProblemParagraphMapping).bulk_update(problem_paragraph_mapping_list,
                                                                   ['document_id'])
+                # 修改向量段落信息
                 ListenerManagement.update_embedding_document_id(UpdateEmbeddingDocumentIdArgs(
                     [paragraph.id for paragraph in paragraph_list],
-                    target_document_id))
+                    target_document_id, target_dataset_id))
+                # 修改段落信息
                 paragraph_list.update(document_id=target_document_id)
             # 不同数据集迁移
             else:
@@ -362,14 +364,11 @@ def migrate(self, with_valid=True):
                 # 修改mapping
                 QuerySet(ProblemParagraphMapping).bulk_update(problem_paragraph_mapping_list,
                                                               ['problem_id', 'dataset_id', 'document_id'])
+                # 修改向量段落信息
                 ListenerManagement.update_embedding_document_id(UpdateEmbeddingDocumentIdArgs(
-                    [*[problem_paragraph_mapping.id for problem_paragraph_mapping in problem_paragraph_mapping_list]
-                        , *[paragraph.id for paragraph in paragraph_list]],
-                    target_document_id))
-                ListenerManagement.update_embedding_dataset_id(UpdateEmbeddingDatasetIdArgs(
-                    [*[problem_paragraph_mapping.id for problem_paragraph_mapping in problem_paragraph_mapping_list]
-                        , *[paragraph.id for paragraph in paragraph_list]],
-                    target_dataset_id))
+                    [paragraph.id for paragraph in paragraph_list],
+                    target_document_id, target_dataset_id))
+                # 修改段落信息
                 paragraph_list.update(dataset_id=target_dataset_id, document_id=target_document_id)
 
         @staticmethod
diff --git a/apps/embedding/vector/base_vector.py b/apps/embedding/vector/base_vector.py
index 4c42319b97c..2bfd0e977d5 100644
--- a/apps/embedding/vector/base_vector.py
+++ b/apps/embedding/vector/base_vector.py
@@ -130,6 +130,10 @@ def hit_test(self, query_text, dataset_id: list[str], exclude_document_id_list:
     def update_by_paragraph_id(self, paragraph_id: str, instance: Dict):
         pass
 
+    @abstractmethod
+    def update_by_paragraph_ids(self, paragraph_ids: str, instance: Dict):
+        pass
+
     @abstractmethod
     def update_by_source_id(self, source_id: str, instance: Dict):
         pass
diff --git a/apps/embedding/vector/pg_vector.py b/apps/embedding/vector/pg_vector.py
index 8665c01ef0d..5c0d045363b 100644
--- a/apps/embedding/vector/pg_vector.py
+++ b/apps/embedding/vector/pg_vector.py
@@ -119,6 +119,9 @@ def update_by_source_id(self, source_id: str, instance: Dict):
     def update_by_paragraph_id(self, paragraph_id: str, instance: Dict):
         QuerySet(Embedding).filter(paragraph_id=paragraph_id).update(**instance)
 
+    def update_by_paragraph_ids(self, paragraph_id: str, instance: Dict):
+        QuerySet(Embedding).filter(paragraph_id__in=paragraph_id).update(**instance)
+
     def delete_by_dataset_id(self, dataset_id: str):
         QuerySet(Embedding).filter(dataset_id=dataset_id).delete()
 

From a6c06ea82e49365c3fe092e99e6e78a31e0fb1cf Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Wed, 8 May 2024 10:08:30 +0800
Subject: [PATCH 7/9] =?UTF-8?q?feat:=20=E7=9F=A5=E8=AF=86=E5=BA=93?=
 =?UTF-8?q?=E5=88=86=E6=AE=B5=E8=BF=81=E7=A7=BB=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/views/paragraph/index.vue | 34 +++++++++++++++++++++++++-------
 1 file changed, 27 insertions(+), 7 deletions(-)

diff --git a/ui/src/views/paragraph/index.vue b/ui/src/views/paragraph/index.vue
index 66c70f74def..468fe9f68a9 100644
--- a/ui/src/views/paragraph/index.vue
+++ b/ui/src/views/paragraph/index.vue
@@ -107,11 +107,25 @@
                   <template #footer>
                     <div class="footer-content flex-between">
                       <span> {{ numberFormat(item?.content.length) || 0 }} 个 字符 </span>
-                      <el-tooltip effect="dark" content="删除" placement="top">
-                        <el-button text @click.stop="deleteParagraph(item)" class="delete-button">
-                          <el-icon><Delete /></el-icon>
-                        </el-button>
-                      </el-tooltip>
+
+                      <span @click.stop>
+                        <el-dropdown trigger="click">
+                          <el-button text>
+                            <el-icon><MoreFilled /></el-icon>
+                          </el-button>
+                          <template #dropdown>
+                            <el-dropdown-menu>
+                              <el-dropdown-item @click="openSelectDocumentDialog(item)">
+                                <AppIcon iconName="app-migrate"></AppIcon>
+                                迁移</el-dropdown-item
+                              >
+                              <el-dropdown-item icon="Delete" @click.stop="deleteParagraph(item)"
+                                >删除</el-dropdown-item
+                              >
+                            </el-dropdown-menu>
+                          </template>
+                        </el-dropdown>
+                      </span>
                     </div>
                   </template>
                 </CardBox>
@@ -180,8 +194,14 @@ function refreshMigrateParagraph() {
   MsgSuccess('迁移删除成功')
 }
 
-function openSelectDocumentDialog() {
-  SelectDocumentDialogRef.value.open(multipleSelection.value)
+function openSelectDocumentDialog(row?: any) {
+  let arr: string[] = []
+  if (row) {
+    arr.push(row.id)
+  } else {
+    arr = multipleSelection.value
+  }
+  SelectDocumentDialogRef.value.open(arr)
 }
 function deleteMulParagraph() {
   MsgConfirm(

From f579dc7df708757b1b66ad1812c6c7bd9e2ad9bf Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Wed, 8 May 2024 10:28:23 +0800
Subject: [PATCH 8/9] =?UTF-8?q?feat:=20=E7=9F=A5=E8=AF=86=E5=BA=93?=
 =?UTF-8?q?=E5=88=86=E6=AE=B5=E8=BF=81=E7=A7=BB=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/views/paragraph/index.vue | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/ui/src/views/paragraph/index.vue b/ui/src/views/paragraph/index.vue
index 468fe9f68a9..1f993edc4c1 100644
--- a/ui/src/views/paragraph/index.vue
+++ b/ui/src/views/paragraph/index.vue
@@ -195,13 +195,10 @@ function refreshMigrateParagraph() {
 }
 
 function openSelectDocumentDialog(row?: any) {
-  let arr: string[] = []
   if (row) {
-    arr.push(row.id)
-  } else {
-    arr = multipleSelection.value
+    multipleSelection.value = [row.id]
   }
-  SelectDocumentDialogRef.value.open(arr)
+  SelectDocumentDialogRef.value.open(multipleSelection.value)
 }
 function deleteMulParagraph() {
   MsgConfirm(

From ad7b2596a6936e25db41a9860bba3a56d957422b Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Wed, 8 May 2024 10:31:18 +0800
Subject: [PATCH 9/9] =?UTF-8?q?feat:=20=E7=9F=A5=E8=AF=86=E5=BA=93?=
 =?UTF-8?q?=E5=88=86=E6=AE=B5=E8=BF=81=E7=A7=BB=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/views/paragraph/index.vue | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/ui/src/views/paragraph/index.vue b/ui/src/views/paragraph/index.vue
index 1f993edc4c1..0307ab723a5 100644
--- a/ui/src/views/paragraph/index.vue
+++ b/ui/src/views/paragraph/index.vue
@@ -8,8 +8,12 @@
         >）</el-text
       >
       <div class="document-detail__header">
-        <el-button @click="isBatch = true" v-if="isBatch === false"> 批量选择 </el-button>
-        <el-button @click="cancelSelectedHandle" v-if="isBatch === true"> 取消选择 </el-button>
+        <el-button @click="batchSelectedHandle(true)" v-if="isBatch === false">
+          批量选择
+        </el-button>
+        <el-button @click="batchSelectedHandle(false)" v-if="isBatch === true">
+          取消选择
+        </el-button>
         <el-button
           @click="addParagraph"
           type="primary"
@@ -223,8 +227,8 @@ function deleteMulParagraph() {
     .catch(() => {})
 }
 
-function cancelSelectedHandle() {
-  isBatch.value = false
+function batchSelectedHandle(bool: boolean) {
+  isBatch.value = bool
   multipleSelection.value = []
 }