From dc637f70b0a826a50b5d5a4c2e3c3783f877a40a Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Mon, 18 Aug 2025 13:23:57 +0200
Subject: [PATCH] fix: fixes add datapoints params

---
 cognee/tasks/storage/add_data_points.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/tasks/storage/add_data_points.py b/cognee/tasks/storage/add_data_points.py
index 27054ca66..a29f5a5f7 100644
--- a/cognee/tasks/storage/add_data_points.py
+++ b/cognee/tasks/storage/add_data_points.py
@@ -42,7 +42,7 @@ async def add_data_points(
         - Optionally updates the edge index via `index_graph_edges`.
     """
     
-async def add_data_points(data_points: List[DataPoint]) -> List[DataPoint]:
+async def add_data_points(data_points: List[DataPoint], update_edge_collection: bool = True) -> List[DataPoint]:
     if not isinstance(data_points, list):
         raise InvalidDataPointsInAddDataPointsError("data_points must be a list.")
     if not all(isinstance(dp, DataPoint) for dp in data_points):