From 126ca8a30685c30e69700eb516d6bbb0a8506706 Mon Sep 17 00:00:00 2001 From: Daulet Amirkhanov Date: Mon, 1 Sep 2025 13:07:38 +0100 Subject: [PATCH] ruff format --- cognee/api/v1/cognify/routers/get_cognify_router.py | 4 +--- .../api/v1/datasets/routers/get_datasets_router.py | 12 +++--------- .../v1/permissions/routers/get_permissions_router.py | 4 +--- cognee/api/v1/search/routers/get_search_router.py | 4 +--- 4 files changed, 6 insertions(+), 18 deletions(-) diff --git a/cognee/api/v1/cognify/routers/get_cognify_router.py b/cognee/api/v1/cognify/routers/get_cognify_router.py index 31873632c..6809f089a 100644 --- a/cognee/api/v1/cognify/routers/get_cognify_router.py +++ b/cognee/api/v1/cognify/routers/get_cognify_router.py @@ -46,9 +46,7 @@ def get_cognify_router() -> APIRouter: router = APIRouter() @router.post("", response_model=dict) - async def cognify( - payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user) - ): + async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user)): """ Transform datasets into structured knowledge graphs through cognitive processing. diff --git a/cognee/api/v1/datasets/routers/get_datasets_router.py b/cognee/api/v1/datasets/routers/get_datasets_router.py index d43cd166d..ff310e4b4 100644 --- a/cognee/api/v1/datasets/routers/get_datasets_router.py +++ b/cognee/api/v1/datasets/routers/get_datasets_router.py @@ -176,9 +176,7 @@ def get_datasets_router() -> APIRouter: @router.delete( "/{dataset_id}", response_model=None, responses={404: {"model": ErrorResponseDTO}} ) - async def delete_dataset( - dataset_id: UUID, user: User = Depends(get_authenticated_user) - ): + async def delete_dataset(dataset_id: UUID, user: User = Depends(get_authenticated_user)): """ Delete a dataset by its ID. @@ -266,9 +264,7 @@ def get_datasets_router() -> APIRouter: await delete_data(data) @router.get("/{dataset_id}/graph", response_model=GraphDTO) - async def get_dataset_graph( - dataset_id: UUID, user: User = Depends(get_authenticated_user) - ): + async def get_dataset_graph(dataset_id: UUID, user: User = Depends(get_authenticated_user)): """ Get the knowledge graph visualization for a dataset. @@ -298,9 +294,7 @@ def get_datasets_router() -> APIRouter: response_model=list[DataDTO], responses={404: {"model": ErrorResponseDTO}}, ) - async def get_dataset_data( - dataset_id: UUID, user: User = Depends(get_authenticated_user) - ): + async def get_dataset_data(dataset_id: UUID, user: User = Depends(get_authenticated_user)): """ Get all data items in a dataset. diff --git a/cognee/api/v1/permissions/routers/get_permissions_router.py b/cognee/api/v1/permissions/routers/get_permissions_router.py index 7a2cdfeaa..89603ac46 100644 --- a/cognee/api/v1/permissions/routers/get_permissions_router.py +++ b/cognee/api/v1/permissions/routers/get_permissions_router.py @@ -183,9 +183,7 @@ def get_permissions_router() -> APIRouter: return JSONResponse(status_code=200, content={"message": "User added to tenant"}) @permissions_router.post("/tenants") - async def create_tenant( - tenant_name: str, user: User = Depends(get_authenticated_user) - ): + async def create_tenant(tenant_name: str, user: User = Depends(get_authenticated_user)): """ Create a new tenant. diff --git a/cognee/api/v1/search/routers/get_search_router.py b/cognee/api/v1/search/routers/get_search_router.py index ea60e59e3..0ceeb1abb 100644 --- a/cognee/api/v1/search/routers/get_search_router.py +++ b/cognee/api/v1/search/routers/get_search_router.py @@ -66,9 +66,7 @@ def get_search_router() -> APIRouter: return JSONResponse(status_code=500, content={"error": str(error)}) @router.post("", response_model=list) - async def search( - payload: SearchPayloadDTO, user: User = Depends(get_authenticated_user) - ): + async def search(payload: SearchPayloadDTO, user: User = Depends(get_authenticated_user)): """ Search for nodes in the graph database.