From 0a11fe0371e12ea515043afea1d8939a61bde288 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Thu, 19 Jun 2025 18:08:54 +0200 Subject: [PATCH] feat: Add dataset_id return value from add endpoint --- cognee/api/v1/add/routers/get_add_router.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/cognee/api/v1/add/routers/get_add_router.py b/cognee/api/v1/add/routers/get_add_router.py index 53741ba41..c792883bf 100644 --- a/cognee/api/v1/add/routers/get_add_router.py +++ b/cognee/api/v1/add/routers/get_add_router.py @@ -10,6 +10,7 @@ import requests from cognee.modules.users.models import User from cognee.modules.users.methods import get_authenticated_user +from cognee.modules.data.methods import get_unique_dataset_id logger = get_logger() @@ -52,7 +53,14 @@ def get_add_router() -> APIRouter: else: add_run = await cognee_add(data, datasetName, user=user, dataset_id=datasetId) - return add_run.model_dump() + # Create response json with add function return value + response_json = add_run.model_dump() + + # Add dataset_id to json response + dataset_id = await get_unique_dataset_id(datasetName, user=user) + response_json["dataset_id"] = dataset_id + + return response_json except Exception as error: return JSONResponse(status_code=409, content={"error": str(error)})