From 0f2d8393196829dee1d0918eaf15594d643b3567 Mon Sep 17 00:00:00 2001 From: Igor Ilic <30923996+dexters1@users.noreply.github.com> Date: Fri, 18 Jul 2025 14:00:20 +0200 Subject: [PATCH] fix: Resolve issue with dataset status endpoint (#1103) ## Description ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin. --- .../api/v1/datasets/routers/get_datasets_router.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/cognee/api/v1/datasets/routers/get_datasets_router.py b/cognee/api/v1/datasets/routers/get_datasets_router.py index bd9a6b4d9..6f09fd6d2 100644 --- a/cognee/api/v1/datasets/routers/get_datasets_router.py +++ b/cognee/api/v1/datasets/routers/get_datasets_router.py @@ -331,11 +331,19 @@ def get_datasets_router() -> APIRouter: ## Error Codes - **500 Internal Server Error**: Error retrieving status information """ - from cognee.modules.data.methods import get_dataset_status + from cognee.api.v1.datasets.datasets import datasets as cognee_datasets - dataset_status = await get_dataset_status(datasets, user.id) + try: + # Verify user has permission to read dataset + authorized_datasets = await get_authorized_existing_datasets(datasets, "read", user) - return dataset_status + datasets_statuses = await cognee_datasets.get_status( + [dataset.id for dataset in authorized_datasets] + ) + + return datasets_statuses + except Exception as error: + return JSONResponse(status_code=409, content={"error": str(error)}) @router.get("/{dataset_id}/data/{data_id}/raw", response_class=FileResponse) async def get_raw_data(