From fabbd638a4ef00391f313f8b57df85ea25381f30 Mon Sep 17 00:00:00 2001 From: hajdul88 <52442977+hajdul88@users.noreply.github.com> Date: Wed, 13 Aug 2025 15:03:12 +0200 Subject: [PATCH] chore: renaming error --- cognee/modules/data/exceptions/exceptions.py | 4 ++-- cognee/modules/data/methods/delete_data.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/cognee/modules/data/exceptions/exceptions.py b/cognee/modules/data/exceptions/exceptions.py index c85e11cb2..ac3b68e64 100644 --- a/cognee/modules/data/exceptions/exceptions.py +++ b/cognee/modules/data/exceptions/exceptions.py @@ -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) diff --git a/cognee/modules/data/methods/delete_data.py b/cognee/modules/data/methods/delete_data.py index 6ec055fcc..5425acac3 100644 --- a/cognee/modules/data/methods/delete_data.py +++ b/cognee/modules/data/methods/delete_data.py @@ -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()