From e07240784aeacbc457a4440bc1a4d5ec631a227f Mon Sep 17 00:00:00 2001 From: Anushree2005 Date: Sat, 27 Dec 2025 18:01:25 +0530 Subject: [PATCH] feat(api): include label in dataset data response --- cognee/api/v1/datasets/routers/get_datasets_router.py | 1 + cognee/modules/data/models/Data.py | 1 + 2 files changed, 2 insertions(+) diff --git a/cognee/api/v1/datasets/routers/get_datasets_router.py b/cognee/api/v1/datasets/routers/get_datasets_router.py index ca738dfbe..22866c30f 100644 --- a/cognee/api/v1/datasets/routers/get_datasets_router.py +++ b/cognee/api/v1/datasets/routers/get_datasets_router.py @@ -50,6 +50,7 @@ class DataDTO(OutDTO): mime_type: str raw_data_location: str dataset_id: UUID + label: Optional[str] = None class GraphNodeDTO(OutDTO): diff --git a/cognee/modules/data/models/Data.py b/cognee/modules/data/models/Data.py index 8a3d5e218..76b600100 100644 --- a/cognee/modules/data/models/Data.py +++ b/cognee/modules/data/models/Data.py @@ -56,5 +56,6 @@ class Data(Base): "createdAt": self.created_at.isoformat(), "updatedAt": self.updated_at.isoformat() if self.updated_at else None, "nodeSet": self.node_set, + "label": self.label, # "datasets": [dataset.to_json() for dataset in self.datasets] }