Merge pull request #700 from apecloud/add_rerun_buttom_for_doc_indexing

feat: add rerun button for doc indexing

Author: nayutah

aperag/tasks/index.py

@@ -300,7 +300,7 @@ def update_index_for_document(self, document_id):
     document.save()
 
     try:
-        relate_ids = json.loads(document.relate_ids)
+        relate_ids = json.loads(document.relate_ids) if document.relate_ids.strip() else {}
         source = get_source(json.loads(document.collection.config))
         metadata = json.loads(document.metadata)
         local_doc = source.prepare_document(name=document.name, metadata=metadata)

aperag/views/main.py

@@ -689,7 +689,7 @@ async def update_document(
     # if user add labels for a document, we need to update index
     update_index_for_document.delay(instance.id)
 
-    related_questions = await sync_to_async(document.question_set.exclude)(status=QuestionStatus.DELETED)
+    related_questions = await sync_to_async(instance.question_set.exclude)(status=QuestionStatus.DELETED)
     async for question in related_questions:
         question.status = QuestionStatus.WARNING
         await question.asave()