From 3da11564e2cda9e32e73727579c254f9b9acbf13 Mon Sep 17 00:00:00 2001 From: Anushree2005 Date: Sat, 27 Dec 2025 17:58:33 +0530 Subject: [PATCH] feat(ingestion): persist optional label when provided --- cognee/tasks/ingestion/ingest_data.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/cognee/tasks/ingestion/ingest_data.py b/cognee/tasks/ingestion/ingest_data.py index 5987f38d5..ef17b04b1 100644 --- a/cognee/tasks/ingestion/ingest_data.py +++ b/cognee/tasks/ingestion/ingest_data.py @@ -120,6 +120,9 @@ async def ingest_data( # TODO: Maybe allow getting of external metadata through ingestion loader? ext_metadata = get_external_metadata_dict(data_item) + # Extract optional label from data_item if present + label = getattr(data_item, "label", None) + if node_set: ext_metadata["node_set"] = node_set @@ -138,6 +141,7 @@ async def ingest_data( data_point.file_size = original_file_metadata["file_size"] data_point.external_metadata = ext_metadata data_point.node_set = json.dumps(node_set) if node_set else None + data_point.label = label data_point.tenant_id = user.tenant_id if user.tenant_id else None # Check if data is already in dataset @@ -167,6 +171,7 @@ async def ingest_data( node_set=json.dumps(node_set) if node_set else None, data_size=original_file_metadata["file_size"], tenant_id=user.tenant_id if user.tenant_id else None, + label=label, pipeline_status={}, token_count=-1, )