From 74bc7c942054c3203ad20639ef3c08c7d59aea5f Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Mon, 29 Sep 2025 21:22:21 +0200 Subject: [PATCH] refactor: set node_set to None for endpoint --- cognee/api/v1/add/routers/get_add_router.py | 6 +++++- cognee/api/v1/update/routers/get_update_router.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/cognee/api/v1/add/routers/get_add_router.py b/cognee/api/v1/add/routers/get_add_router.py index dfa7d275b..4d0063cc9 100644 --- a/cognee/api/v1/add/routers/get_add_router.py +++ b/cognee/api/v1/add/routers/get_add_router.py @@ -73,7 +73,11 @@ def get_add_router() -> APIRouter: try: add_run = await cognee_add( - data, datasetName, user=user, dataset_id=datasetId, node_set=node_set + data, + datasetName, + user=user, + dataset_id=datasetId, + node_set=node_set if node_set else None, ) if isinstance(add_run, PipelineRunErrored): diff --git a/cognee/api/v1/update/routers/get_update_router.py b/cognee/api/v1/update/routers/get_update_router.py index 144bdd9e6..4101e1e31 100644 --- a/cognee/api/v1/update/routers/get_update_router.py +++ b/cognee/api/v1/update/routers/get_update_router.py @@ -75,7 +75,7 @@ def get_update_router() -> APIRouter: data=data, dataset_id=dataset_id, user=user, - node_set=node_set, + node_set=node_set if node_set else None, ) # If any cognify run errored return JSONResponse with proper error status code