From dbcb35a6dafab18e8c2b61e120228c0e5fc2dae4 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 2 Dec 2025 13:09:45 +0100 Subject: [PATCH] chore: remove unused imports, add optional for delete dataset statement --- .../dataset_database_handler_interface.py | 2 +- .../databases/graph/kuzu/KuzuDatasetDatabaseHandler.py | 3 --- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py b/cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py index 01ee46c48..1811feb12 100644 --- a/cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py +++ b/cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py @@ -68,7 +68,7 @@ class DatasetDatabaseHandlerInterface(ABC): @classmethod @abstractmethod - async def delete_dataset(cls, dataset_id: UUID, user: User) -> None: + async def delete_dataset(cls, dataset_id: Optional[UUID], user: Optional[User]) -> None: """ Delete the graph or vector database for the given dataset. Function should auto handle deleting of the actual database or send a request to the proper service to delete/mark the database as not needed for the given dataset. diff --git a/cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py b/cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py index a2b2da8f4..242249e00 100644 --- a/cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py +++ b/cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py @@ -1,6 +1,3 @@ -import os -import asyncio -import requests from uuid import UUID from typing import Optional