diff --git a/backend/data_export/pipeline/repositories.py b/backend/data_export/pipeline/repositories.py index 71420b7c..1423e235 100644 --- a/backend/data_export/pipeline/repositories.py +++ b/backend/data_export/pipeline/repositories.py @@ -205,3 +205,11 @@ class IntentDetectionSlotFillingRepository(TextRepository): label.setdefault("cats", []) label.setdefault("entities", []) return label_per_user + + def reduce_user(self, label_per_user: Dict[str, Any]): + cats = [] + entities = [] + for user, label in label_per_user.items(): + cats.extend(label.get("cats", [])) + entities.extend(label.get("entities", [])) + return {"all": {"entities": entities, "cats": cats}}