Return data_points in descriptive metric calculation task

topoteretes · alekszievr · Feb 3, 2025 · Jan 28, 2025 · Jan 28, 2025 · Jan 28, 2025
commit 44fa2cd1336c3b264e97007893500d6bd6f512b7
diff --git a/cognee/tasks/storage/descriptive_metrics.py b/cognee/tasks/storage/descriptive_metrics.py
@@ -45,3 +45,4 @@ async def store_descriptive_metrics(data_points: list[DataPoint]):
     metrics_dict = {"id": uuid.uuid4(), "num_tokens": token_count_sum} | graph_metrics
 
     await db_engine.insert_data(table_name, metrics_dict)
+    return data_points
Original file line number	Diff line number	Diff line change
Expand Up		@@ -45,3 +45,4 @@ async def store_descriptive_metrics(data_points: list[DataPoint]):
		metrics_dict = {"id": uuid.uuid4(), "num_tokens": token_count_sum} \| graph_metrics

		await db_engine.insert_data(table_name, metrics_dict)
		return data_points