fix: Prevent compute metrics for draft datasets (#5624)

# Description
<!-- Please include a summary of the changes and the related issue.
Please also include relevant motivation and context. List any
dependencies that are required for this change. -->

Dataset progress and user metrics are not available for draft datasets.

**Type of change**
<!-- Please delete options that are not relevant. Remember to title the
PR according to the type of change -->

- Bug fix (non-breaking change which fixes an issue)

**How Has This Been Tested**
<!-- Please add some reference about how your feature has been tested.
-->

**Checklist**
<!-- Please go over the list and make sure you've taken everything into
account -->

- I added relevant documentation
- I followed the style guidelines of this project
- I did a self-review of my code
- I made corresponding changes to the documentation
- I confirm My changes generate no new warnings
- I have added tests that prove my fix is effective or that my feature
works
- I have added relevant notes to the CHANGELOG.md file (See
https://keepachangelog.com/)

Author: frascuchon

argilla-server/src/argilla_server/search_engine/commons.py

@@ -395,13 +395,19 @@ async def delete_record_suggestion(self, suggestion: Suggestion):
         )
 
     async def get_dataset_progress(self, dataset: Dataset) -> dict:
+        if dataset.is_draft:
+            return {}
+
         index_name = es_index_name_for_dataset(dataset)
 
         metrics = await self._compute_terms_metrics_for(index_name, "status")
 
         return {"total": metrics.total, **{metric.term: metric.count for metric in metrics.values}}
 
     async def get_dataset_user_progress(self, dataset: Dataset, user: User) -> dict:
+        if dataset.is_draft:
+            return {}
+
         index_name = es_index_name_for_dataset(dataset)
 
         result = await self._compute_terms_metrics_for(

argilla-server/tests/unit/search_engine/test_commons.py

@@ -28,6 +28,7 @@
     SimilarityOrder,
     RecordStatus,
     SortOrder,
+    DatasetStatus,
 )
 from argilla_server.models import Dataset, Question, Record, User, VectorSettings, Vector
 from argilla_server.search_engine import (
@@ -120,6 +121,7 @@ async def test_banking_sentiment_dataset_non_indexed():
             await FloatMetadataPropertyFactory.create(name="seq_float"),
         ],
         questions=[text_question, rating_question],
+        status=DatasetStatus.ready,
     )
 
     records = [
@@ -1352,6 +1354,13 @@ async def test_get_dataset_user_progress_with_response(
             "submitted": 1,
         }
 
+    async def test_get_dataset_user_progress_for_draft_dataset(self, search_engine: BaseElasticAndOpenSearchEngine):
+        dataset = await DatasetFactory.create(status=DatasetStatus.draft)
+        user = await UserFactory.create()
+
+        progress = await search_engine.get_dataset_user_progress(dataset, user=user)
+        assert progress == {}
+
     async def test_get_dataset_progress_with_pending_records(
         self,
         search_engine: BaseElasticAndOpenSearchEngine,
@@ -1388,6 +1397,11 @@ async def test_get_dataset_progress_with_completed_records(
         progress = await search_engine.get_dataset_progress(dataset)
         assert progress == {"total": len(records), "completed": len(records)}
 
+    async def test_get_dataset_progress_for_draft_dataset(self, search_engine: BaseElasticAndOpenSearchEngine):
+        dataset = await DatasetFactory.create(status=DatasetStatus.draft)
+        progress = await search_engine.get_dataset_progress(dataset)
+        assert progress == {}
+
     @pytest.mark.parametrize(
         ("property_name", "expected_metrics"),
         [