gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(api): include label in dataset data response

Browse source
This commit is contained in:
Anushree2005 2025-12-27 18:01:25 +05:30
parent 3da11564e2
commit e07240784a
2 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cognee/api/v1/datasets/routers/get_datasets_router.py
View file

@ -50,6 +50,7 @@ class DataDTO(OutDTO):
mime_type: str mime_type: str
raw_data_location: str raw_data_location: str
dataset_id: UUID dataset_id: UUID
label: Optional[str] = None
class GraphNodeDTO(OutDTO): class GraphNodeDTO(OutDTO):

1
cognee/modules/data/models/Data.py
View file

@ -56,5 +56,6 @@ class Data(Base):
"createdAt": self.created_at.isoformat(), "createdAt": self.created_at.isoformat(),
"updatedAt": self.updated_at.isoformat() if self.updated_at else None, "updatedAt": self.updated_at.isoformat() if self.updated_at else None,
"nodeSet": self.node_set, "nodeSet": self.node_set,
"label": self.label,
# "datasets": [dataset.to_json() for dataset in self.datasets] # "datasets": [dataset.to_json() for dataset in self.datasets]
} }