gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: Add dataset_id return value from add endpoint

Browse source
This commit is contained in:
Igor Ilic 2025-06-19 18:08:54 +02:00
parent 63e9266d28
commit 0a11fe0371
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
cognee/api/v1/add/routers/get_add_router.py
View file

@ -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)})