fix: dataset error

api/controllers/console/datasets/datasets_document.py

@@ -896,13 +896,13 @@ class DocumentRetryApi(DocumentResource):
                             location='json')
         args = parser.parse_args()
         dataset_id = str(dataset_id)
+        dataset = DatasetService.get_dataset(dataset_id)
+        if not dataset:
+            raise NotFound('Dataset not found.')
         for document_id in args['document_ids']:
             try:
                 document_id = str(document_id)
-                dataset = DatasetService.get_dataset(dataset_id)
-                if not dataset:
-                    raise NotFound('Dataset not found.')
-
+                
                 document = DocumentService.get_document(dataset.id, document_id)
 
                 # 404 if document not found

api/services/dataset_service.py

@@ -411,7 +411,7 @@ class DocumentService:
     def get_error_documents_by_dataset_id(dataset_id: str) -> list[Document]:
         documents = db.session.query(Document).filter(
             Document.dataset_id == dataset_id,
-            Document.indexing_status == 'error' or Document.indexing_status == 'paused'
+            Document.indexing_status == 'error' or Document.indexing_status == 'paused' or Document.indexing_status == 'waiting'
         ).all()
 
         return documents
@@ -490,7 +490,7 @@ class DocumentService:
         retry_indexing_cache_key = 'document_{}_is_retried'.format(document.id)
         redis_client.setex(retry_indexing_cache_key, 600, 1)
         # trigger async task
-        retry_document_indexing_task.delay(document.dataset_id, [document.id])
+        retry_document_indexing_task.delay(document.dataset_id, document.id)
 
     @staticmethod
     def get_documents_position(dataset_id):

web/app/components/datasets/documents/list.tsx

@@ -323,7 +323,7 @@ const DocumentList: FC<IDocumentListProps> = ({ embeddingAvailable, documents =
             <td className='w-12'>#</td>
             <td>{t('datasetDocuments.list.table.header.fileName')}</td>
             <td className='w-24'>{t('datasetDocuments.list.table.header.words')}</td>
-            <td className='w-24'>{t('datasetDocuments.list.table.header.hitCount')}</td>
+            <td className='w-44'>{t('datasetDocuments.list.table.header.hitCount')}</td>
             <td className='w-44'>
               <div className='flex justify-between items-center'>
                 {t('datasetDocuments.list.table.header.uploadTime')}