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

2024-05-07 13:50:39 +08:00 · 2024-05-07 13:50:39 +08:00 · 77d71f95a3
commit 77d71f95a3
parent c1b6ec630c
1 changed files with 3 additions and 0 deletions
--- a/apps/dataset/serializers/document_serializers.py
+++ b/apps/dataset/serializers/document_serializers.py
@ -387,6 +387,9 @@ class DocumentSerializers(ApiMixin, serializers.Serializer):
                                        {})

            else:
+                if document.status != Status.embedding.value:
+                    document.status = Status.embedding
+                    document.save()
                ListenerManagement.embedding_by_document_signal.send(document_id)
            return True