20 lines
686 B
Python
20 lines
686 B
Python
from cognee.exceptions import InvalidAttributeError
|
|
from cognee.modules.data.models import Data
|
|
from cognee.infrastructure.databases.relational import get_relational_engine
|
|
|
|
|
|
async def delete_data(data: Data):
|
|
"""Delete a data record from the database.
|
|
|
|
Args:
|
|
data (Data): The data object to be deleted.
|
|
|
|
Raises:
|
|
ValueError: If the data object is invalid.
|
|
"""
|
|
if not hasattr(data, '__tablename__'):
|
|
raise InvalidAttributeError(message="The provided data object is missing the required '__tablename__' attribute.")
|
|
|
|
db_engine = get_relational_engine()
|
|
|
|
return await db_engine.delete_data_by_id(data.__tablename__, data.id)
|