From 9750a550a5afb952d8010fd4a88f630fcad7e069 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Wed, 20 Dec 2023 15:23:59 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=88=9B=E5=BB=BA=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E9=9B=86=E5=90=91=E9=87=8F=E5=8C=96,=E4=BB=A5=E6=96=87?=
 =?UTF-8?q?=E6=A1=A3=E7=9A=84=E7=B2=92=E5=BA=A6=E5=90=91=E9=87=8F=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/common/event/listener_manage.py | 20 +++-----------------
 ui/src/views/document/index.vue      |  4 ++--
 2 files changed, 5 insertions(+), 19 deletions(-)

diff --git a/apps/common/event/listener_manage.py b/apps/common/event/listener_manage.py
index b98a6cc2d..a73a53ef4 100644
--- a/apps/common/event/listener_manage.py
+++ b/apps/common/event/listener_manage.py
@@ -96,23 +96,9 @@ class ListenerManagement:
         :param dataset_id: 知识库id
         :return: None
         """
-        status = Status.success
-        try:
-            data_list = native_search(
-                {'problem': QuerySet(get_dynamics_model({'problem.dataset_id': django.db.models.CharField()})).filter(
-                    **{'problem.dataset_id': dataset_id}),
-                    'paragraph': QuerySet(Paragraph).filter(dataset_id=dataset_id)},
-                select_string=get_file_content(
-                    os.path.join(PROJECT_DIR, "apps", "common", 'sql', 'list_embedding_text.sql')))
-            # 删除知识库相关向量数据
-            VectorStore.get_embedding_vector().delete_by_dataset_id(dataset_id)
-            # 批量向量化
-            VectorStore.get_embedding_vector().batch_save(data_list)
-        except Exception as e:
-            status = Status.error
-        #  修改文档 以及段落的状态
-        QuerySet(Document).filter(dataset_id=dataset_id).update(**{'status': status})
-        QuerySet(Paragraph).filter(dataset_id=dataset_id).update(**{'status': status})
+        document_list = QuerySet(Document).filter(dataset_id=dataset_id)
+        for document in document_list:
+            ListenerManagement.embedding_by_document(document.id)
 
     @staticmethod
     def delete_embedding_by_document(document_id):
diff --git a/ui/src/views/document/index.vue b/ui/src/views/document/index.vue
index f0c0697cf..99f7f777b 100644
--- a/ui/src/views/document/index.vue
+++ b/ui/src/views/document/index.vue
@@ -132,7 +132,7 @@ const paginationConfig = reactive({
 const initInterval = () => {
   interval = setInterval(() => {
     if (documentData.value.length > 0 && documentData.value.every((item) => item.status === '0')) {
-      getList()
+      getList(true)
     }
   }, 6000)
 }
@@ -238,7 +238,7 @@ function cellMouseLeave() {
 
 function handleSizeChange() {
   paginationConfig.current_page = 1
-  getList(true)
+  getList()
 }
 
 function getList(bool?: boolean) {