742540929c
Updated docs to functions related to getting and deleting Data model objects Docs #COG-334
19 lines
615 B
Python
19 lines
615 B
Python
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 ValueError("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)
|