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

chore: renaming error

Browse source
This commit is contained in:
hajdul88 2025-08-13 15:03:12 +02:00
parent d1bfeaa0f2
commit fabbd638a4
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/modules/data/exceptions/exceptions.py
View file

@ -45,11 +45,11 @@ class DatasetTypeError(CogneeValidationError):
super().__init__(message, name, status_code)
class InvalidAttributeError(CogneeValidationError):
class InvalidTableAttributeError(CogneeValidationError):
def __init__(
self,
message: str = "The provided data object is missing the required '__tablename__' attribute.",
name: str = "InvalidAttributeError",
name: str = "InvalidTableAttributeError",
status_code: int = status.HTTP_400_BAD_REQUEST,
):
super().__init__(message, name, status_code)

4
cognee/modules/data/methods/delete_data.py
View file

@ -1,4 +1,4 @@
from cognee.modules.data.exceptions.exceptions import InvalidAttributeError
from cognee.modules.data.exceptions.exceptions import InvalidTableAttributeError
from cognee.modules.data.models import Data
from cognee.infrastructure.databases.relational import get_relational_engine
@ -13,7 +13,7 @@ async def delete_data(data: Data):
ValueError: If the data object is invalid.
"""
if not hasattr(data, "__tablename__"):
raise InvalidAttributeError()
raise InvalidTableAttributeError()
db_engine = get_relational_engine()