fix: 修改文档向量化重试,文档状态处理 (1Panel-dev#373)

lunf · May 7, 2024 · 77d71f9 · 77d71f9
1 parent c1b6ec6
commit 77d71f9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/apps/dataset/serializers/document_serializers.py b/apps/dataset/serializers/document_serializers.py
@@ -387,6 +387,9 @@ def refresh(self, with_valid=True):
                                         {})
 
             else:
+                if document.status != Status.embedding.value:
+                    document.status = Status.embedding
+                    document.save()
                 ListenerManagement.embedding_by_document_signal.send(document_id)
             return True