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

chore: remove unused imports, add optional for delete dataset statement

Browse source
This commit is contained in:
Igor Ilic 2025-12-02 13:09:45 +01:00
parent 5cfc7b1761
commit dbcb35a6da
2 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py
View file

@ -68,7 +68,7 @@ class DatasetDatabaseHandlerInterface(ABC):
@classmethod @classmethod
@abstractmethod @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. 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. 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.

3
cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py
View file

@ -1,6 +1,3 @@
import os
import asyncio
import requests
from uuid import UUID from uuid import UUID
from typing import Optional from typing import Optional