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) super().__init__(message, name, status_code)
class InvalidAttributeError(CogneeValidationError): class InvalidTableAttributeError(CogneeValidationError):
def __init__( def __init__(
self, self,
message: str = "The provided data object is missing the required '__tablename__' attribute.", 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, status_code: int = status.HTTP_400_BAD_REQUEST,
): ):
super().__init__(message, name, status_code) 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.modules.data.models import Data
from cognee.infrastructure.databases.relational import get_relational_engine 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. ValueError: If the data object is invalid.
""" """
if not hasattr(data, "__tablename__"): if not hasattr(data, "__tablename__"):
raise InvalidAttributeError() raise InvalidTableAttributeError()
db_engine = get_relational_engine() db_engine = get_relational_engine()