From bc57f173375dad01120bc771adfd43f59b098730 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Mon, 22 Apr 2024 14:45:53 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=88=86=E8=AF=8D=E6=A3=80=E7=B4=A2?=
 =?UTF-8?q?=E5=8E=86=E5=8F=B2=E6=95=B0=E6=8D=AE=E5=A4=84=E7=90=86=20(#203)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/embedding/migrations/0002_embedding_search_vector.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/apps/embedding/migrations/0002_embedding_search_vector.py b/apps/embedding/migrations/0002_embedding_search_vector.py
index e0cbe733b..3ed58d582 100644
--- a/apps/embedding/migrations/0002_embedding_search_vector.py
+++ b/apps/embedding/migrations/0002_embedding_search_vector.py
@@ -31,12 +31,14 @@ def save_keywords(apps, schema_editor):
                                                                                             'paragraph')
         embedding_update_list = [update_embedding_search_vector(embedding, paragraph_list) for embedding
                                  in embedding_list]
-        child_array = sub_array(embedding_update_list, 20)
+        child_array = sub_array(embedding_update_list, 50)
         for c in child_array:
             try:
                 embedding.objects.using(db_alias).bulk_update(c, ['search_vector'])
             except Exception as e:
                 print(e)
+        document.status = Status.success
+        document.save()
 
 
 class Migration(migrations.Migration):