From ed555a731d2c904a5cf0c24240be565c973ffd88 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 11:27:45 +0200
Subject: [PATCH 01/34] feat: adds new Exception classes

---
 cognee/exceptions/__init__.py   |  7 +++---
 cognee/exceptions/exceptions.py | 43 +++++++++++++++++++++------------
 2 files changed, 31 insertions(+), 19 deletions(-)

diff --git a/cognee/exceptions/__init__.py b/cognee/exceptions/__init__.py
index d1d4ecbf5..496ad63c5 100644
--- a/cognee/exceptions/__init__.py
+++ b/cognee/exceptions/__init__.py
@@ -7,8 +7,7 @@ such as service failures, resource conflicts, and invalid operations.
 
 from .exceptions import (
     CogneeApiError,
-    ServiceError,
-    InvalidValueError,
-    InvalidAttributeError,
-    CriticalError,
+    CogneeSystemError,
+    CogneeValidationError,
+    CogneeConfigurationError,
 )
diff --git a/cognee/exceptions/exceptions.py b/cognee/exceptions/exceptions.py
index 46e0af1a8..e7eb784b8 100644
--- a/cognee/exceptions/exceptions.py
+++ b/cognee/exceptions/exceptions.py
@@ -35,37 +35,50 @@ class CogneeApiError(Exception):
         return f"{self.name}: {self.message} (Status code: {self.status_code})"
 
 
-class ServiceError(CogneeApiError):
-    """Failures in external services or APIs, like a database or a third-party service"""
+class CogneeSystemError(CogneeApiError):
+    """System error"""
 
     def __init__(
         self,
-        message: str = "Service is unavailable.",
-        name: str = "ServiceError",
+        message: str = "A system error occurred.",
+        name: str = "CogneeSystemError",
+        status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+    ):
+        super().__init__(message, name, status_code)
+
+
+class CogneeValidationError(CogneeApiError):
+    """Validation error"""
+
+    def __init__(
+        self,
+        message: str = "A validation error occurred.",
+        name: str = "CogneeValidationError",
         status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
     ):
         super().__init__(message, name, status_code)
 
 
-class InvalidValueError(CogneeApiError):
+class CogneeConfigurationError(CogneeApiError):
+    """SystemConfigError"""
+
     def __init__(
         self,
-        message: str = "Invalid Value.",
-        name: str = "InvalidValueError",
-        status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
+        message: str = "A system configuration error occurred.",
+        name: str = "CogneeConfigurationError",
+        status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
     ):
         super().__init__(message, name, status_code)
 
 
-class InvalidAttributeError(CogneeApiError):
+class CogneeTransientError(CogneeApiError):
+    """TransientError"""
+
     def __init__(
         self,
-        message: str = "Invalid attribute.",
-        name: str = "InvalidAttributeError",
-        status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
+        message: str = "A transient error occurred.",
+        name: str = "CogneeTransientError",
+        status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
     ):
         super().__init__(message, name, status_code)
 
-
-class CriticalError(CogneeApiError):
-    pass

From 0555dd9ddaf48ceec746e65a09adbc748828810c Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 11:29:16 +0200
Subject: [PATCH 02/34] Update __init__.py

---
 cognee/exceptions/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cognee/exceptions/__init__.py b/cognee/exceptions/__init__.py
index 496ad63c5..2f1589291 100644
--- a/cognee/exceptions/__init__.py
+++ b/cognee/exceptions/__init__.py
@@ -2,7 +2,7 @@
 Custom exceptions for the Cognee API.
 
 This module defines a set of exceptions for handling various application errors,
-such as service failures, resource conflicts, and invalid operations.
+such as System, Validation, Configuration or TransientErrors
 """
 
 from .exceptions import (
@@ -10,4 +10,5 @@ from .exceptions import (
     CogneeSystemError,
     CogneeValidationError,
     CogneeConfigurationError,
+    CogneeTransientError,
 )

From 1b3898dd8b2eb174e8f8e46fe4d4533dce6133f4 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 11:38:38 +0200
Subject: [PATCH 03/34] feat: updates rel_db exceptions with the new error
 classes

---
 .../databases/exceptions/exceptions.py         | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/cognee/infrastructure/databases/exceptions/exceptions.py b/cognee/infrastructure/databases/exceptions/exceptions.py
index 7a4220358..4aba09d37 100644
--- a/cognee/infrastructure/databases/exceptions/exceptions.py
+++ b/cognee/infrastructure/databases/exceptions/exceptions.py
@@ -1,13 +1,13 @@
 from fastapi import status
-from cognee.exceptions import CogneeApiError, CriticalError
+from cognee.exceptions import CogneeSystemError, CogneeValidationError, CogneeConfigurationError
 
 
-class DatabaseNotCreatedError(CriticalError):
+class DatabaseNotCreatedError(CogneeSystemError):
     """
     Represents an error indicating that the database has not been created. This error should
     be raised when an attempt is made to access the database before it has been initialized.
 
-    Inherits from CriticalError. Overrides the constructor to include a default message and
+    Inherits from CogneeSystemError. Overrides the constructor to include a default message and
     status code.
     """
 
@@ -20,10 +20,10 @@ class DatabaseNotCreatedError(CriticalError):
         super().__init__(message, name, status_code)
 
 
-class EntityNotFoundError(CogneeApiError):
+class EntityNotFoundError(CogneeValidationError):
     """
     Represents an error when a requested entity is not found in the database. This class
-    inherits from CogneeApiError.
+    inherits from CogneeValidationError.
 
     Public methods:
 
@@ -49,11 +49,11 @@ class EntityNotFoundError(CogneeApiError):
         # super().__init__(message, name, status_code) :TODO: This is not an error anymore with the dynamic exception handling therefore we shouldn't log error
 
 
-class EntityAlreadyExistsError(CogneeApiError):
+class EntityAlreadyExistsError(CogneeValidationError):
     """
     Represents an error when an entity creation is attempted but the entity already exists.
 
-    This class is derived from CogneeApiError and is used to signal a conflict in operations
+    This class is derived from CogneeValidationError and is used to signal a conflict in operations
     involving resource creation.
     """
 
@@ -66,11 +66,11 @@ class EntityAlreadyExistsError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class NodesetFilterNotSupportedError(CogneeApiError):
+class NodesetFilterNotSupportedError(CogneeConfigurationError):
     """
     Raise an exception when a nodeset filter is not supported by the current database.
 
-    This exception inherits from `CogneeApiError` and is designed to provide information
+    This exception inherits from `CogneeConfigurationError` and is designed to provide information
     about the specific issue of unsupported nodeset filters in the context of graph
     databases.
     """

From b40dbf3c68a2def746cd83a78e6d61408ef7c502 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 11:50:24 +0200
Subject: [PATCH 04/34] feat: adds new error classes to litellm instructor

---
 cognee/infrastructure/llm/exceptions.py       | 21 +++++++++++++++++--
 .../litellm_instructor/llm/get_llm_client.py  |  5 ++++-
 2 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/cognee/infrastructure/llm/exceptions.py b/cognee/infrastructure/llm/exceptions.py
index af3aa5832..287820448 100644
--- a/cognee/infrastructure/llm/exceptions.py
+++ b/cognee/infrastructure/llm/exceptions.py
@@ -1,5 +1,22 @@
-from cognee.exceptions.exceptions import CriticalError
+from cognee.exceptions.exceptions import CogneeValidationError
 
 
-class ContentPolicyFilterError(CriticalError):
+class ContentPolicyFilterError(CogneeValidationError):
     pass
+
+
+class LLMAPIKeyNotSetError(CogneeValidationError):
+    """
+    Raised when the LLM API key is not set in the configuration.
+    """
+    def __init__(self, message: str = "LLM API key is not set."):
+        super().__init__(message=message, name="LLMAPIKeyNotSetError")
+
+
+class UnsupportedLLMProviderError(CogneeValidationError):
+    """
+    Raised when an unsupported LLM provider is specified in the configuration.
+    """
+    def __init__(self, provider: str):
+        message = f"Unsupported LLM provider: {provider}"
+        super().__init__(message=message, name="UnsupportedLLMProviderError")
diff --git a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
index 22d101077..b486fa089 100644
--- a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
@@ -2,11 +2,14 @@
 
 from enum import Enum
 
-from cognee.exceptions import InvalidValueError
 from cognee.infrastructure.llm import get_llm_config
 from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.ollama.adapter import (
     OllamaAPIAdapter,
 )
+from cognee.infrastructure.llm.exceptions import (
+    LLMAPIKeyNotSetError,
+    UnsupportedLLMProviderError,
+)
 
 
 # Define an Enum for LLM Providers

From 623148ec6af1a4848862e20808dd03be2e9b45fa Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 11:53:29 +0200
Subject: [PATCH 05/34] feat: adds new errors to litellm client

---
 .../litellm_instructor/llm/get_llm_client.py              | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
index b486fa089..3006a795b 100644
--- a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
@@ -83,7 +83,7 @@ def get_llm_client():
 
     elif provider == LLMProvider.OLLAMA:
         if llm_config.llm_api_key is None:
-            raise InvalidValueError(message="LLM API key is not set.")
+            raise LLMAPIKeyNotSetError()
 
         from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api.adapter import (
             GenericAPIAdapter,
@@ -106,7 +106,7 @@ def get_llm_client():
 
     elif provider == LLMProvider.CUSTOM:
         if llm_config.llm_api_key is None:
-            raise InvalidValueError(message="LLM API key is not set.")
+            raise LLMAPIKeyNotSetError()
 
         from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api.adapter import (
             GenericAPIAdapter,
@@ -125,7 +125,7 @@ def get_llm_client():
 
     elif provider == LLMProvider.GEMINI:
         if llm_config.llm_api_key is None:
-            raise InvalidValueError(message="LLM API key is not set.")
+            raise LLMAPIKeyNotSetError()
 
         from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.gemini.adapter import (
             GeminiAdapter,
@@ -141,4 +141,4 @@ def get_llm_client():
         )
 
     else:
-        raise InvalidValueError(message=f"Unsupported LLM provider: {provider}")
+        raise UnsupportedLLMProviderError(provider)

From 7bd2660d080ea36477e02c4eeb585eb29d601717 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 11:58:32 +0200
Subject: [PATCH 06/34] feat: setting base classes of data exceptions to the
 new ones

---
 cognee/modules/data/exceptions/exceptions.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/cognee/modules/data/exceptions/exceptions.py b/cognee/modules/data/exceptions/exceptions.py
index 214bf5381..4eff8f9b8 100644
--- a/cognee/modules/data/exceptions/exceptions.py
+++ b/cognee/modules/data/exceptions/exceptions.py
@@ -1,8 +1,11 @@
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import (
+    CogneeValidationError,
+    CogneeConfigurationError,
+)
 from fastapi import status
 
 
-class UnstructuredLibraryImportError(CogneeApiError):
+class UnstructuredLibraryImportError(CogneeConfigurationError):
     def __init__(
         self,
         message: str = "Import error. Unstructured library is not installed.",
@@ -12,7 +15,7 @@ class UnstructuredLibraryImportError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class UnauthorizedDataAccessError(CogneeApiError):
+class UnauthorizedDataAccessError(CogneeValidationError):
     def __init__(
         self,
         message: str = "User does not have permission to access this data.",
@@ -22,7 +25,7 @@ class UnauthorizedDataAccessError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class DatasetNotFoundError(CogneeApiError):
+class DatasetNotFoundError(CogneeValidationError):
     def __init__(
         self,
         message: str = "Dataset not found.",
@@ -32,7 +35,7 @@ class DatasetNotFoundError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class DatasetTypeError(CogneeApiError):
+class DatasetTypeError(CogneeValidationError):
     def __init__(
         self,
         message: str = "Dataset type not supported.",

From 6870bba5a90285e87f066095a3a878865c343fad Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 12:03:18 +0200
Subject: [PATCH 07/34] feat: adds new error to delete

---
 cognee/modules/data/exceptions/exceptions.py | 10 ++++++++++
 cognee/modules/data/methods/delete_data.py   |  6 ++----
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/cognee/modules/data/exceptions/exceptions.py b/cognee/modules/data/exceptions/exceptions.py
index 4eff8f9b8..c85e11cb2 100644
--- a/cognee/modules/data/exceptions/exceptions.py
+++ b/cognee/modules/data/exceptions/exceptions.py
@@ -43,3 +43,13 @@ class DatasetTypeError(CogneeValidationError):
         status_code=status.HTTP_400_BAD_REQUEST,
     ):
         super().__init__(message, name, status_code)
+
+
+class InvalidAttributeError(CogneeValidationError):
+    def __init__(
+        self,
+        message: str = "The provided data object is missing the required '__tablename__' attribute.",
+        name: str = "InvalidAttributeError",
+        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 2d87d73a5..6ec055fcc 100644
--- a/cognee/modules/data/methods/delete_data.py
+++ b/cognee/modules/data/methods/delete_data.py
@@ -1,4 +1,4 @@
-from cognee.exceptions import InvalidAttributeError
+from cognee.modules.data.exceptions.exceptions import InvalidAttributeError
 from cognee.modules.data.models import Data
 from cognee.infrastructure.databases.relational import get_relational_engine
 
@@ -13,9 +13,7 @@ async def delete_data(data: Data):
         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."
-        )
+        raise InvalidAttributeError()
 
     db_engine = get_relational_engine()
 

From ebd4403c2f74bf6f65f717ad389b89e869a7efc2 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 12:14:42 +0200
Subject: [PATCH 08/34] adds new error classes to keywordextration error and
 defines new error

---
 .../data/exceptions/__init__.py               |  9 ++++++++
 .../data/exceptions/exceptions.py             | 22 +++++++++++++++++++
 .../data/utils/extract_keywords.py            |  4 ++--
 3 files changed, 33 insertions(+), 2 deletions(-)
 create mode 100644 cognee/infrastructure/data/exceptions/__init__.py
 create mode 100644 cognee/infrastructure/data/exceptions/exceptions.py

diff --git a/cognee/infrastructure/data/exceptions/__init__.py b/cognee/infrastructure/data/exceptions/__init__.py
new file mode 100644
index 000000000..6735200ed
--- /dev/null
+++ b/cognee/infrastructure/data/exceptions/__init__.py
@@ -0,0 +1,9 @@
+"""
+Custom exceptions for the Cognee API.
+
+This module defines a set of exceptions for handling various data errors
+"""
+
+from .exceptions import (
+    KeywordExtractionError
+)
diff --git a/cognee/infrastructure/data/exceptions/exceptions.py b/cognee/infrastructure/data/exceptions/exceptions.py
new file mode 100644
index 000000000..5c36d6754
--- /dev/null
+++ b/cognee/infrastructure/data/exceptions/exceptions.py
@@ -0,0 +1,22 @@
+from cognee.exceptions import (
+    CogneeValidationError,
+)
+from fastapi import status
+
+
+class KeywordExtractionError(CogneeValidationError):
+    """
+    Raised when a provided value is syntactically valid but semantically unacceptable
+    for the given operation.
+
+    Example:
+        - Passing an empty string to a keyword extraction function.
+    """
+
+    def __init__(
+        self,
+        message: str = "Extract_keywords cannot extract keywords from empty text.",
+        name: str = "KeywordExtractionError",
+        status_code: int = status.HTTP_400_BAD_REQUEST,
+    ):
+        super().__init__(message, name, status_code)
diff --git a/cognee/infrastructure/data/utils/extract_keywords.py b/cognee/infrastructure/data/utils/extract_keywords.py
index bd4fedd56..2915131a4 100644
--- a/cognee/infrastructure/data/utils/extract_keywords.py
+++ b/cognee/infrastructure/data/utils/extract_keywords.py
@@ -1,6 +1,6 @@
 from sklearn.feature_extraction.text import TfidfVectorizer
 
-from cognee.exceptions import InvalidValueError
+from cognee.infrastructure.data.exceptions.exceptions import KeywordExtractionError
 from cognee.shared.utils import extract_pos_tags
 
 
@@ -25,7 +25,7 @@ def extract_keywords(text: str) -> list[str]:
           with more than 3 characters.
     """
     if len(text) == 0:
-        raise InvalidValueError(message="extract_keywords cannot extract keywords from empty text.")
+        raise KeywordExtractionError()
 
     tags = extract_pos_tags(text)
     nouns = [word for (word, tag) in tags if tag == "NN"]

From 657c775cbe673f10059028f5c3f5c51af239c5bf Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 12:22:03 +0200
Subject: [PATCH 09/34] feat: adds configattributeerror

---
 cognee/api/v1/config/config.py         |  6 ++----
 cognee/api/v1/exceptions/__init__.py   |  9 +++++++++
 cognee/api/v1/exceptions/exceptions.py | 15 +++++++++++++++
 3 files changed, 26 insertions(+), 4 deletions(-)
 create mode 100644 cognee/api/v1/exceptions/__init__.py
 create mode 100644 cognee/api/v1/exceptions/exceptions.py

diff --git a/cognee/api/v1/config/config.py b/cognee/api/v1/config/config.py
index 9970b7471..03df700cb 100644
--- a/cognee/api/v1/config/config.py
+++ b/cognee/api/v1/config/config.py
@@ -2,7 +2,6 @@
 
 import os
 from cognee.base_config import get_base_config
-from cognee.exceptions import InvalidValueError, InvalidAttributeError
 from cognee.modules.cognify.config import get_cognify_config
 from cognee.infrastructure.data.chunking.config import get_chunk_config
 from cognee.infrastructure.databases.vector import get_vectordb_config
@@ -11,6 +10,7 @@ from cognee.infrastructure.llm.config import (
     get_llm_config,
 )
 from cognee.infrastructure.databases.relational import get_relational_config, get_migration_config
+from cognee.api.v1.exceptions.exceptions import InvalidConfigAttributeError
 
 
 class config:
@@ -92,9 +92,7 @@ class config:
             if hasattr(llm_config, key):
                 object.__setattr__(llm_config, key, value)
             else:
-                raise InvalidAttributeError(
-                    message=f"'{key}' is not a valid attribute of the config."
-                )
+                raise InvalidConfigAttributeError(attribute=key)
 
     @staticmethod
     def set_chunk_strategy(chunk_strategy: object):
diff --git a/cognee/api/v1/exceptions/__init__.py b/cognee/api/v1/exceptions/__init__.py
new file mode 100644
index 000000000..f18b9f99a
--- /dev/null
+++ b/cognee/api/v1/exceptions/__init__.py
@@ -0,0 +1,9 @@
+"""
+Custom exceptions for the Cognee API.
+
+This module defines a set of exceptions for handling various data errors
+"""
+
+from .exceptions import (
+    InvalidConfigAttributeError
+)
diff --git a/cognee/api/v1/exceptions/exceptions.py b/cognee/api/v1/exceptions/exceptions.py
new file mode 100644
index 000000000..9875d179b
--- /dev/null
+++ b/cognee/api/v1/exceptions/exceptions.py
@@ -0,0 +1,15 @@
+from cognee.exceptions import (
+    CogneeConfigurationError,
+)
+from fastapi import status
+
+
+class InvalidConfigAttributeError(CogneeConfigurationError):
+    def __init__(
+        self,
+        attribute: str,
+        name: str = "InvalidConfigAttributeError",
+        status_code: int = status.HTTP_400_BAD_REQUEST,
+    ):
+        message = f"'{attribute}' is not a valid attribute of the configuration."
+        super().__init__(message, name, status_code)

From 5bc00f1143d18ef000beb3b7f92411b8160913f7 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 12:29:35 +0200
Subject: [PATCH 10/34] feat: adds new search classes to search.py

---
 cognee/modules/search/exceptions/__init__.py   |  9 +++++++++
 cognee/modules/search/exceptions/exceptions.py | 15 +++++++++++++++
 cognee/modules/search/methods/search.py        |  5 ++---
 3 files changed, 26 insertions(+), 3 deletions(-)
 create mode 100644 cognee/modules/search/exceptions/__init__.py
 create mode 100644 cognee/modules/search/exceptions/exceptions.py

diff --git a/cognee/modules/search/exceptions/__init__.py b/cognee/modules/search/exceptions/__init__.py
new file mode 100644
index 000000000..ffb30f428
--- /dev/null
+++ b/cognee/modules/search/exceptions/__init__.py
@@ -0,0 +1,9 @@
+"""
+Custom exceptions for the Cognee API.
+
+This module defines a set of exceptions for handling various data errors
+"""
+
+from .exceptions import (
+    UnsupportedSearchTypeError
+)
diff --git a/cognee/modules/search/exceptions/exceptions.py b/cognee/modules/search/exceptions/exceptions.py
new file mode 100644
index 000000000..ddc877700
--- /dev/null
+++ b/cognee/modules/search/exceptions/exceptions.py
@@ -0,0 +1,15 @@
+from cognee.exceptions import (
+    CogneeValidationError,
+)
+from fastapi import status
+
+
+class UnsupportedSearchTypeError(CogneeValidationError):
+    def __init__(
+        self,
+        search_type: str,
+        name: str = "UnsupportedSearchTypeError",
+        status_code: int = status.HTTP_400_BAD_REQUEST,
+    ):
+        message = f"Unsupported search type: {search_type}"
+        super().__init__(message, name, status_code)
diff --git a/cognee/modules/search/methods/search.py b/cognee/modules/search/methods/search.py
index 365920019..f431a498e 100644
--- a/cognee/modules/search/methods/search.py
+++ b/cognee/modules/search/methods/search.py
@@ -3,9 +3,8 @@ import json
 import asyncio
 from uuid import UUID
 from typing import Callable, List, Optional, Type, Union
-
+from cognee.modules.search.exceptions import UnsupportedSearchTypeError
 from cognee.context_global_variables import set_database_global_context_variables
-from cognee.exceptions import InvalidValueError
 from cognee.modules.retrieval.chunks_retriever import ChunksRetriever
 from cognee.modules.retrieval.insights_retriever import InsightsRetriever
 from cognee.modules.retrieval.summaries_retriever import SummariesRetriever
@@ -136,7 +135,7 @@ async def specific_search(
     search_task = search_tasks.get(query_type)
 
     if search_task is None:
-        raise InvalidValueError(message=f"Unsupported search type: {query_type}")
+        raise UnsupportedSearchTypeError(str(query_type))
 
     send_telemetry("cognee.search EXECUTION STARTED", user.id)
 

From 9fb9f68c42a9b1179f5bfca59e9e9976fcec323f Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 12:36:31 +0200
Subject: [PATCH 11/34] adds new base errors to retrieval exceptions

---
 .../modules/retrieval/exceptions/exceptions.py | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/cognee/modules/retrieval/exceptions/exceptions.py b/cognee/modules/retrieval/exceptions/exceptions.py
index a403f3e44..3e934909b 100644
--- a/cognee/modules/retrieval/exceptions/exceptions.py
+++ b/cognee/modules/retrieval/exceptions/exceptions.py
@@ -1,8 +1,8 @@
 from fastapi import status
-from cognee.exceptions import CogneeApiError, CriticalError
+from cognee.exceptions import CogneeValidationError, CogneeSystemError
 
 
-class SearchTypeNotSupported(CogneeApiError):
+class SearchTypeNotSupported(CogneeValidationError):
     def __init__(
         self,
         message: str = "CYPHER search type not supported by the adapter.",
@@ -12,7 +12,7 @@ class SearchTypeNotSupported(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class CypherSearchError(CogneeApiError):
+class CypherSearchError(CogneeSystemError):
     def __init__(
         self,
         message: str = "An error occurred during the execution of the Cypher query.",
@@ -22,11 +22,17 @@ class CypherSearchError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class NoDataError(CriticalError):
-    message: str = "No data found in the system, please add data first."
+class NoDataError(CogneeValidationError):
+    def __init__(
+        self,
+        message: str = "No data found in the system, please add data first.",
+        name: str = "NoDataError",
+        status_code: int = status.HTTP_404_NOT_FOUND,
+    ):
+        super().__init__(message, name, status_code)
 
 
-class CollectionDistancesNotFoundError(CogneeApiError):
+class CollectionDistancesNotFoundError(CogneeValidationError):
     def __init__(
         self,
         message: str = "No collection distances found for the given query.",

From 91b9c11cd01109794c0bedef0f5fe6580435dd96 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 12:39:34 +0200
Subject: [PATCH 12/34] changes base class for vectordb exceptions

---
 .../databases/vector/exceptions/exceptions.py             | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/cognee/infrastructure/databases/vector/exceptions/exceptions.py b/cognee/infrastructure/databases/vector/exceptions/exceptions.py
index b6a8237d8..48d9976e0 100644
--- a/cognee/infrastructure/databases/vector/exceptions/exceptions.py
+++ b/cognee/infrastructure/databases/vector/exceptions/exceptions.py
@@ -1,8 +1,8 @@
 from fastapi import status
-from cognee.exceptions import CriticalError
+from cognee.exceptions import CogneeValidationError
 
 
-class CollectionNotFoundError(CriticalError):
+class CollectionNotFoundError(CogneeValidationError):
     """
     Represents an error that occurs when a requested collection cannot be found.
 
@@ -16,7 +16,5 @@ class CollectionNotFoundError(CriticalError):
         message,
         name: str = "CollectionNotFoundError",
         status_code: int = status.HTTP_422_UNPROCESSABLE_ENTITY,
-        log=True,
-        log_level="DEBUG",
     ):
-        super().__init__(message, name, status_code, log, log_level)
+        super().__init__(message, name, status_code)

From 6dcd59c73cee29ae3da396724e37e5f02d0b9636 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 13:07:04 +0200
Subject: [PATCH 13/34] feat: Adds changes to cognee graph part

---
 .../modules/graph/cognee_graph/CogneeGraph.py |  6 ++---
 .../graph/cognee_graph/CogneeGraphElements.py | 16 ++++--------
 cognee/modules/graph/exceptions/__init__.py   |  2 ++
 cognee/modules/graph/exceptions/exceptions.py | 26 ++++++++++++++++---
 4 files changed, 32 insertions(+), 18 deletions(-)

diff --git a/cognee/modules/graph/cognee_graph/CogneeGraph.py b/cognee/modules/graph/cognee_graph/CogneeGraph.py
index ca1984dfe..bdafaf238 100644
--- a/cognee/modules/graph/cognee_graph/CogneeGraph.py
+++ b/cognee/modules/graph/cognee_graph/CogneeGraph.py
@@ -2,8 +2,7 @@ import time
 from cognee.shared.logging_utils import get_logger
 from typing import List, Dict, Union, Optional, Type
 
-from cognee.exceptions import InvalidValueError
-from cognee.modules.graph.exceptions import EntityNotFoundError, EntityAlreadyExistsError
+from cognee.modules.graph.exceptions import EntityNotFoundError, EntityAlreadyExistsError, InvalidDimensionsError
 from cognee.infrastructure.databases.graph.graph_db_interface import GraphDBInterface
 from cognee.modules.graph.cognee_graph.CogneeGraphElements import Node, Edge
 from cognee.modules.graph.cognee_graph.CogneeAbstractGraph import CogneeAbstractGraph
@@ -66,8 +65,7 @@ class CogneeGraph(CogneeAbstractGraph):
         node_name: Optional[List[str]] = None,
     ) -> None:
         if node_dimension < 1 or edge_dimension < 1:
-            raise InvalidValueError(message="Dimensions must be positive integers")
-
+            raise InvalidDimensionsError()
         try:
             import time
 
diff --git a/cognee/modules/graph/cognee_graph/CogneeGraphElements.py b/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
index 063248ee2..c22cc5c18 100644
--- a/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
+++ b/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
@@ -1,8 +1,6 @@
 import numpy as np
 from typing import List, Dict, Optional, Any, Union
-
-from cognee.exceptions import InvalidValueError
-
+from cognee.modules.graph.exceptions import InvalidDimensionsError, DimensionOutOfRangeError
 
 class Node:
     """
@@ -24,7 +22,7 @@ class Node:
         self, node_id: str, attributes: Optional[Dict[str, Any]] = None, dimension: int = 1
     ):
         if dimension <= 0:
-            raise InvalidValueError(message="Dimension must be a positive integer")
+            raise InvalidDimensionsError()
         self.id = node_id
         self.attributes = attributes if attributes is not None else {}
         self.attributes["vector_distance"] = float("inf")
@@ -58,9 +56,7 @@ class Node:
 
     def is_node_alive_in_dimension(self, dimension: int) -> bool:
         if dimension < 0 or dimension >= len(self.status):
-            raise InvalidValueError(
-                message=f"Dimension {dimension} is out of range. Valid range is 0 to {len(self.status) - 1}."
-            )
+            raise DimensionOutOfRangeError(dimension=dimension, max_index=len(self.status) - 1)
         return self.status[dimension] == 1
 
     def add_attribute(self, key: str, value: Any) -> None:
@@ -110,7 +106,7 @@ class Edge:
         dimension: int = 1,
     ):
         if dimension <= 0:
-            raise InvalidValueError(message="Dimensions must be a positive integer.")
+            InvalidDimensionsError()
         self.node1 = node1
         self.node2 = node2
         self.attributes = attributes if attributes is not None else {}
@@ -120,9 +116,7 @@ class Edge:
 
     def is_edge_alive_in_dimension(self, dimension: int) -> bool:
         if dimension < 0 or dimension >= len(self.status):
-            raise InvalidValueError(
-                message=f"Dimension {dimension} is out of range. Valid range is 0 to {len(self.status) - 1}."
-            )
+            raise DimensionOutOfRangeError(dimension=dimension, max_index=len(self.status) - 1)
         return self.status[dimension] == 1
 
     def add_attribute(self, key: str, value: Any) -> None:
diff --git a/cognee/modules/graph/exceptions/__init__.py b/cognee/modules/graph/exceptions/__init__.py
index 5cf600099..04bec74ad 100644
--- a/cognee/modules/graph/exceptions/__init__.py
+++ b/cognee/modules/graph/exceptions/__init__.py
@@ -7,4 +7,6 @@ This module defines a set of exceptions for handling various graph errors
 from .exceptions import (
     EntityNotFoundError,
     EntityAlreadyExistsError,
+    InvalidDimensionsError,
+    DimensionOutOfRangeError,
 )
diff --git a/cognee/modules/graph/exceptions/exceptions.py b/cognee/modules/graph/exceptions/exceptions.py
index 854e620ff..6fb2d400d 100644
--- a/cognee/modules/graph/exceptions/exceptions.py
+++ b/cognee/modules/graph/exceptions/exceptions.py
@@ -1,8 +1,8 @@
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import CogneeValidationError
 from fastapi import status
 
 
-class EntityNotFoundError(CogneeApiError):
+class EntityNotFoundError(CogneeValidationError):
     """Database returns nothing"""
 
     def __init__(
@@ -14,7 +14,7 @@ class EntityNotFoundError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class EntityAlreadyExistsError(CogneeApiError):
+class EntityAlreadyExistsError(CogneeValidationError):
     """Conflict detected, like trying to create a resource that already exists"""
 
     def __init__(
@@ -24,3 +24,23 @@ class EntityAlreadyExistsError(CogneeApiError):
         status_code=status.HTTP_409_CONFLICT,
     ):
         super().__init__(message, name, status_code)
+
+class InvalidDimensionsError(CogneeValidationError):
+    def __init__(
+        self,
+        name: str = "InvalidDimensionsError",
+        status_code: int = status.HTTP_400_BAD_REQUEST,
+    ):
+        message = "Dimensions must be positive integers."
+        super().__init__(message, name, status_code)
+
+class DimensionOutOfRangeError(CogneeValidationError):
+    def __init__(
+        self,
+        dimension: int,
+        max_index: int,
+        name: str = "DimensionOutOfRangeError",
+        status_code: int = status.HTTP_400_BAD_REQUEST,
+    ):
+        message = f"Dimension {dimension} is out of range. Valid range is 0 to {max_index}."
+        super().__init__(message, name, status_code)

From 32996aa0d08b0c90dd234607bd6436b79e72e5b1 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 13:40:50 +0200
Subject: [PATCH 14/34] feat: adds new error classes to llm and databases +
 introduces loglevel and logging from child error

---
 cognee/exceptions/exceptions.py               | 16 +++++++++++----
 .../exceptions/EmbeddingException.py          | 20 -------------------
 .../databases/exceptions/__init__.py          |  1 +
 .../databases/exceptions/exceptions.py        | 18 +++++++++++++++++
 .../embeddings/LiteLLMEmbeddingEngine.py      |  2 +-
 .../databases/vector/exceptions/exceptions.py |  4 +++-
 .../vector/pgvector/PGVectorAdapter.py        |  4 ++--
 cognee/infrastructure/llm/exceptions.py       |  8 ++++++++
 .../litellm_instructor/llm/openai/adapter.py  |  5 ++---
 cognee/tasks/storage/index_data_points.py     |  2 +-
 10 files changed, 48 insertions(+), 32 deletions(-)
 delete mode 100644 cognee/infrastructure/databases/exceptions/EmbeddingException.py

diff --git a/cognee/exceptions/exceptions.py b/cognee/exceptions/exceptions.py
index e7eb784b8..9b6cef21d 100644
--- a/cognee/exceptions/exceptions.py
+++ b/cognee/exceptions/exceptions.py
@@ -43,8 +43,10 @@ class CogneeSystemError(CogneeApiError):
         message: str = "A system error occurred.",
         name: str = "CogneeSystemError",
         status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+        log=True,
+        log_level="ERROR",
     ):
-        super().__init__(message, name, status_code)
+        super().__init__(message, name, status_code, log, log_level)
 
 
 class CogneeValidationError(CogneeApiError):
@@ -55,8 +57,10 @@ class CogneeValidationError(CogneeApiError):
         message: str = "A validation error occurred.",
         name: str = "CogneeValidationError",
         status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
+        log=True,
+        log_level="ERROR",
     ):
-        super().__init__(message, name, status_code)
+        super().__init__(message, name, status_code, log, log_level)
 
 
 class CogneeConfigurationError(CogneeApiError):
@@ -67,8 +71,10 @@ class CogneeConfigurationError(CogneeApiError):
         message: str = "A system configuration error occurred.",
         name: str = "CogneeConfigurationError",
         status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+        log=True,
+        log_level="ERROR",
     ):
-        super().__init__(message, name, status_code)
+        super().__init__(message, name, status_code, log, log_level)
 
 
 class CogneeTransientError(CogneeApiError):
@@ -79,6 +85,8 @@ class CogneeTransientError(CogneeApiError):
         message: str = "A transient error occurred.",
         name: str = "CogneeTransientError",
         status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
+        log=True,
+        log_level="ERROR",
     ):
-        super().__init__(message, name, status_code)
+        super().__init__(message, name, status_code, log, log_level)
 
diff --git a/cognee/infrastructure/databases/exceptions/EmbeddingException.py b/cognee/infrastructure/databases/exceptions/EmbeddingException.py
deleted file mode 100644
index 62616899c..000000000
--- a/cognee/infrastructure/databases/exceptions/EmbeddingException.py
+++ /dev/null
@@ -1,20 +0,0 @@
-from cognee.exceptions import CogneeApiError
-from fastapi import status
-
-
-class EmbeddingException(CogneeApiError):
-    """
-    Custom exception for handling embedding-related errors.
-
-    This exception class is designed to indicate issues specifically related to embeddings
-    within the application. It extends the base exception class CogneeApiError and allows
-    for customization of the error message, name, and status code.
-    """
-
-    def __init__(
-        self,
-        message: str = "Embedding Exception.",
-        name: str = "EmbeddingException",
-        status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
-    ):
-        super().__init__(message, name, status_code)
diff --git a/cognee/infrastructure/databases/exceptions/__init__.py b/cognee/infrastructure/databases/exceptions/__init__.py
index 9d8d18567..c7d2a8feb 100644
--- a/cognee/infrastructure/databases/exceptions/__init__.py
+++ b/cognee/infrastructure/databases/exceptions/__init__.py
@@ -8,4 +8,5 @@ from .exceptions import (
     EntityNotFoundError,
     EntityAlreadyExistsError,
     DatabaseNotCreatedError,
+    EmbeddingException,
 )
diff --git a/cognee/infrastructure/databases/exceptions/exceptions.py b/cognee/infrastructure/databases/exceptions/exceptions.py
index 4aba09d37..66740fa5e 100644
--- a/cognee/infrastructure/databases/exceptions/exceptions.py
+++ b/cognee/infrastructure/databases/exceptions/exceptions.py
@@ -84,3 +84,21 @@ class NodesetFilterNotSupportedError(CogneeConfigurationError):
         self.message = message
         self.name = name
         self.status_code = status_code
+
+
+class EmbeddingException(CogneeConfigurationError):
+    """
+    Custom exception for handling embedding-related errors.
+
+    This exception class is designed to indicate issues specifically related to embeddings
+    within the application. It extends the base exception class CogneeConfigurationError allows
+    for customization of the error message, name, and status code.
+    """
+
+    def __init__(
+        self,
+        message: str = "Embedding Exception.",
+        name: str = "EmbeddingException",
+        status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
+    ):
+        super().__init__(message, name, status_code)
diff --git a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
index e3cdaea00..dae664907 100644
--- a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
+++ b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
@@ -6,7 +6,7 @@ import math
 import litellm
 import os
 from cognee.infrastructure.databases.vector.embeddings.EmbeddingEngine import EmbeddingEngine
-from cognee.infrastructure.databases.exceptions.EmbeddingException import EmbeddingException
+from cognee.infrastructure.databases.exceptions import EmbeddingException
 from cognee.infrastructure.llm.tokenizer.Gemini import (
     GeminiTokenizer,
 )
diff --git a/cognee/infrastructure/databases/vector/exceptions/exceptions.py b/cognee/infrastructure/databases/vector/exceptions/exceptions.py
index 48d9976e0..ee0712433 100644
--- a/cognee/infrastructure/databases/vector/exceptions/exceptions.py
+++ b/cognee/infrastructure/databases/vector/exceptions/exceptions.py
@@ -16,5 +16,7 @@ class CollectionNotFoundError(CogneeValidationError):
         message,
         name: str = "CollectionNotFoundError",
         status_code: int = status.HTTP_422_UNPROCESSABLE_ENTITY,
+        log=True,
+        log_level="DEBUG",
     ):
-        super().__init__(message, name, status_code)
+        super().__init__(message, name, status_code, log, log_level)
diff --git a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
index ac40b91b8..96b2056c4 100644
--- a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
+++ b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
@@ -9,7 +9,7 @@ from sqlalchemy.exc import ProgrammingError
 from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_exponential
 from asyncpg import DeadlockDetectedError, DuplicateTableError, UniqueViolationError
 
-from cognee.exceptions import InvalidValueError
+
 from cognee.shared.logging_utils import get_logger
 from cognee.infrastructure.engine import DataPoint
 from cognee.infrastructure.engine.utils import parse_id
@@ -275,7 +275,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
                 return metadata.tables[collection_name]
             else:
                 raise CollectionNotFoundError(
-                    f"Collection '{collection_name}' not found!", log_level="DEBUG"
+                    f"Collection '{collection_name}' not found!",
                 )
 
     async def retrieve(self, collection_name: str, data_point_ids: List[str]):
diff --git a/cognee/infrastructure/llm/exceptions.py b/cognee/infrastructure/llm/exceptions.py
index 287820448..c8f5726fb 100644
--- a/cognee/infrastructure/llm/exceptions.py
+++ b/cognee/infrastructure/llm/exceptions.py
@@ -20,3 +20,11 @@ class UnsupportedLLMProviderError(CogneeValidationError):
     def __init__(self, provider: str):
         message = f"Unsupported LLM provider: {provider}"
         super().__init__(message=message, name="UnsupportedLLMProviderError")
+
+class MissingSystemPromptPathError(CogneeValidationError):
+    def __init__(
+        self,
+        name: str = "MissingSystemPromptPathError",
+    ):
+        message = "No system prompt path provided."
+        super().__init__(message, name)
diff --git a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
index 69c1bac1c..38e1bc82e 100644
--- a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
@@ -7,12 +7,11 @@ from openai import ContentFilterFinishReasonError
 from litellm.exceptions import ContentPolicyViolationError
 from instructor.exceptions import InstructorRetryException
 
-from cognee.exceptions import InvalidValueError
 from cognee.infrastructure.llm.LLMGateway import LLMGateway
 from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.llm_interface import (
     LLMInterface,
 )
-from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError
+from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError, MissingSystemPromptPathError
 from cognee.infrastructure.files.utils.open_data_file import open_data_file
 from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     rate_limit_async,
@@ -325,7 +324,7 @@ class OpenAIAdapter(LLMInterface):
         if not text_input:
             text_input = "No user input provided."
         if not system_prompt:
-            raise InvalidValueError(message="No system prompt path provided.")
+            raise MissingSystemPromptPathError()
         system_prompt = LLMGateway.read_query_prompt(system_prompt)
 
         formatted_prompt = (
diff --git a/cognee/tasks/storage/index_data_points.py b/cognee/tasks/storage/index_data_points.py
index 51b6c2d6e..9c363c04c 100644
--- a/cognee/tasks/storage/index_data_points.py
+++ b/cognee/tasks/storage/index_data_points.py
@@ -1,6 +1,6 @@
 from cognee.shared.logging_utils import get_logger
 
-from cognee.infrastructure.databases.exceptions.EmbeddingException import EmbeddingException
+from cognee.infrastructure.databases.exceptions import EmbeddingException
 from cognee.infrastructure.databases.vector import get_vector_engine
 from cognee.infrastructure.engine import DataPoint
 

From bed523a36b18b8b6ac76de69f344973baf5cd043 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 13:59:12 +0200
Subject: [PATCH 15/34] feat: api error handling restruct

---
 cognee/api/v1/config/config.py         | 12 ++-----
 cognee/api/v1/delete/delete.py         |  2 +-
 cognee/api/v1/delete/exceptions.py     | 50 --------------------------
 cognee/api/v1/exceptions/__init__.py   |  6 +++-
 cognee/api/v1/exceptions/exceptions.py | 37 +++++++++++++++++++
 5 files changed, 46 insertions(+), 61 deletions(-)
 delete mode 100644 cognee/api/v1/delete/exceptions.py

diff --git a/cognee/api/v1/config/config.py b/cognee/api/v1/config/config.py
index 03df700cb..464753438 100644
--- a/cognee/api/v1/config/config.py
+++ b/cognee/api/v1/config/config.py
@@ -129,9 +129,7 @@ class config:
             if hasattr(relational_db_config, key):
                 object.__setattr__(relational_db_config, key, value)
             else:
-                raise InvalidAttributeError(
-                    message=f"'{key}' is not a valid attribute of the config."
-                )
+                raise InvalidConfigAttributeError(attribute=key)
 
     @staticmethod
     def set_migration_db_config(config_dict: dict):
@@ -143,9 +141,7 @@ class config:
             if hasattr(migration_db_config, key):
                 object.__setattr__(migration_db_config, key, value)
             else:
-                raise InvalidAttributeError(
-                    message=f"'{key}' is not a valid attribute of the config."
-                )
+                raise InvalidConfigAttributeError(attribute=key)
 
     @staticmethod
     def set_graph_db_config(config_dict: dict) -> None:
@@ -169,9 +165,7 @@ class config:
             if hasattr(vector_db_config, key):
                 object.__setattr__(vector_db_config, key, value)
             else:
-                raise InvalidAttributeError(
-                    message=f"'{key}' is not a valid attribute of the config."
-                )
+                InvalidConfigAttributeError(attribute=key)
 
     @staticmethod
     def set_vector_db_key(db_key: str):
diff --git a/cognee/api/v1/delete/delete.py b/cognee/api/v1/delete/delete.py
index 98f6cb9fc..73f264670 100644
--- a/cognee/api/v1/delete/delete.py
+++ b/cognee/api/v1/delete/delete.py
@@ -16,7 +16,7 @@ from cognee.modules.users.methods import get_default_user
 from cognee.modules.data.methods import get_authorized_existing_datasets
 from cognee.context_global_variables import set_database_global_context_variables
 
-from cognee.api.v1.delete.exceptions import (
+from cognee.api.v1.exceptions import (
     DocumentNotFoundError,
     DatasetNotFoundError,
     DocumentSubgraphNotFoundError,
diff --git a/cognee/api/v1/delete/exceptions.py b/cognee/api/v1/delete/exceptions.py
deleted file mode 100644
index a4d2a77ed..000000000
--- a/cognee/api/v1/delete/exceptions.py
+++ /dev/null
@@ -1,50 +0,0 @@
-from cognee.exceptions import CogneeApiError
-from fastapi import status
-
-
-class DocumentNotFoundError(CogneeApiError):
-    """Raised when a document cannot be found in the database."""
-
-    def __init__(
-        self,
-        message: str = "Document not found in database.",
-        name: str = "DocumentNotFoundError",
-        status_code: int = status.HTTP_404_NOT_FOUND,
-    ):
-        super().__init__(message, name, status_code)
-
-
-class DatasetNotFoundError(CogneeApiError):
-    """Raised when a dataset cannot be found."""
-
-    def __init__(
-        self,
-        message: str = "Dataset not found.",
-        name: str = "DatasetNotFoundError",
-        status_code: int = status.HTTP_404_NOT_FOUND,
-    ):
-        super().__init__(message, name, status_code)
-
-
-class DataNotFoundError(CogneeApiError):
-    """Raised when a dataset cannot be found."""
-
-    def __init__(
-        self,
-        message: str = "Data not found.",
-        name: str = "DataNotFoundError",
-        status_code: int = status.HTTP_404_NOT_FOUND,
-    ):
-        super().__init__(message, name, status_code)
-
-
-class DocumentSubgraphNotFoundError(CogneeApiError):
-    """Raised when a document's subgraph cannot be found in the graph database."""
-
-    def __init__(
-        self,
-        message: str = "Document subgraph not found in graph database.",
-        name: str = "DocumentSubgraphNotFoundError",
-        status_code: int = status.HTTP_404_NOT_FOUND,
-    ):
-        super().__init__(message, name, status_code)
diff --git a/cognee/api/v1/exceptions/__init__.py b/cognee/api/v1/exceptions/__init__.py
index f18b9f99a..b2948934c 100644
--- a/cognee/api/v1/exceptions/__init__.py
+++ b/cognee/api/v1/exceptions/__init__.py
@@ -5,5 +5,9 @@ This module defines a set of exceptions for handling various data errors
 """
 
 from .exceptions import (
-    InvalidConfigAttributeError
+    InvalidConfigAttributeError,
+    DocumentNotFoundError,
+    DatasetNotFoundError,
+    DataNotFoundError,
+    DocumentSubgraphNotFoundError
 )
diff --git a/cognee/api/v1/exceptions/exceptions.py b/cognee/api/v1/exceptions/exceptions.py
index 9875d179b..8b65691f1 100644
--- a/cognee/api/v1/exceptions/exceptions.py
+++ b/cognee/api/v1/exceptions/exceptions.py
@@ -1,5 +1,6 @@
 from cognee.exceptions import (
     CogneeConfigurationError,
+    CogneeValidationError
 )
 from fastapi import status
 
@@ -13,3 +14,39 @@ class InvalidConfigAttributeError(CogneeConfigurationError):
     ):
         message = f"'{attribute}' is not a valid attribute of the configuration."
         super().__init__(message, name, status_code)
+
+class DocumentNotFoundError(CogneeValidationError):
+    def __init__(
+        self,
+        message: str = "Document not found in database.",
+        name: str = "DocumentNotFoundError",
+        status_code: int = status.HTTP_404_NOT_FOUND,
+    ):
+        super().__init__(message, name, status_code)
+
+class DatasetNotFoundError(CogneeValidationError):
+    def __init__(
+        self,
+        message: str = "Dataset not found.",
+        name: str = "DatasetNotFoundError",
+        status_code: int = status.HTTP_404_NOT_FOUND,
+    ):
+        super().__init__(message, name, status_code)
+
+class DataNotFoundError(CogneeValidationError):
+    def __init__(
+        self,
+        message: str = "Data not found.",
+        name: str = "DataNotFoundError",
+        status_code: int = status.HTTP_404_NOT_FOUND,
+    ):
+        super().__init__(message, name, status_code)
+
+class DocumentSubgraphNotFoundError(CogneeValidationError):
+    def __init__(
+        self,
+        message: str = "Document subgraph not found in graph database.",
+        name: str = "DocumentSubgraphNotFoundError",
+        status_code: int = status.HTTP_404_NOT_FOUND,
+    ):
+        super().__init__(message, name, status_code)

From 544e08930b3ab43f5a936024e9d41542e60ec297 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 14:42:57 +0200
Subject: [PATCH 16/34] feat: removing invalidValueErrors

---
 cognee-mcp/src/server.py                      | 16 -----------
 cognee/api/v1/add/add.py                      |  5 ----
 cognee/api/v1/cognify/cognify.py              |  8 ------
 cognee/api/v1/search/search.py                |  7 -----
 .../data/utils/extract_keywords.py            |  2 +-
 .../databases/exceptions/__init__.py          |  2 ++
 .../databases/exceptions/exceptions.py        | 28 +++++++++++++++++++
 .../hybrid/falkordb/FalkorDBAdapter.py        |  4 +--
 .../NeptuneAnalyticsAdapter.py                |  9 +++---
 .../vector/chromadb/ChromaDBAdapter.py        |  4 +--
 .../vector/lancedb/LanceDBAdapter.py          |  4 +--
 .../vector/pgvector/PGVectorAdapter.py        |  3 +-
 .../databases/vector/qdrant/QDrantAdapter.py  |  6 ++--
 .../vector/weaviate_db/WeaviateAdapter.py     |  6 ++--
 .../llm/anthropic/adapter.py                  |  4 +--
 .../litellm_instructor/llm/gemini/adapter.py  |  6 ++--
 .../litellm_instructor/llm/get_llm_client.py  |  4 +--
 .../litellm_instructor/llm/openai/adapter.py  |  2 +-
 .../graph/cognee_graph_elements_test.py       | 11 ++++----
 19 files changed, 62 insertions(+), 69 deletions(-)

diff --git a/cognee-mcp/src/server.py b/cognee-mcp/src/server.py
index a657225f5..3e65a5eb7 100755
--- a/cognee-mcp/src/server.py
+++ b/cognee-mcp/src/server.py
@@ -221,14 +221,6 @@ async def cognify(data: str, graph_model_file: str = None, graph_model_name: str
     - The actual cognify process may take significant time depending on text length
     - Use the cognify_status tool to check the progress of the operation
 
-    Raises
-    ------
-    InvalidValueError
-        If LLM_API_KEY is not set
-    ValueError
-        If chunks exceed max token limits (reduce chunk_size)
-    DatabaseNotCreatedError
-        If databases are not properly initialized
     """
 
     async def cognify_task(
@@ -512,14 +504,6 @@ async def search(search_query: str, search_type: str) -> list:
     - Different search types produce different output formats
     - The function handles the conversion between Cognee's internal result format and MCP's output format
 
-    Raises
-    ------
-    InvalidValueError
-        If LLM_API_KEY is not set (for LLM-based search types)
-    ValueError
-        If query_text is empty or search parameters are invalid
-    NoDataError
-        If no relevant data found for the search query
     """
 
     async def search_task(search_query: str, search_type: str) -> str:
diff --git a/cognee/api/v1/add/add.py b/cognee/api/v1/add/add.py
index 3e4aaae49..a9ad42923 100644
--- a/cognee/api/v1/add/add.py
+++ b/cognee/api/v1/add/add.py
@@ -133,11 +133,6 @@ async def add(
         - VECTOR_DB_PROVIDER: "lancedb" (default), "chromadb", "qdrant", "weaviate"
         - GRAPH_DATABASE_PROVIDER: "kuzu" (default), "neo4j", "networkx"
 
-    Raises:
-        FileNotFoundError: If specified file paths don't exist
-        PermissionError: If user lacks access to files or dataset
-        UnsupportedFileTypeError: If file format cannot be processed
-        InvalidValueError: If LLM_API_KEY is not set or invalid
     """
     tasks = [
         Task(resolve_data_directories, include_subdirectories=True),
diff --git a/cognee/api/v1/cognify/cognify.py b/cognee/api/v1/cognify/cognify.py
index c6508f3a7..23984b9a6 100644
--- a/cognee/api/v1/cognify/cognify.py
+++ b/cognee/api/v1/cognify/cognify.py
@@ -177,14 +177,6 @@ async def cognify(
         - LLM_PROVIDER, LLM_MODEL, VECTOR_DB_PROVIDER, GRAPH_DATABASE_PROVIDER
         - LLM_RATE_LIMIT_ENABLED: Enable rate limiting (default: False)
         - LLM_RATE_LIMIT_REQUESTS: Max requests per interval (default: 60)
-
-    Raises:
-        DatasetNotFoundError: If specified datasets don't exist
-        PermissionError: If user lacks processing rights
-        InvalidValueError: If LLM_API_KEY is not set
-        OntologyParsingError: If ontology file is malformed
-        ValueError: If chunks exceed max token limits (reduce chunk_size)
-        DatabaseNotCreatedError: If databases are not properly initialized
     """
     tasks = await get_default_tasks(user, graph_model, chunker, chunk_size, ontology_file_path)
 
diff --git a/cognee/api/v1/search/search.py b/cognee/api/v1/search/search.py
index 66ce48cc2..f4f4831c1 100644
--- a/cognee/api/v1/search/search.py
+++ b/cognee/api/v1/search/search.py
@@ -158,13 +158,6 @@ async def search(
         - VECTOR_DB_PROVIDER: Must match what was used during cognify
         - GRAPH_DATABASE_PROVIDER: Must match what was used during cognify
 
-    Raises:
-        DatasetNotFoundError: If specified datasets don't exist or aren't accessible
-        PermissionDeniedError: If user lacks read access to requested datasets
-        NoDataError: If no relevant data found for the search query
-        InvalidValueError: If LLM_API_KEY is not set (for LLM-based search types)
-        ValueError: If query_text is empty or search parameters are invalid
-        CollectionNotFoundError: If vector collection not found (data not processed)
     """
     # We use lists from now on for datasets
     if isinstance(datasets, UUID) or isinstance(datasets, str):
diff --git a/cognee/infrastructure/data/utils/extract_keywords.py b/cognee/infrastructure/data/utils/extract_keywords.py
index 2915131a4..8085459c9 100644
--- a/cognee/infrastructure/data/utils/extract_keywords.py
+++ b/cognee/infrastructure/data/utils/extract_keywords.py
@@ -8,7 +8,7 @@ def extract_keywords(text: str) -> list[str]:
     """
     Extract keywords from the provided text string.
 
-    This function raises an InvalidValueError if the input text is empty. It processes the
+    This function raises an KeyWordExtractionError if the input text is empty. It processes the
     text to extract parts of speech, focusing on nouns, and uses TF-IDF to identify the most
     relevant keywords based on their frequency. The function returns a list of up to 15
     keywords, each having more than 3 characters.
diff --git a/cognee/infrastructure/databases/exceptions/__init__.py b/cognee/infrastructure/databases/exceptions/__init__.py
index c7d2a8feb..56deaac74 100644
--- a/cognee/infrastructure/databases/exceptions/__init__.py
+++ b/cognee/infrastructure/databases/exceptions/__init__.py
@@ -9,4 +9,6 @@ from .exceptions import (
     EntityAlreadyExistsError,
     DatabaseNotCreatedError,
     EmbeddingException,
+    MissingQueryParameterError,
+    MutuallyExclusiveQueryParametersError
 )
diff --git a/cognee/infrastructure/databases/exceptions/exceptions.py b/cognee/infrastructure/databases/exceptions/exceptions.py
index 66740fa5e..6e1bb74f6 100644
--- a/cognee/infrastructure/databases/exceptions/exceptions.py
+++ b/cognee/infrastructure/databases/exceptions/exceptions.py
@@ -102,3 +102,31 @@ class EmbeddingException(CogneeConfigurationError):
         status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
     ):
         super().__init__(message, name, status_code)
+
+class MissingQueryParameterError(CogneeValidationError):
+    """
+    Raised when neither 'query_text' nor 'query_vector' is provided,
+    and at least one is required to perform the operation.
+    """
+    def __init__(
+        self,
+        name: str = "MissingQueryParameterError",
+        status_code: int = status.HTTP_400_BAD_REQUEST,
+    ):
+        message = "One of query_text or query_vector must be provided!"
+        super().__init__(message, name, status_code)
+
+class MutuallyExclusiveQueryParametersError(CogneeValidationError):
+    """
+    Raised when both 'text' and 'embedding' are provided to the search function,
+    but only one type of input is allowed at a time.
+    """
+    def __init__(
+        self,
+        name: str = "MutuallyExclusiveQueryParametersError",
+        status_code: int = status.HTTP_400_BAD_REQUEST,
+    ):
+        message = (
+            "The search function accepts either text or embedding as input, but not both."
+        )
+        super().__init__(message, name, status_code)
diff --git a/cognee/infrastructure/databases/hybrid/falkordb/FalkorDBAdapter.py b/cognee/infrastructure/databases/hybrid/falkordb/FalkorDBAdapter.py
index 35ce7c77e..cb6899925 100644
--- a/cognee/infrastructure/databases/hybrid/falkordb/FalkorDBAdapter.py
+++ b/cognee/infrastructure/databases/hybrid/falkordb/FalkorDBAdapter.py
@@ -9,7 +9,7 @@ from typing import List, Dict, Any, Optional, Tuple, Type, Union
 
 from falkordb import FalkorDB
 
-from cognee.exceptions import InvalidValueError
+from cognee.infrastructure.databases.exceptions import MissingQueryParameterError
 from cognee.infrastructure.databases.graph.graph_db_interface import (
     GraphDBInterface,
     record_graph_changes,
@@ -721,7 +721,7 @@ class FalkorDBAdapter(VectorDBInterface, GraphDBInterface):
             Returns the search results as a result set from the graph database.
         """
         if query_text is None and query_vector is None:
-            raise InvalidValueError(message="One of query_text or query_vector must be provided!")
+            raise MissingQueryParameterError()
 
         if query_text and not query_vector:
             query_vector = (await self.embed_data([query_text]))[0]
diff --git a/cognee/infrastructure/databases/hybrid/neptune_analytics/NeptuneAnalyticsAdapter.py b/cognee/infrastructure/databases/hybrid/neptune_analytics/NeptuneAnalyticsAdapter.py
index a04e6f09e..4baf8ff13 100644
--- a/cognee/infrastructure/databases/hybrid/neptune_analytics/NeptuneAnalyticsAdapter.py
+++ b/cognee/infrastructure/databases/hybrid/neptune_analytics/NeptuneAnalyticsAdapter.py
@@ -5,7 +5,8 @@ import json
 from typing import List, Optional, Any, Dict, Type, Tuple
 from uuid import UUID
 
-from cognee.exceptions import InvalidValueError
+from cognee.infrastructure.databases.exceptions import MissingQueryParameterError
+from cognee.infrastructure.databases.exceptions import MutuallyExclusiveQueryParametersError
 from cognee.infrastructure.databases.graph.neptune_driver.adapter import NeptuneGraphDB
 from cognee.infrastructure.databases.vector.vector_db_interface import VectorDBInterface
 from cognee.infrastructure.engine import DataPoint
@@ -274,11 +275,9 @@ class NeptuneAnalyticsAdapter(NeptuneGraphDB, VectorDBInterface):
             limit = self._TOPK_UPPER_BOUND
 
         if query_vector and query_text:
-            raise InvalidValueError(
-                message="The search function accepts either text or embedding as input, but not both."
-            )
+            raise MutuallyExclusiveQueryParametersError()
         elif query_text is None and query_vector is None:
-            raise InvalidValueError(message="One of query_text or query_vector must be provided!")
+            raise MissingQueryParameterError()
         elif query_vector:
             embedding = query_vector
         else:
diff --git a/cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py b/cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py
index d4b858348..81f47dddd 100644
--- a/cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py
+++ b/cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py
@@ -3,13 +3,13 @@ from uuid import UUID
 from typing import List, Optional
 from chromadb import AsyncHttpClient, Settings
 
-from cognee.exceptions import InvalidValueError
 from cognee.shared.logging_utils import get_logger
 from cognee.modules.storage.utils import get_own_properties
 from cognee.infrastructure.engine import DataPoint
 from cognee.infrastructure.engine.utils import parse_id
 from cognee.infrastructure.databases.vector.exceptions import CollectionNotFoundError
 from cognee.infrastructure.databases.vector.models.ScoredResult import ScoredResult
+from cognee.infrastructure.databases.exceptions import MissingQueryParameterError
 
 from ..embeddings.EmbeddingEngine import EmbeddingEngine
 from ..vector_db_interface import VectorDBInterface
@@ -373,7 +373,7 @@ class ChromaDBAdapter(VectorDBInterface):
             Returns a list of ScoredResult instances representing the search results.
         """
         if query_text is None and query_vector is None:
-            raise InvalidValueError(message="One of query_text or query_vector must be provided!")
+            raise MissingQueryParameterError()
 
         if query_text and not query_vector:
             query_vector = (await self.embedding_engine.embed_text([query_text]))[0]
diff --git a/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py b/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py
index f37c83113..0184ec3ee 100644
--- a/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py
+++ b/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py
@@ -5,7 +5,7 @@ from pydantic import BaseModel
 from lancedb.pydantic import LanceModel, Vector
 from typing import Generic, List, Optional, TypeVar, Union, get_args, get_origin, get_type_hints
 
-from cognee.exceptions import InvalidValueError
+from cognee.infrastructure.databases.exceptions import MissingQueryParameterError
 from cognee.infrastructure.engine import DataPoint
 from cognee.infrastructure.engine.utils import parse_id
 from cognee.infrastructure.files.storage import get_file_storage
@@ -228,7 +228,7 @@ class LanceDBAdapter(VectorDBInterface):
         normalized: bool = True,
     ):
         if query_text is None and query_vector is None:
-            raise InvalidValueError(message="One of query_text or query_vector must be provided!")
+            raise MissingQueryParameterError()
 
         if query_text and not query_vector:
             query_vector = (await self.embedding_engine.embed_text([query_text]))[0]
diff --git a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
index 96b2056c4..4dfd9792f 100644
--- a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
+++ b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
@@ -17,6 +17,7 @@ from cognee.infrastructure.databases.relational import get_relational_engine
 
 from distributed.utils import override_distributed
 from distributed.tasks.queued_add_data_points import queued_add_data_points
+from cognee.infrastructure.databases.exceptions import MissingQueryParameterError
 
 from ...relational.ModelBase import Base
 from ...relational.sqlalchemy.SqlAlchemyAdapter import SQLAlchemyAdapter
@@ -302,7 +303,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
         with_vector: bool = False,
     ) -> List[ScoredResult]:
         if query_text is None and query_vector is None:
-            raise InvalidValueError(message="One of query_text or query_vector must be provided!")
+            raise MissingQueryParameterError()
 
         if query_text and not query_vector:
             query_vector = (await self.embedding_engine.embed_text([query_text]))[0]
diff --git a/cognee/infrastructure/databases/vector/qdrant/QDrantAdapter.py b/cognee/infrastructure/databases/vector/qdrant/QDrantAdapter.py
index 716fc969f..aa1368716 100644
--- a/cognee/infrastructure/databases/vector/qdrant/QDrantAdapter.py
+++ b/cognee/infrastructure/databases/vector/qdrant/QDrantAdapter.py
@@ -4,7 +4,7 @@ from qdrant_client import AsyncQdrantClient, models
 
 from cognee.shared.logging_utils import get_logger
 from cognee.infrastructure.engine.utils import parse_id
-from cognee.exceptions import InvalidValueError
+from cognee.infrastructure.databases.exceptions import MissingQueryParameterError
 from cognee.infrastructure.engine import DataPoint
 from cognee.infrastructure.databases.vector.exceptions import CollectionNotFoundError
 from cognee.infrastructure.databases.vector.models.ScoredResult import ScoredResult
@@ -363,7 +363,7 @@ class QDrantAdapter(VectorDBInterface):
         Search for data points in a collection based on either a textual query or a vector
         query.
 
-        Raises InvalidValueError if both query_text and query_vector are None.
+        Raises MissingQueryParameterError if both query_text and query_vector are None.
 
         Returns a list of scored results that match the search criteria.
 
@@ -388,7 +388,7 @@ class QDrantAdapter(VectorDBInterface):
         from qdrant_client.http.exceptions import UnexpectedResponse
 
         if query_text is None and query_vector is None:
-            raise InvalidValueError(message="One of query_text or query_vector must be provided!")
+            raise MissingQueryParameterError()
 
         if not await self.has_collection(collection_name):
             return []
diff --git a/cognee/infrastructure/databases/vector/weaviate_db/WeaviateAdapter.py b/cognee/infrastructure/databases/vector/weaviate_db/WeaviateAdapter.py
index 00a6a0411..db68e004e 100644
--- a/cognee/infrastructure/databases/vector/weaviate_db/WeaviateAdapter.py
+++ b/cognee/infrastructure/databases/vector/weaviate_db/WeaviateAdapter.py
@@ -3,7 +3,7 @@ from typing import List, Optional
 from tenacity import retry, retry_if_exception, stop_after_attempt, wait_exponential
 
 from cognee.shared.logging_utils import get_logger
-from cognee.exceptions import InvalidValueError
+from cognee.infrastructure.databases.exceptions import MissingQueryParameterError
 from cognee.infrastructure.engine import DataPoint
 from cognee.infrastructure.engine.utils import parse_id
 from cognee.infrastructure.databases.vector.exceptions import CollectionNotFoundError
@@ -390,7 +390,7 @@ class WeaviateAdapter(VectorDBInterface):
         """
         Perform a search on a collection using either a text query or a vector query.
 
-        Return scored results based on the search criteria provided. Raise InvalidValueError if
+        Return scored results based on the search criteria provided. Raise MissingQueryParameterError if
         no query is provided.
 
         Parameters:
@@ -413,7 +413,7 @@ class WeaviateAdapter(VectorDBInterface):
         import weaviate.exceptions
 
         if query_text is None and query_vector is None:
-            raise InvalidValueError(message="One of query_text or query_vector must be provided!")
+            raise MissingQueryParameterError()
 
         if query_vector is None:
             query_vector = (await self.embed_data([query_text]))[0]
diff --git a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py
index 636e6c0f2..6845fb6aa 100644
--- a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py
@@ -2,7 +2,7 @@ from typing import Type
 from pydantic import BaseModel
 import instructor
 
-from cognee.exceptions import InvalidValueError
+from cognee.infrastructure.llm.exceptions import MissingSystemPromptPathError
 from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.llm_interface import (
     LLMInterface,
 )
@@ -89,7 +89,7 @@ class AnthropicAdapter(LLMInterface):
         if not text_input:
             text_input = "No user input provided."
         if not system_prompt:
-            raise InvalidValueError(message="No system prompt path provided.")
+            raise MissingSystemPromptPathError()
 
         system_prompt = LLMGateway.read_query_prompt(system_prompt)
 
diff --git a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py
index 61d42ff5f..3cde1fdc4 100644
--- a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py
@@ -5,7 +5,7 @@ from litellm import acompletion, JSONSchemaValidationError
 
 from cognee.shared.logging_utils import get_logger
 from cognee.modules.observability.get_observe import get_observe
-from cognee.exceptions import InvalidValueError
+from cognee.infrastructure.llm.exceptions import MissingSystemPromptPathError
 from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.llm_interface import (
     LLMInterface,
 )
@@ -118,7 +118,7 @@ class GeminiAdapter(LLMInterface):
         """
         Format and display the prompt for a user query.
 
-        Raises an InvalidValueError if no system prompt is provided.
+        Raises an MissingQueryParameterError if no system prompt is provided.
 
         Parameters:
         -----------
@@ -135,7 +135,7 @@ class GeminiAdapter(LLMInterface):
         if not text_input:
             text_input = "No user input provided."
         if not system_prompt:
-            raise InvalidValueError(message="No system prompt path provided.")
+            raise MissingSystemPromptPathError()
         system_prompt = LLMGateway.read_query_prompt(system_prompt)
 
         formatted_prompt = (
diff --git a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
index 3006a795b..fd347aef3 100644
--- a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
@@ -38,7 +38,7 @@ def get_llm_client():
 
     This function retrieves the configuration for the LLM provider and model, and
     initializes the appropriate LLM client adapter accordingly. It raises an
-    InvalidValueError if the LLM API key is not set for certain providers or if the provider
+    LLMAPIKeyNotSetError if the LLM API key is not set for certain providers or if the provider
     is unsupported.
 
     Returns:
@@ -62,7 +62,7 @@ def get_llm_client():
 
     if provider == LLMProvider.OPENAI:
         if llm_config.llm_api_key is None:
-            raise InvalidValueError(message="LLM API key is not set.")
+            raise LLMAPIKeyNotSetError()
 
         from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.openai.adapter import (
             OpenAIAdapter,
diff --git a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
index 38e1bc82e..4126b3e13 100644
--- a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
@@ -307,7 +307,7 @@ class OpenAIAdapter(LLMInterface):
         Format and display the prompt for a user query.
 
         This method formats the prompt using the provided user input and system prompt,
-        returning a string representation. Raises InvalidValueError if the system prompt is not
+        returning a string representation. Raises MissingSystemPromptPathError if the system prompt is not
         provided.
 
         Parameters:
diff --git a/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py b/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
index 000856b12..a852bcee3 100644
--- a/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
+++ b/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
@@ -1,9 +1,8 @@
 import numpy as np
 import pytest
 
-from cognee.exceptions import InvalidValueError
 from cognee.modules.graph.cognee_graph.CogneeGraphElements import Edge, Node
-
+from cognee.modules.graph.exceptions import InvalidDimensionsError, DimensionOutOfRangeError
 
 def test_node_initialization():
     """Test that a Node is initialized correctly."""
@@ -16,7 +15,7 @@ def test_node_initialization():
 
 def test_node_invalid_dimension():
     """Test that initializing a Node with a non-positive dimension raises an error."""
-    with pytest.raises(InvalidValueError, match="Dimension must be a positive integer"):
+    with pytest.raises(InvalidDimensionsError, match="Dimensions must be a positive integers"):
         Node("node1", dimension=0)
 
 
@@ -69,7 +68,7 @@ def test_is_node_alive_in_dimension():
 def test_node_alive_invalid_dimension():
     """Test that checking alive status with an invalid dimension raises an error."""
     node = Node("node1", dimension=1)
-    with pytest.raises(InvalidValueError, match="Dimension 1 is out of range"):
+    with pytest.raises(DimensionOutOfRangeError, match="Dimension 1 is out of range"):
         node.is_node_alive_in_dimension(1)
 
 
@@ -106,7 +105,7 @@ def test_edge_invalid_dimension():
     """Test that initializing an Edge with a non-positive dimension raises an error."""
     node1 = Node("node1")
     node2 = Node("node2")
-    with pytest.raises(InvalidValueError, match="Dimensions must be a positive integer."):
+    with pytest.raises(DimensionOutOfRangeError, match="Dimensions must be a positive integer."):
         Edge(node1, node2, dimension=0)
 
 
@@ -125,7 +124,7 @@ def test_edge_alive_invalid_dimension():
     node1 = Node("node1")
     node2 = Node("node2")
     edge = Edge(node1, node2, dimension=1)
-    with pytest.raises(InvalidValueError, match="Dimension 1 is out of range"):
+    with pytest.raises(DimensionOutOfRangeError, match="Dimension 1 is out of range"):
         edge.is_edge_alive_in_dimension(1)
 
 

From 885f7c3f9915adada0d2ef62e679d60e4ac335e7 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 14:58:56 +0200
Subject: [PATCH 17/34] chore: fixing graph elements tests

---
 cognee/modules/graph/cognee_graph/CogneeGraphElements.py  | 2 +-
 .../unit/modules/graph/cognee_graph_elements_test.py      | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/cognee/modules/graph/cognee_graph/CogneeGraphElements.py b/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
index c22cc5c18..2d8917d4a 100644
--- a/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
+++ b/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
@@ -106,7 +106,7 @@ class Edge:
         dimension: int = 1,
     ):
         if dimension <= 0:
-            InvalidDimensionsError()
+            raise InvalidDimensionsError()
         self.node1 = node1
         self.node2 = node2
         self.attributes = attributes if attributes is not None else {}
diff --git a/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py b/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
index a852bcee3..a8fb17268 100644
--- a/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
+++ b/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
@@ -15,7 +15,7 @@ def test_node_initialization():
 
 def test_node_invalid_dimension():
     """Test that initializing a Node with a non-positive dimension raises an error."""
-    with pytest.raises(InvalidDimensionsError, match="Dimensions must be a positive integers"):
+    with pytest.raises(InvalidDimensionsError):
         Node("node1", dimension=0)
 
 
@@ -68,7 +68,7 @@ def test_is_node_alive_in_dimension():
 def test_node_alive_invalid_dimension():
     """Test that checking alive status with an invalid dimension raises an error."""
     node = Node("node1", dimension=1)
-    with pytest.raises(DimensionOutOfRangeError, match="Dimension 1 is out of range"):
+    with pytest.raises(DimensionOutOfRangeError):
         node.is_node_alive_in_dimension(1)
 
 
@@ -105,7 +105,7 @@ def test_edge_invalid_dimension():
     """Test that initializing an Edge with a non-positive dimension raises an error."""
     node1 = Node("node1")
     node2 = Node("node2")
-    with pytest.raises(DimensionOutOfRangeError, match="Dimensions must be a positive integer."):
+    with pytest.raises(InvalidDimensionsError):
         Edge(node1, node2, dimension=0)
 
 
@@ -124,7 +124,7 @@ def test_edge_alive_invalid_dimension():
     node1 = Node("node1")
     node2 = Node("node2")
     edge = Edge(node1, node2, dimension=1)
-    with pytest.raises(DimensionOutOfRangeError, match="Dimension 1 is out of range"):
+    with pytest.raises(DimensionOutOfRangeError):
         edge.is_edge_alive_in_dimension(1)
 
 

From d1bfeaa0f27a8f53d2b58819dd88d155e36a9a9d Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 15:00:25 +0200
Subject: [PATCH 18/34] fix: fixes search unit test error expectation

---
 cognee/tests/unit/modules/search/search_methods_test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cognee/tests/unit/modules/search/search_methods_test.py b/cognee/tests/unit/modules/search/search_methods_test.py
index 14712f6d2..8e9afff1c 100644
--- a/cognee/tests/unit/modules/search/search_methods_test.py
+++ b/cognee/tests/unit/modules/search/search_methods_test.py
@@ -5,7 +5,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 import pytest
 from pylint.checkers.utils import node_type
 
-from cognee.exceptions import InvalidValueError
+from cognee.modules.search.exceptions import UnsupportedSearchTypeError
 from cognee.modules.search.methods.search import search, specific_search
 from cognee.modules.search.types import SearchType
 from cognee.modules.users.models import User
@@ -217,7 +217,7 @@ async def test_specific_search_invalid_type(mock_user):
     query_type = "INVALID_TYPE"  # Not a valid SearchType
 
     # Execute and verify
-    with pytest.raises(InvalidValueError) as excinfo:
+    with pytest.raises(UnsupportedSearchTypeError) as excinfo:
         await specific_search(query_type, query, mock_user)
 
     assert "Unsupported search type" in str(excinfo.value)

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 19/34] 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()
 

From 59c9204a74c8c357e784335223f6136c133e1599 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 15:03:44 +0200
Subject: [PATCH 20/34] Update exceptions.py

---
 cognee/infrastructure/databases/vector/exceptions/exceptions.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/infrastructure/databases/vector/exceptions/exceptions.py b/cognee/infrastructure/databases/vector/exceptions/exceptions.py
index ee0712433..ecd106c0d 100644
--- a/cognee/infrastructure/databases/vector/exceptions/exceptions.py
+++ b/cognee/infrastructure/databases/vector/exceptions/exceptions.py
@@ -6,7 +6,7 @@ class CollectionNotFoundError(CogneeValidationError):
     """
     Represents an error that occurs when a requested collection cannot be found.
 
-    This class extends the CriticalError to handle specific cases where a requested
+    This class extends the CogneeValidationError to handle specific cases where a requested
     collection is unavailable. It can be initialized with a custom message and allows for
     logging options including log level and whether to log the error.
     """

From 68327d3ab9fe0b99aa6aef7fc4fe3bd78554ef1c Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 15:14:43 +0200
Subject: [PATCH 21/34] chore: Changing base classes for exceptions that were
 already defined

---
 .../databases/graph/neptune_driver/exceptions.py     |  4 ++--
 cognee/modules/ingestion/exceptions/exceptions.py    |  4 ++--
 cognee/modules/ontology/exceptions/exceptions.py     |  8 ++++----
 cognee/modules/pipelines/exceptions/exceptions.py    |  4 ++--
 cognee/modules/users/exceptions/exceptions.py        | 12 ++++++------
 cognee/shared/exceptions/exceptions.py               |  4 ++--
 cognee/tasks/completion/exceptions/exceptions.py     |  4 ++--
 7 files changed, 20 insertions(+), 20 deletions(-)

diff --git a/cognee/infrastructure/databases/graph/neptune_driver/exceptions.py b/cognee/infrastructure/databases/graph/neptune_driver/exceptions.py
index 57d54d74d..d78936202 100644
--- a/cognee/infrastructure/databases/graph/neptune_driver/exceptions.py
+++ b/cognee/infrastructure/databases/graph/neptune_driver/exceptions.py
@@ -3,11 +3,11 @@
 This module defines custom exceptions for Neptune Analytics operations.
 """
 
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import CogneeSystemError
 from fastapi import status
 
 
-class NeptuneAnalyticsError(CogneeApiError):
+class NeptuneAnalyticsError(CogneeSystemError):
     """Base exception for Neptune Analytics operations."""
 
     def __init__(
diff --git a/cognee/modules/ingestion/exceptions/exceptions.py b/cognee/modules/ingestion/exceptions/exceptions.py
index 08991a946..d43b69d60 100644
--- a/cognee/modules/ingestion/exceptions/exceptions.py
+++ b/cognee/modules/ingestion/exceptions/exceptions.py
@@ -1,8 +1,8 @@
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import CogneeValidationError
 from fastapi import status
 
 
-class IngestionError(CogneeApiError):
+class IngestionError(CogneeValidationError):
     def __init__(
         self,
         message: str = "Type of data sent to classify not supported.",
diff --git a/cognee/modules/ontology/exceptions/exceptions.py b/cognee/modules/ontology/exceptions/exceptions.py
index 511e41524..daa8dcdb5 100644
--- a/cognee/modules/ontology/exceptions/exceptions.py
+++ b/cognee/modules/ontology/exceptions/exceptions.py
@@ -1,8 +1,8 @@
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import CogneeSystemError
 from fastapi import status
 
 
-class OntologyInitializationError(CogneeApiError):
+class OntologyInitializationError(CogneeSystemError):
     def __init__(
         self,
         message: str = "Ontology initialization failed",
@@ -12,7 +12,7 @@ class OntologyInitializationError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class FindClosestMatchError(CogneeApiError):
+class FindClosestMatchError(CogneeSystemError):
     def __init__(
         self,
         message: str = "Error in find_closest_match",
@@ -22,7 +22,7 @@ class FindClosestMatchError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class GetSubgraphError(CogneeApiError):
+class GetSubgraphError(CogneeSystemError):
     def __init__(
         self,
         message: str = "Failed to retrieve subgraph",
diff --git a/cognee/modules/pipelines/exceptions/exceptions.py b/cognee/modules/pipelines/exceptions/exceptions.py
index 0a4863075..646a51286 100644
--- a/cognee/modules/pipelines/exceptions/exceptions.py
+++ b/cognee/modules/pipelines/exceptions/exceptions.py
@@ -1,8 +1,8 @@
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import CogneeSystemError
 from fastapi import status
 
 
-class PipelineRunFailedError(CogneeApiError):
+class PipelineRunFailedError(CogneeSystemError):
     def __init__(
         self,
         message: str = "Pipeline run failed.",
diff --git a/cognee/modules/users/exceptions/exceptions.py b/cognee/modules/users/exceptions/exceptions.py
index a7484c05e..85ca8ab63 100644
--- a/cognee/modules/users/exceptions/exceptions.py
+++ b/cognee/modules/users/exceptions/exceptions.py
@@ -1,8 +1,8 @@
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import CogneeValidationError
 from fastapi import status
 
 
-class RoleNotFoundError(CogneeApiError):
+class RoleNotFoundError(CogneeValidationError):
     """User group not found"""
 
     def __init__(
@@ -14,7 +14,7 @@ class RoleNotFoundError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class TenantNotFoundError(CogneeApiError):
+class TenantNotFoundError(CogneeValidationError):
     """User group not found"""
 
     def __init__(
@@ -26,7 +26,7 @@ class TenantNotFoundError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class UserNotFoundError(CogneeApiError):
+class UserNotFoundError(CogneeValidationError):
     """User not found"""
 
     def __init__(
@@ -38,7 +38,7 @@ class UserNotFoundError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class PermissionDeniedError(CogneeApiError):
+class PermissionDeniedError(CogneeValidationError):
     def __init__(
         self,
         message: str = "User does not have permission on documents.",
@@ -48,7 +48,7 @@ class PermissionDeniedError(CogneeApiError):
         super().__init__(message, name, status_code)
 
 
-class PermissionNotFoundError(CogneeApiError):
+class PermissionNotFoundError(CogneeValidationError):
     def __init__(
         self,
         message: str = "Permission type does not exist.",
diff --git a/cognee/shared/exceptions/exceptions.py b/cognee/shared/exceptions/exceptions.py
index 4b4164995..43084e04b 100644
--- a/cognee/shared/exceptions/exceptions.py
+++ b/cognee/shared/exceptions/exceptions.py
@@ -1,8 +1,8 @@
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import CogneeValidationError
 from fastapi import status
 
 
-class IngestionError(CogneeApiError):
+class IngestionError(CogneeValidationError):
     def __init__(
         self,
         message: str = "Failed to load data.",
diff --git a/cognee/tasks/completion/exceptions/exceptions.py b/cognee/tasks/completion/exceptions/exceptions.py
index ac105a966..5052e2a77 100644
--- a/cognee/tasks/completion/exceptions/exceptions.py
+++ b/cognee/tasks/completion/exceptions/exceptions.py
@@ -1,8 +1,8 @@
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import CogneeValidationError
 from fastapi import status
 
 
-class NoRelevantDataError(CogneeApiError):
+class NoRelevantDataError(CogneeValidationError):
     """
     Represents an error when no relevant data is found during a search. This class is a
     subclass of CogneeApiError.

From da40365932ce7056719dd323d6c721214e6919fa Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 15:15:39 +0200
Subject: [PATCH 22/34] ruff formatting

---
 cognee/api/v1/exceptions/__init__.py                     | 2 +-
 cognee/api/v1/exceptions/exceptions.py                   | 9 +++++----
 cognee/exceptions/exceptions.py                          | 1 -
 cognee/infrastructure/data/exceptions/__init__.py        | 4 +---
 cognee/infrastructure/databases/exceptions/__init__.py   | 2 +-
 cognee/infrastructure/databases/exceptions/exceptions.py | 8 +++++---
 cognee/infrastructure/llm/exceptions.py                  | 3 +++
 .../litellm_instructor/llm/openai/adapter.py             | 5 ++++-
 cognee/modules/graph/cognee_graph/CogneeGraph.py         | 6 +++++-
 cognee/modules/graph/cognee_graph/CogneeGraphElements.py | 1 +
 cognee/modules/graph/exceptions/exceptions.py            | 2 ++
 cognee/modules/search/exceptions/__init__.py             | 4 +---
 .../unit/modules/graph/cognee_graph_elements_test.py     | 1 +
 13 files changed, 30 insertions(+), 18 deletions(-)

diff --git a/cognee/api/v1/exceptions/__init__.py b/cognee/api/v1/exceptions/__init__.py
index b2948934c..5767a5801 100644
--- a/cognee/api/v1/exceptions/__init__.py
+++ b/cognee/api/v1/exceptions/__init__.py
@@ -9,5 +9,5 @@ from .exceptions import (
     DocumentNotFoundError,
     DatasetNotFoundError,
     DataNotFoundError,
-    DocumentSubgraphNotFoundError
+    DocumentSubgraphNotFoundError,
 )
diff --git a/cognee/api/v1/exceptions/exceptions.py b/cognee/api/v1/exceptions/exceptions.py
index 8b65691f1..12dfa9d37 100644
--- a/cognee/api/v1/exceptions/exceptions.py
+++ b/cognee/api/v1/exceptions/exceptions.py
@@ -1,7 +1,4 @@
-from cognee.exceptions import (
-    CogneeConfigurationError,
-    CogneeValidationError
-)
+from cognee.exceptions import CogneeConfigurationError, CogneeValidationError
 from fastapi import status
 
 
@@ -15,6 +12,7 @@ class InvalidConfigAttributeError(CogneeConfigurationError):
         message = f"'{attribute}' is not a valid attribute of the configuration."
         super().__init__(message, name, status_code)
 
+
 class DocumentNotFoundError(CogneeValidationError):
     def __init__(
         self,
@@ -24,6 +22,7 @@ class DocumentNotFoundError(CogneeValidationError):
     ):
         super().__init__(message, name, status_code)
 
+
 class DatasetNotFoundError(CogneeValidationError):
     def __init__(
         self,
@@ -33,6 +32,7 @@ class DatasetNotFoundError(CogneeValidationError):
     ):
         super().__init__(message, name, status_code)
 
+
 class DataNotFoundError(CogneeValidationError):
     def __init__(
         self,
@@ -42,6 +42,7 @@ class DataNotFoundError(CogneeValidationError):
     ):
         super().__init__(message, name, status_code)
 
+
 class DocumentSubgraphNotFoundError(CogneeValidationError):
     def __init__(
         self,
diff --git a/cognee/exceptions/exceptions.py b/cognee/exceptions/exceptions.py
index 9b6cef21d..d956d9cef 100644
--- a/cognee/exceptions/exceptions.py
+++ b/cognee/exceptions/exceptions.py
@@ -89,4 +89,3 @@ class CogneeTransientError(CogneeApiError):
         log_level="ERROR",
     ):
         super().__init__(message, name, status_code, log, log_level)
-
diff --git a/cognee/infrastructure/data/exceptions/__init__.py b/cognee/infrastructure/data/exceptions/__init__.py
index 6735200ed..cc5f6044f 100644
--- a/cognee/infrastructure/data/exceptions/__init__.py
+++ b/cognee/infrastructure/data/exceptions/__init__.py
@@ -4,6 +4,4 @@ Custom exceptions for the Cognee API.
 This module defines a set of exceptions for handling various data errors
 """
 
-from .exceptions import (
-    KeywordExtractionError
-)
+from .exceptions import KeywordExtractionError
diff --git a/cognee/infrastructure/databases/exceptions/__init__.py b/cognee/infrastructure/databases/exceptions/__init__.py
index 56deaac74..2969b1c59 100644
--- a/cognee/infrastructure/databases/exceptions/__init__.py
+++ b/cognee/infrastructure/databases/exceptions/__init__.py
@@ -10,5 +10,5 @@ from .exceptions import (
     DatabaseNotCreatedError,
     EmbeddingException,
     MissingQueryParameterError,
-    MutuallyExclusiveQueryParametersError
+    MutuallyExclusiveQueryParametersError,
 )
diff --git a/cognee/infrastructure/databases/exceptions/exceptions.py b/cognee/infrastructure/databases/exceptions/exceptions.py
index 6e1bb74f6..c240d3133 100644
--- a/cognee/infrastructure/databases/exceptions/exceptions.py
+++ b/cognee/infrastructure/databases/exceptions/exceptions.py
@@ -103,11 +103,13 @@ class EmbeddingException(CogneeConfigurationError):
     ):
         super().__init__(message, name, status_code)
 
+
 class MissingQueryParameterError(CogneeValidationError):
     """
     Raised when neither 'query_text' nor 'query_vector' is provided,
     and at least one is required to perform the operation.
     """
+
     def __init__(
         self,
         name: str = "MissingQueryParameterError",
@@ -116,17 +118,17 @@ class MissingQueryParameterError(CogneeValidationError):
         message = "One of query_text or query_vector must be provided!"
         super().__init__(message, name, status_code)
 
+
 class MutuallyExclusiveQueryParametersError(CogneeValidationError):
     """
     Raised when both 'text' and 'embedding' are provided to the search function,
     but only one type of input is allowed at a time.
     """
+
     def __init__(
         self,
         name: str = "MutuallyExclusiveQueryParametersError",
         status_code: int = status.HTTP_400_BAD_REQUEST,
     ):
-        message = (
-            "The search function accepts either text or embedding as input, but not both."
-        )
+        message = "The search function accepts either text or embedding as input, but not both."
         super().__init__(message, name, status_code)
diff --git a/cognee/infrastructure/llm/exceptions.py b/cognee/infrastructure/llm/exceptions.py
index c8f5726fb..1d390a951 100644
--- a/cognee/infrastructure/llm/exceptions.py
+++ b/cognee/infrastructure/llm/exceptions.py
@@ -9,6 +9,7 @@ class LLMAPIKeyNotSetError(CogneeValidationError):
     """
     Raised when the LLM API key is not set in the configuration.
     """
+
     def __init__(self, message: str = "LLM API key is not set."):
         super().__init__(message=message, name="LLMAPIKeyNotSetError")
 
@@ -17,10 +18,12 @@ class UnsupportedLLMProviderError(CogneeValidationError):
     """
     Raised when an unsupported LLM provider is specified in the configuration.
     """
+
     def __init__(self, provider: str):
         message = f"Unsupported LLM provider: {provider}"
         super().__init__(message=message, name="UnsupportedLLMProviderError")
 
+
 class MissingSystemPromptPathError(CogneeValidationError):
     def __init__(
         self,
diff --git a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
index 4126b3e13..c3c215896 100644
--- a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
@@ -11,7 +11,10 @@ from cognee.infrastructure.llm.LLMGateway import LLMGateway
 from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.llm_interface import (
     LLMInterface,
 )
-from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError, MissingSystemPromptPathError
+from cognee.infrastructure.llm.exceptions import (
+    ContentPolicyFilterError,
+    MissingSystemPromptPathError,
+)
 from cognee.infrastructure.files.utils.open_data_file import open_data_file
 from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     rate_limit_async,
diff --git a/cognee/modules/graph/cognee_graph/CogneeGraph.py b/cognee/modules/graph/cognee_graph/CogneeGraph.py
index bdafaf238..ed867ae24 100644
--- a/cognee/modules/graph/cognee_graph/CogneeGraph.py
+++ b/cognee/modules/graph/cognee_graph/CogneeGraph.py
@@ -2,7 +2,11 @@ import time
 from cognee.shared.logging_utils import get_logger
 from typing import List, Dict, Union, Optional, Type
 
-from cognee.modules.graph.exceptions import EntityNotFoundError, EntityAlreadyExistsError, InvalidDimensionsError
+from cognee.modules.graph.exceptions import (
+    EntityNotFoundError,
+    EntityAlreadyExistsError,
+    InvalidDimensionsError,
+)
 from cognee.infrastructure.databases.graph.graph_db_interface import GraphDBInterface
 from cognee.modules.graph.cognee_graph.CogneeGraphElements import Node, Edge
 from cognee.modules.graph.cognee_graph.CogneeAbstractGraph import CogneeAbstractGraph
diff --git a/cognee/modules/graph/cognee_graph/CogneeGraphElements.py b/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
index 2d8917d4a..0ca9c4fb9 100644
--- a/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
+++ b/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
@@ -2,6 +2,7 @@ import numpy as np
 from typing import List, Dict, Optional, Any, Union
 from cognee.modules.graph.exceptions import InvalidDimensionsError, DimensionOutOfRangeError
 
+
 class Node:
     """
     Represents a node in a graph.
diff --git a/cognee/modules/graph/exceptions/exceptions.py b/cognee/modules/graph/exceptions/exceptions.py
index 6fb2d400d..67f4200ff 100644
--- a/cognee/modules/graph/exceptions/exceptions.py
+++ b/cognee/modules/graph/exceptions/exceptions.py
@@ -25,6 +25,7 @@ class EntityAlreadyExistsError(CogneeValidationError):
     ):
         super().__init__(message, name, status_code)
 
+
 class InvalidDimensionsError(CogneeValidationError):
     def __init__(
         self,
@@ -34,6 +35,7 @@ class InvalidDimensionsError(CogneeValidationError):
         message = "Dimensions must be positive integers."
         super().__init__(message, name, status_code)
 
+
 class DimensionOutOfRangeError(CogneeValidationError):
     def __init__(
         self,
diff --git a/cognee/modules/search/exceptions/__init__.py b/cognee/modules/search/exceptions/__init__.py
index ffb30f428..a019da249 100644
--- a/cognee/modules/search/exceptions/__init__.py
+++ b/cognee/modules/search/exceptions/__init__.py
@@ -4,6 +4,4 @@ Custom exceptions for the Cognee API.
 This module defines a set of exceptions for handling various data errors
 """
 
-from .exceptions import (
-    UnsupportedSearchTypeError
-)
+from .exceptions import UnsupportedSearchTypeError
diff --git a/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py b/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
index a8fb17268..37ba113b5 100644
--- a/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
+++ b/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
@@ -4,6 +4,7 @@ import pytest
 from cognee.modules.graph.cognee_graph.CogneeGraphElements import Edge, Node
 from cognee.modules.graph.exceptions import InvalidDimensionsError, DimensionOutOfRangeError
 
+
 def test_node_initialization():
     """Test that a Node is initialized correctly."""
     node = Node("node1", {"attr1": "value1"}, dimension=2)

From 7af4775c562265a87c5da92c72aeee0789a84fa9 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 15:17:01 +0200
Subject: [PATCH 23/34] chore: changed pypdf error base class

---
 .../data/processing/document_types/exceptions/exceptions.py   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cognee/modules/data/processing/document_types/exceptions/exceptions.py b/cognee/modules/data/processing/document_types/exceptions/exceptions.py
index b5126a8a7..a05e64d60 100644
--- a/cognee/modules/data/processing/document_types/exceptions/exceptions.py
+++ b/cognee/modules/data/processing/document_types/exceptions/exceptions.py
@@ -1,8 +1,8 @@
-from cognee.exceptions import CogneeApiError
+from cognee.exceptions import CogneeSystemError
 from fastapi import status
 
 
-class PyPdfInternalError(CogneeApiError):
+class PyPdfInternalError(CogneeSystemError):
     """Internal pypdf error"""
 
     def __init__(

From 66d2c7512811bcb38ed9ffa31d81753d1a916d2a Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 15:20:24 +0200
Subject: [PATCH 24/34] chore: changing docstring

---
 cognee/tasks/completion/exceptions/exceptions.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/tasks/completion/exceptions/exceptions.py b/cognee/tasks/completion/exceptions/exceptions.py
index 5052e2a77..e5ab5b67c 100644
--- a/cognee/tasks/completion/exceptions/exceptions.py
+++ b/cognee/tasks/completion/exceptions/exceptions.py
@@ -5,7 +5,7 @@ from fastapi import status
 class NoRelevantDataError(CogneeValidationError):
     """
     Represents an error when no relevant data is found during a search. This class is a
-    subclass of CogneeApiError.
+    subclass of CogneeValidationError.
 
     Public methods:
 

From 8bd2416bc75b0037915098d69cd24b953ce65ec9 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 15:36:17 +0200
Subject: [PATCH 25/34] chore: fixing delete exception import in dataset router

---
 cognee/api/v1/datasets/routers/get_datasets_router.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/api/v1/datasets/routers/get_datasets_router.py b/cognee/api/v1/datasets/routers/get_datasets_router.py
index 627be226d..a6938d764 100644
--- a/cognee/api/v1/datasets/routers/get_datasets_router.py
+++ b/cognee/api/v1/datasets/routers/get_datasets_router.py
@@ -13,7 +13,7 @@ from cognee.infrastructure.databases.relational import get_relational_engine
 from cognee.modules.data.methods import get_authorized_existing_datasets
 from cognee.modules.data.methods import create_dataset, get_datasets_by_name
 from cognee.shared.logging_utils import get_logger
-from cognee.api.v1.delete.exceptions import DataNotFoundError, DatasetNotFoundError
+from cognee.api.v1.exceptions import DataNotFoundError, DatasetNotFoundError
 from cognee.modules.users.models import User
 from cognee.modules.users.methods import get_authenticated_user
 from cognee.modules.users.permissions.methods import (

From d14d31adbe9f32e92c625909b9d8462156c6c912 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 15:39:11 +0200
Subject: [PATCH 26/34] chore: updating delete_by_id test

---
 cognee/tests/test_delete_by_id.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/tests/test_delete_by_id.py b/cognee/tests/test_delete_by_id.py
index 36e0a57f5..8fc5395eb 100644
--- a/cognee/tests/test_delete_by_id.py
+++ b/cognee/tests/test_delete_by_id.py
@@ -7,7 +7,7 @@ from cognee.shared.logging_utils import get_logger
 from cognee.modules.users.methods import get_default_user, create_user
 from cognee.modules.users.permissions.methods import authorized_give_permission_on_datasets
 from cognee.modules.data.methods import get_dataset_data, get_datasets_by_name
-from cognee.api.v1.delete.exceptions import DocumentNotFoundError, DatasetNotFoundError
+from cognee.api.v1.exceptions import DocumentNotFoundError, DatasetNotFoundError
 
 logger = get_logger()
 

From 748e9fad86d924edb9d78d00f22242c9d936b9d5 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 16:17:58 +0200
Subject: [PATCH 27/34] feat: adds s3 file system not found error to ingestion

---
 cognee/tasks/ingestion/exceptions/__init__.py      | 10 ++++++++++
 cognee/tasks/ingestion/exceptions/exceptions.py    | 12 ++++++++++++
 cognee/tasks/ingestion/resolve_data_directories.py |  5 +++++
 3 files changed, 27 insertions(+)
 create mode 100644 cognee/tasks/ingestion/exceptions/__init__.py
 create mode 100644 cognee/tasks/ingestion/exceptions/exceptions.py

diff --git a/cognee/tasks/ingestion/exceptions/__init__.py b/cognee/tasks/ingestion/exceptions/__init__.py
new file mode 100644
index 000000000..5c0a0b02f
--- /dev/null
+++ b/cognee/tasks/ingestion/exceptions/__init__.py
@@ -0,0 +1,10 @@
+"""
+Custom exceptions for the Cognee API.
+
+This module defines a set of exceptions for handling various application errors,
+such as System, Validation, Configuration or TransientErrors
+"""
+
+from .exceptions import (
+    S3FileSystemNotFoundError
+)
diff --git a/cognee/tasks/ingestion/exceptions/exceptions.py b/cognee/tasks/ingestion/exceptions/exceptions.py
new file mode 100644
index 000000000..9d07b9ab3
--- /dev/null
+++ b/cognee/tasks/ingestion/exceptions/exceptions.py
@@ -0,0 +1,12 @@
+from cognee.exceptions import CogneeSystemError
+from fastapi import status
+
+
+class S3FileSystemNotFoundError(CogneeSystemError):
+    def __init__(
+        self,
+        name: str = "S3FileSystemNotFoundError",
+        status_code: int = status.HTTP_500_INTERNAL_SERVER_ERROR,
+    ):
+        message = "Could not find S3FileSystem."
+        super().__init__(message, name, status_code)
diff --git a/cognee/tasks/ingestion/resolve_data_directories.py b/cognee/tasks/ingestion/resolve_data_directories.py
index 0f2f2a85f..1d3124a0c 100644
--- a/cognee/tasks/ingestion/resolve_data_directories.py
+++ b/cognee/tasks/ingestion/resolve_data_directories.py
@@ -1,6 +1,9 @@
 import os
 from urllib.parse import urlparse
 from typing import List, Union, BinaryIO
+
+from cognee.tasks.ingestion.exceptions import S3FileSystemNotFoundError
+from cognee.exceptions import CogneeSystemError
 from cognee.infrastructure.files.storage.s3_config import get_s3_config
 
 
@@ -54,6 +57,8 @@ async def resolve_data_directories(
                             else:
                                 s3_files.append(key)
                     resolved_data.extend(s3_files)
+                else:
+                    raise S3FileSystemNotFoundError()
 
             elif os.path.isdir(item):  # If it's a directory
                 if include_subdirectories:

From dbb967fda8ef5e09c0b0b7be102d17ab2219fc64 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 16:28:05 +0200
Subject: [PATCH 28/34] chore: updates neptune exception base + ruff

---
 .../graph/neptune_driver/exceptions.py        | 23 +++++++++++--------
 cognee/tasks/ingestion/exceptions/__init__.py |  4 +---
 2 files changed, 15 insertions(+), 12 deletions(-)

diff --git a/cognee/infrastructure/databases/graph/neptune_driver/exceptions.py b/cognee/infrastructure/databases/graph/neptune_driver/exceptions.py
index d78936202..a200aad39 100644
--- a/cognee/infrastructure/databases/graph/neptune_driver/exceptions.py
+++ b/cognee/infrastructure/databases/graph/neptune_driver/exceptions.py
@@ -3,7 +3,12 @@
 This module defines custom exceptions for Neptune Analytics operations.
 """
 
-from cognee.exceptions import CogneeSystemError
+from cognee.exceptions import (
+    CogneeSystemError,
+    CogneeTransientError,
+    CogneeValidationError,
+    CogneeConfigurationError,
+)
 from fastapi import status
 
 
@@ -19,7 +24,7 @@ class NeptuneAnalyticsError(CogneeSystemError):
         super().__init__(message, name, status_code)
 
 
-class NeptuneAnalyticsConnectionError(NeptuneAnalyticsError):
+class NeptuneAnalyticsConnectionError(CogneeTransientError):
     """Exception raised when connection to Neptune Analytics fails."""
 
     def __init__(
@@ -31,7 +36,7 @@ class NeptuneAnalyticsConnectionError(NeptuneAnalyticsError):
         super().__init__(message, name, status_code)
 
 
-class NeptuneAnalyticsQueryError(NeptuneAnalyticsError):
+class NeptuneAnalyticsQueryError(CogneeValidationError):
     """Exception raised when a query execution fails."""
 
     def __init__(
@@ -43,7 +48,7 @@ class NeptuneAnalyticsQueryError(NeptuneAnalyticsError):
         super().__init__(message, name, status_code)
 
 
-class NeptuneAnalyticsAuthenticationError(NeptuneAnalyticsError):
+class NeptuneAnalyticsAuthenticationError(CogneeConfigurationError):
     """Exception raised when authentication with Neptune Analytics fails."""
 
     def __init__(
@@ -55,7 +60,7 @@ class NeptuneAnalyticsAuthenticationError(NeptuneAnalyticsError):
         super().__init__(message, name, status_code)
 
 
-class NeptuneAnalyticsConfigurationError(NeptuneAnalyticsError):
+class NeptuneAnalyticsConfigurationError(CogneeConfigurationError):
     """Exception raised when Neptune Analytics configuration is invalid."""
 
     def __init__(
@@ -67,7 +72,7 @@ class NeptuneAnalyticsConfigurationError(NeptuneAnalyticsError):
         super().__init__(message, name, status_code)
 
 
-class NeptuneAnalyticsTimeoutError(NeptuneAnalyticsError):
+class NeptuneAnalyticsTimeoutError(CogneeTransientError):
     """Exception raised when a Neptune Analytics operation times out."""
 
     def __init__(
@@ -79,7 +84,7 @@ class NeptuneAnalyticsTimeoutError(NeptuneAnalyticsError):
         super().__init__(message, name, status_code)
 
 
-class NeptuneAnalyticsThrottlingError(NeptuneAnalyticsError):
+class NeptuneAnalyticsThrottlingError(CogneeTransientError):
     """Exception raised when requests are throttled by Neptune Analytics."""
 
     def __init__(
@@ -91,7 +96,7 @@ class NeptuneAnalyticsThrottlingError(NeptuneAnalyticsError):
         super().__init__(message, name, status_code)
 
 
-class NeptuneAnalyticsResourceNotFoundError(NeptuneAnalyticsError):
+class NeptuneAnalyticsResourceNotFoundError(CogneeValidationError):
     """Exception raised when a Neptune Analytics resource is not found."""
 
     def __init__(
@@ -103,7 +108,7 @@ class NeptuneAnalyticsResourceNotFoundError(NeptuneAnalyticsError):
         super().__init__(message, name, status_code)
 
 
-class NeptuneAnalyticsInvalidParameterError(NeptuneAnalyticsError):
+class NeptuneAnalyticsInvalidParameterError(CogneeValidationError):
     """Exception raised when invalid parameters are provided to Neptune Analytics."""
 
     def __init__(
diff --git a/cognee/tasks/ingestion/exceptions/__init__.py b/cognee/tasks/ingestion/exceptions/__init__.py
index 5c0a0b02f..f24792d1f 100644
--- a/cognee/tasks/ingestion/exceptions/__init__.py
+++ b/cognee/tasks/ingestion/exceptions/__init__.py
@@ -5,6 +5,4 @@ This module defines a set of exceptions for handling various application errors,
 such as System, Validation, Configuration or TransientErrors
 """
 
-from .exceptions import (
-    S3FileSystemNotFoundError
-)
+from .exceptions import S3FileSystemNotFoundError

From c99b453d965df490b02e370710bb94070f97993d Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 14 Aug 2025 10:57:16 +0200
Subject: [PATCH 29/34] feat: adds WrongDataDocumentError to classify documents

---
 cognee/tasks/documents/classify_documents.py    |  5 +++++
 cognee/tasks/documents/exceptions/__init__.py   |  9 +++++++++
 cognee/tasks/documents/exceptions/exceptions.py | 17 +++++++++++++++++
 3 files changed, 31 insertions(+)
 create mode 100644 cognee/tasks/documents/exceptions/__init__.py
 create mode 100644 cognee/tasks/documents/exceptions/exceptions.py

diff --git a/cognee/tasks/documents/classify_documents.py b/cognee/tasks/documents/classify_documents.py
index 673e17c75..6567e5825 100644
--- a/cognee/tasks/documents/classify_documents.py
+++ b/cognee/tasks/documents/classify_documents.py
@@ -10,6 +10,7 @@ from cognee.modules.data.processing.document_types import (
 )
 from cognee.modules.engine.models.node_set import NodeSet
 from cognee.modules.engine.utils.generate_node_id import generate_node_id
+from cognee.tasks.documents.exceptions import WrongDataDocumentInputError
 
 EXTENSION_TO_DOCUMENT_CLASS = {
     "pdf": PdfDocument,  # Text documents
@@ -111,8 +112,12 @@ async def classify_documents(data_documents: list[Data]) -> list[Document]:
         - list[Document]: A list of Document objects created based on the classified data
           documents.
     """
+    if not isinstance(data_documents, list):
+        raise WrongDataDocumentInputError("data_documents")
+
     documents = []
     for data_item in data_documents:
+
         document = EXTENSION_TO_DOCUMENT_CLASS[data_item.extension](
             id=data_item.id,
             title=f"{data_item.name}.{data_item.extension}",
diff --git a/cognee/tasks/documents/exceptions/__init__.py b/cognee/tasks/documents/exceptions/__init__.py
new file mode 100644
index 000000000..cdd50c6b0
--- /dev/null
+++ b/cognee/tasks/documents/exceptions/__init__.py
@@ -0,0 +1,9 @@
+"""
+Custom exceptions for the Cognee API.
+
+This module defines a set of exceptions for handling various data errors
+"""
+
+from .exceptions import (
+    WrongDataDocumentInputError,
+)
diff --git a/cognee/tasks/documents/exceptions/exceptions.py b/cognee/tasks/documents/exceptions/exceptions.py
new file mode 100644
index 000000000..a1fcb1d4d
--- /dev/null
+++ b/cognee/tasks/documents/exceptions/exceptions.py
@@ -0,0 +1,17 @@
+from cognee.exceptions import (
+    CogneeValidationError,
+    CogneeConfigurationError,
+)
+from fastapi import status
+
+
+class WrongDataDocumentInputError(CogneeValidationError):
+    """Raised when a wrong data document is provided."""
+    def __init__(
+        self,
+        field: str,
+        name: str = "WrongDataDocumentInputError",
+        status_code: int = status.HTTP_422_UNPROCESSABLE_ENTITY,
+    ):
+        message = f"Missing of invalid parameter: '{field}'."
+        super().__init__(message, name, status_code)
\ No newline at end of file

From df3a3df117026aea7bab81521e9c06b96f152bb7 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 14 Aug 2025 13:12:08 +0200
Subject: [PATCH 30/34] feat: adds errors to classify, and chunking top level

---
 cognee/tasks/documents/classify_documents.py  |  1 -
 cognee/tasks/documents/exceptions/__init__.py |  2 ++
 .../tasks/documents/exceptions/exceptions.py  | 21 ++++++++++++++++++-
 .../extract_chunks_from_documents.py          | 10 +++++++--
 4 files changed, 30 insertions(+), 4 deletions(-)

diff --git a/cognee/tasks/documents/classify_documents.py b/cognee/tasks/documents/classify_documents.py
index 6567e5825..9fa512906 100644
--- a/cognee/tasks/documents/classify_documents.py
+++ b/cognee/tasks/documents/classify_documents.py
@@ -117,7 +117,6 @@ async def classify_documents(data_documents: list[Data]) -> list[Document]:
 
     documents = []
     for data_item in data_documents:
-
         document = EXTENSION_TO_DOCUMENT_CLASS[data_item.extension](
             id=data_item.id,
             title=f"{data_item.name}.{data_item.extension}",
diff --git a/cognee/tasks/documents/exceptions/__init__.py b/cognee/tasks/documents/exceptions/__init__.py
index cdd50c6b0..a8602d6f5 100644
--- a/cognee/tasks/documents/exceptions/__init__.py
+++ b/cognee/tasks/documents/exceptions/__init__.py
@@ -6,4 +6,6 @@ This module defines a set of exceptions for handling various data errors
 
 from .exceptions import (
     WrongDataDocumentInputError,
+    InvalidChunkSizeError,
+    InvalidChunkerError,
 )
diff --git a/cognee/tasks/documents/exceptions/exceptions.py b/cognee/tasks/documents/exceptions/exceptions.py
index a1fcb1d4d..27907aaf1 100644
--- a/cognee/tasks/documents/exceptions/exceptions.py
+++ b/cognee/tasks/documents/exceptions/exceptions.py
@@ -7,6 +7,7 @@ from fastapi import status
 
 class WrongDataDocumentInputError(CogneeValidationError):
     """Raised when a wrong data document is provided."""
+
     def __init__(
         self,
         field: str,
@@ -14,4 +15,22 @@ class WrongDataDocumentInputError(CogneeValidationError):
         status_code: int = status.HTTP_422_UNPROCESSABLE_ENTITY,
     ):
         message = f"Missing of invalid parameter: '{field}'."
-        super().__init__(message, name, status_code)
\ No newline at end of file
+        super().__init__(message, name, status_code)
+
+
+class InvalidChunkSizeError(CogneeValidationError):
+    def __init__(self, value):
+        super().__init__(
+            message=f"max_chunk_size must be a positive integer (got {value}).",
+            name="InvalidChunkSizeError",
+            status_code=status.HTTP_400_BAD_REQUEST,
+        )
+
+
+class InvalidChunkerError(CogneeValidationError):
+    def __init__(self):
+        super().__init__(
+            message=f"chunker must be a valid Chunker class.",
+            name="InvalidChunkerError",
+            status_code=status.HTTP_400_BAD_REQUEST,
+        )
diff --git a/cognee/tasks/documents/extract_chunks_from_documents.py b/cognee/tasks/documents/extract_chunks_from_documents.py
index 216185495..d52380c8d 100644
--- a/cognee/tasks/documents/extract_chunks_from_documents.py
+++ b/cognee/tasks/documents/extract_chunks_from_documents.py
@@ -8,6 +8,7 @@ from cognee.modules.data.models import Data
 from cognee.infrastructure.databases.relational import get_relational_engine
 from cognee.modules.chunking.TextChunker import TextChunker
 from cognee.modules.chunking.Chunker import Chunker
+from cognee.tasks.documents.exceptions import InvalidChunkSizeError, InvalidChunkerError
 
 
 async def update_document_token_count(document_id: UUID, token_count: int) -> None:
@@ -37,6 +38,13 @@ async def extract_chunks_from_documents(
         - The `read` method of the `Document` class must be implemented to support the chunking operation.
         - The `chunker` parameter determines the chunking logic and should align with the document type.
     """
+    if not isinstance(max_chunk_size, int) or max_chunk_size <= 0:
+        raise InvalidChunkSizeError(max_chunk_size)
+    if not isinstance(chunker, type):
+        raise InvalidChunkerError()
+    if not hasattr(chunker, "read"):
+        raise InvalidChunkerError()
+
     for document in documents:
         document_token_count = 0
 
@@ -48,5 +56,3 @@ async def extract_chunks_from_documents(
             yield document_chunk
 
         await update_document_token_count(document.id, document_token_count)
-
-        # todo rita

From 9f965c44b4cdebda8129a7fdd487d96d27a1a5ea Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 14 Aug 2025 13:53:39 +0200
Subject: [PATCH 31/34] feat: adds input checks for extract graph from data

---
 cognee/tasks/graph/exceptions/__init__.py     | 12 ++++++
 cognee/tasks/graph/exceptions/exceptions.py   | 37 +++++++++++++++++++
 cognee/tasks/graph/extract_graph_from_data.py | 25 ++++++++++++-
 3 files changed, 73 insertions(+), 1 deletion(-)
 create mode 100644 cognee/tasks/graph/exceptions/__init__.py
 create mode 100644 cognee/tasks/graph/exceptions/exceptions.py

diff --git a/cognee/tasks/graph/exceptions/__init__.py b/cognee/tasks/graph/exceptions/__init__.py
new file mode 100644
index 000000000..3516ba151
--- /dev/null
+++ b/cognee/tasks/graph/exceptions/__init__.py
@@ -0,0 +1,12 @@
+"""
+Custom exceptions for the Cognee API.
+
+This module defines a set of exceptions for handling various data errors
+"""
+
+from .exceptions import (
+    InvalidDataChunksError,
+    InvalidGraphModelError,
+    InvalidOntologyAdapterError,
+    InvalidChunkGraphInputError
+)
diff --git a/cognee/tasks/graph/exceptions/exceptions.py b/cognee/tasks/graph/exceptions/exceptions.py
new file mode 100644
index 000000000..0708bab31
--- /dev/null
+++ b/cognee/tasks/graph/exceptions/exceptions.py
@@ -0,0 +1,37 @@
+from cognee.exceptions import (
+    CogneeValidationError,
+    CogneeConfigurationError,
+)
+from fastapi import status
+
+
+class InvalidDataChunksError(CogneeValidationError):
+    def __init__(self, detail: str):
+        super().__init__(
+            message=f"Invalid data_chunks: {detail}",
+            name="InvalidDataChunksError",
+            status_code=status.HTTP_400_BAD_REQUEST,
+        )
+
+class InvalidGraphModelError(CogneeValidationError):
+    def __init__(self, got):
+        super().__init__(
+            message=f"graph_model must be a subclass of BaseModel (got {got}).",
+            name="InvalidGraphModelError",
+            status_code=status.HTTP_400_BAD_REQUEST,
+        )
+
+class InvalidOntologyAdapterError(CogneeConfigurationError):
+    def __init__(self, got):
+        super().__init__(
+            message=f"ontology_adapter lacks required interface (got {got}).",
+            name="InvalidOntologyAdapterError",
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+
+class InvalidChunkGraphInputError(CogneeValidationError):
+    def __init__(self, detail: str):
+        super().__init__(
+            message=f"Invalid chunk inputs or LLM Chunkgraphs: {detail}",
+            name="InvalidChunkGraphInputError",
+            status_code=status.HTTP_400_BAD_REQUEST)
\ No newline at end of file
diff --git a/cognee/tasks/graph/extract_graph_from_data.py b/cognee/tasks/graph/extract_graph_from_data.py
index aa415d504..2ddd85075 100644
--- a/cognee/tasks/graph/extract_graph_from_data.py
+++ b/cognee/tasks/graph/extract_graph_from_data.py
@@ -12,7 +12,12 @@ from cognee.modules.graph.utils import (
 )
 from cognee.shared.data_models import KnowledgeGraph
 from cognee.infrastructure.llm.LLMGateway import LLMGateway
-
+from cognee.tasks.graph.exceptions import (
+    InvalidGraphModelError,
+    InvalidDataChunksError,
+    InvalidChunkGraphInputError,
+    InvalidOntologyAdapterError,
+)
 
 async def integrate_chunk_graphs(
     data_chunks: list[DocumentChunk],
@@ -21,6 +26,16 @@ async def integrate_chunk_graphs(
     ontology_adapter: OntologyResolver,
 ) -> List[DocumentChunk]:
     """Updates DocumentChunk objects, integrates data points and edges into databases."""
+
+    if not isinstance(data_chunks, list) or not isinstance(chunk_graphs, list):
+        raise InvalidChunkGraphInputError("data_chunks and chunk_graphs must be lists.")
+    if len(data_chunks) != len(chunk_graphs):
+        raise InvalidChunkGraphInputError(f"length mismatch: {len(data_chunks)} chunks vs {len(chunk_graphs)} graphs.")
+    if not isinstance(graph_model, type) or not issubclass(graph_model, BaseModel):
+        raise InvalidGraphModelError(graph_model)
+    if ontology_adapter is None or not hasattr(ontology_adapter, "get_subgraph"):
+        raise InvalidOntologyAdapterError(type(ontology_adapter).__name__ if ontology_adapter else "None")
+
     graph_engine = await get_graph_engine()
 
     if graph_model is not KnowledgeGraph:
@@ -55,6 +70,14 @@ async def extract_graph_from_data(
     """
     Extracts and integrates a knowledge graph from the text content of document chunks using a specified graph model.
     """
+
+    if not isinstance(data_chunks, list) or not data_chunks:
+        raise InvalidDataChunksError("must be a non-empty list of DocumentChunk.")
+    if not all(hasattr(c, "text") for c in data_chunks):
+        raise InvalidDataChunksError("each chunk must have a 'text' attribute")
+    if not isinstance(graph_model, type) or not issubclass(graph_model, BaseModel):
+        raise InvalidGraphModelError(graph_model)
+
     chunk_graphs = await asyncio.gather(
         *[LLMGateway.extract_content_graph(chunk.text, graph_model) for chunk in data_chunks]
     )

From 63d071f0d86d54246e85eb112f24aea7263db86c Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 14 Aug 2025 14:17:13 +0200
Subject: [PATCH 32/34] feat: adds input checks for add datapoints and
 summarization tasks

---
 cognee/tasks/storage/add_data_points.py             |  9 +++++++++
 cognee/tasks/storage/exceptions/__init__.py         |  9 +++++++++
 cognee/tasks/storage/exceptions/exceptions.py       | 13 +++++++++++++
 cognee/tasks/summarization/exceptions/__init__.py   |  9 +++++++++
 cognee/tasks/summarization/exceptions/exceptions.py | 13 +++++++++++++
 cognee/tasks/summarization/summarize_text.py        | 10 +++++++++-
 6 files changed, 62 insertions(+), 1 deletion(-)
 create mode 100644 cognee/tasks/storage/exceptions/__init__.py
 create mode 100644 cognee/tasks/storage/exceptions/exceptions.py
 create mode 100644 cognee/tasks/summarization/exceptions/__init__.py
 create mode 100644 cognee/tasks/summarization/exceptions/exceptions.py

diff --git a/cognee/tasks/storage/add_data_points.py b/cognee/tasks/storage/add_data_points.py
index 9b5c36c37..41bda954f 100644
--- a/cognee/tasks/storage/add_data_points.py
+++ b/cognee/tasks/storage/add_data_points.py
@@ -5,9 +5,18 @@ from cognee.infrastructure.databases.graph import get_graph_engine
 from cognee.modules.graph.utils import deduplicate_nodes_and_edges, get_graph_from_model
 from .index_data_points import index_data_points
 from .index_graph_edges import index_graph_edges
+from cognee.tasks.storage.exceptions import (
+    InvalidDataPointsInAddDataPointsError,
+)
 
 
 async def add_data_points(data_points: List[DataPoint]) -> List[DataPoint]:
+
+    if not isinstance(data_points, list):
+        raise InvalidDataPointsInAddDataPointsError("data_points must be a list.")
+    if not all(isinstance(dp, DataPoint) for dp in data_points):
+        raise InvalidDataPointsInAddDataPointsError("data_points: each item must be a DataPoint.")
+
     nodes = []
     edges = []
 
diff --git a/cognee/tasks/storage/exceptions/__init__.py b/cognee/tasks/storage/exceptions/__init__.py
new file mode 100644
index 000000000..33dd215d6
--- /dev/null
+++ b/cognee/tasks/storage/exceptions/__init__.py
@@ -0,0 +1,9 @@
+"""
+Custom exceptions for the Cognee API.
+
+This module defines a set of exceptions for handling various data errors
+"""
+
+from .exceptions import (
+    InvalidDataPointsInAddDataPointsError,
+)
diff --git a/cognee/tasks/storage/exceptions/exceptions.py b/cognee/tasks/storage/exceptions/exceptions.py
new file mode 100644
index 000000000..2da6a84aa
--- /dev/null
+++ b/cognee/tasks/storage/exceptions/exceptions.py
@@ -0,0 +1,13 @@
+from cognee.exceptions import (
+    CogneeValidationError,
+)
+from fastapi import status
+
+
+class InvalidDataPointsInAddDataPointsError(CogneeValidationError):
+    def __init__(self, detail: str):
+        super().__init__(
+            message=f"Invalid data_points: {detail}",
+            name="InvalidDataPointsInAddDataPointsError",
+            status_code=status.HTTP_400_BAD_REQUEST)
+
diff --git a/cognee/tasks/summarization/exceptions/__init__.py b/cognee/tasks/summarization/exceptions/__init__.py
new file mode 100644
index 000000000..502f973e7
--- /dev/null
+++ b/cognee/tasks/summarization/exceptions/__init__.py
@@ -0,0 +1,9 @@
+"""
+Custom exceptions for the Cognee API.
+
+This module defines a set of exceptions for handling various data errors
+"""
+
+from .exceptions import (
+    InvalidSummaryInputsError,
+)
diff --git a/cognee/tasks/summarization/exceptions/exceptions.py b/cognee/tasks/summarization/exceptions/exceptions.py
new file mode 100644
index 000000000..d0c9cfec6
--- /dev/null
+++ b/cognee/tasks/summarization/exceptions/exceptions.py
@@ -0,0 +1,13 @@
+from cognee.exceptions import (
+    CogneeValidationError,
+    CogneeConfigurationError,
+)
+from fastapi import status
+
+
+class InvalidSummaryInputsError(CogneeValidationError):
+    def __init__(self, detail: str):
+        super().__init__(
+            message=f"Invalid summarize_text inputs: {detail}",
+            name="InvalidSummaryInputsError",
+            status_code=status.HTTP_400_BAD_REQUEST)
\ No newline at end of file
diff --git a/cognee/tasks/summarization/summarize_text.py b/cognee/tasks/summarization/summarize_text.py
index ca6964f83..b482b5b4a 100644
--- a/cognee/tasks/summarization/summarize_text.py
+++ b/cognee/tasks/summarization/summarize_text.py
@@ -3,10 +3,11 @@ from typing import Type
 from uuid import uuid5
 from pydantic import BaseModel
 
+from cognee.tasks.summarization.exceptions import InvalidSummaryInputsError
 from cognee.modules.chunking.models.DocumentChunk import DocumentChunk
 from cognee.infrastructure.llm.LLMGateway import LLMGateway
 from cognee.modules.cognify.config import get_cognify_config
-from .models import TextSummary
+from cognee.tasks.summarization.models import TextSummary
 
 
 async def summarize_text(
@@ -35,6 +36,13 @@ async def summarize_text(
         A list of TextSummary objects, each containing the summary of a corresponding
         DocumentChunk.
     """
+
+    if not isinstance(data_chunks, list):
+        raise InvalidSummaryInputsError("data_chunks must be a list.")
+    if not all(hasattr(c, "text") for c in data_chunks):
+        raise InvalidSummaryInputsError("each DocumentChunk must have a 'text' attribute.")
+
+
     if len(data_chunks) == 0:
         return data_chunks
 

From affbc557d266c45dc5794c09e3c4e5bbab66097d Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 14 Aug 2025 14:17:35 +0200
Subject: [PATCH 33/34] chore: ruff formatting

---
 cognee/tasks/graph/exceptions/__init__.py           | 2 +-
 cognee/tasks/graph/exceptions/exceptions.py         | 8 ++++++--
 cognee/tasks/graph/extract_graph_from_data.py       | 9 +++++++--
 cognee/tasks/storage/add_data_points.py             | 1 -
 cognee/tasks/storage/exceptions/exceptions.py       | 4 ++--
 cognee/tasks/summarization/exceptions/exceptions.py | 3 ++-
 cognee/tasks/summarization/summarize_text.py        | 1 -
 7 files changed, 18 insertions(+), 10 deletions(-)

diff --git a/cognee/tasks/graph/exceptions/__init__.py b/cognee/tasks/graph/exceptions/__init__.py
index 3516ba151..d91bbbd36 100644
--- a/cognee/tasks/graph/exceptions/__init__.py
+++ b/cognee/tasks/graph/exceptions/__init__.py
@@ -8,5 +8,5 @@ from .exceptions import (
     InvalidDataChunksError,
     InvalidGraphModelError,
     InvalidOntologyAdapterError,
-    InvalidChunkGraphInputError
+    InvalidChunkGraphInputError,
 )
diff --git a/cognee/tasks/graph/exceptions/exceptions.py b/cognee/tasks/graph/exceptions/exceptions.py
index 0708bab31..c09ee1c08 100644
--- a/cognee/tasks/graph/exceptions/exceptions.py
+++ b/cognee/tasks/graph/exceptions/exceptions.py
@@ -13,6 +13,7 @@ class InvalidDataChunksError(CogneeValidationError):
             status_code=status.HTTP_400_BAD_REQUEST,
         )
 
+
 class InvalidGraphModelError(CogneeValidationError):
     def __init__(self, got):
         super().__init__(
@@ -21,12 +22,14 @@ class InvalidGraphModelError(CogneeValidationError):
             status_code=status.HTTP_400_BAD_REQUEST,
         )
 
+
 class InvalidOntologyAdapterError(CogneeConfigurationError):
     def __init__(self, got):
         super().__init__(
             message=f"ontology_adapter lacks required interface (got {got}).",
             name="InvalidOntologyAdapterError",
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+        )
 
 
 class InvalidChunkGraphInputError(CogneeValidationError):
@@ -34,4 +37,5 @@ class InvalidChunkGraphInputError(CogneeValidationError):
         super().__init__(
             message=f"Invalid chunk inputs or LLM Chunkgraphs: {detail}",
             name="InvalidChunkGraphInputError",
-            status_code=status.HTTP_400_BAD_REQUEST)
\ No newline at end of file
+            status_code=status.HTTP_400_BAD_REQUEST,
+        )
diff --git a/cognee/tasks/graph/extract_graph_from_data.py b/cognee/tasks/graph/extract_graph_from_data.py
index 2ddd85075..019e9e4a1 100644
--- a/cognee/tasks/graph/extract_graph_from_data.py
+++ b/cognee/tasks/graph/extract_graph_from_data.py
@@ -19,6 +19,7 @@ from cognee.tasks.graph.exceptions import (
     InvalidOntologyAdapterError,
 )
 
+
 async def integrate_chunk_graphs(
     data_chunks: list[DocumentChunk],
     chunk_graphs: list,
@@ -30,11 +31,15 @@ async def integrate_chunk_graphs(
     if not isinstance(data_chunks, list) or not isinstance(chunk_graphs, list):
         raise InvalidChunkGraphInputError("data_chunks and chunk_graphs must be lists.")
     if len(data_chunks) != len(chunk_graphs):
-        raise InvalidChunkGraphInputError(f"length mismatch: {len(data_chunks)} chunks vs {len(chunk_graphs)} graphs.")
+        raise InvalidChunkGraphInputError(
+            f"length mismatch: {len(data_chunks)} chunks vs {len(chunk_graphs)} graphs."
+        )
     if not isinstance(graph_model, type) or not issubclass(graph_model, BaseModel):
         raise InvalidGraphModelError(graph_model)
     if ontology_adapter is None or not hasattr(ontology_adapter, "get_subgraph"):
-        raise InvalidOntologyAdapterError(type(ontology_adapter).__name__ if ontology_adapter else "None")
+        raise InvalidOntologyAdapterError(
+            type(ontology_adapter).__name__ if ontology_adapter else "None"
+        )
 
     graph_engine = await get_graph_engine()
 
diff --git a/cognee/tasks/storage/add_data_points.py b/cognee/tasks/storage/add_data_points.py
index 41bda954f..6cdc90ac9 100644
--- a/cognee/tasks/storage/add_data_points.py
+++ b/cognee/tasks/storage/add_data_points.py
@@ -11,7 +11,6 @@ from cognee.tasks.storage.exceptions import (
 
 
 async def add_data_points(data_points: List[DataPoint]) -> List[DataPoint]:
-
     if not isinstance(data_points, list):
         raise InvalidDataPointsInAddDataPointsError("data_points must be a list.")
     if not all(isinstance(dp, DataPoint) for dp in data_points):
diff --git a/cognee/tasks/storage/exceptions/exceptions.py b/cognee/tasks/storage/exceptions/exceptions.py
index 2da6a84aa..9b2de9efd 100644
--- a/cognee/tasks/storage/exceptions/exceptions.py
+++ b/cognee/tasks/storage/exceptions/exceptions.py
@@ -9,5 +9,5 @@ class InvalidDataPointsInAddDataPointsError(CogneeValidationError):
         super().__init__(
             message=f"Invalid data_points: {detail}",
             name="InvalidDataPointsInAddDataPointsError",
-            status_code=status.HTTP_400_BAD_REQUEST)
-
+            status_code=status.HTTP_400_BAD_REQUEST,
+        )
diff --git a/cognee/tasks/summarization/exceptions/exceptions.py b/cognee/tasks/summarization/exceptions/exceptions.py
index d0c9cfec6..9e8e7197e 100644
--- a/cognee/tasks/summarization/exceptions/exceptions.py
+++ b/cognee/tasks/summarization/exceptions/exceptions.py
@@ -10,4 +10,5 @@ class InvalidSummaryInputsError(CogneeValidationError):
         super().__init__(
             message=f"Invalid summarize_text inputs: {detail}",
             name="InvalidSummaryInputsError",
-            status_code=status.HTTP_400_BAD_REQUEST)
\ No newline at end of file
+            status_code=status.HTTP_400_BAD_REQUEST,
+        )
diff --git a/cognee/tasks/summarization/summarize_text.py b/cognee/tasks/summarization/summarize_text.py
index b482b5b4a..f6dcc54a2 100644
--- a/cognee/tasks/summarization/summarize_text.py
+++ b/cognee/tasks/summarization/summarize_text.py
@@ -42,7 +42,6 @@ async def summarize_text(
     if not all(hasattr(c, "text") for c in data_chunks):
         raise InvalidSummaryInputsError("each DocumentChunk must have a 'text' attribute.")
 
-
     if len(data_chunks) == 0:
         return data_chunks
 

From a7d7e12d4cb808cb52b8f688718a25e351a65109 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 14 Aug 2025 14:48:35 +0200
Subject: [PATCH 34/34] ruff fix

---
 cognee/tasks/documents/exceptions/exceptions.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/tasks/documents/exceptions/exceptions.py b/cognee/tasks/documents/exceptions/exceptions.py
index 27907aaf1..737e9a3d1 100644
--- a/cognee/tasks/documents/exceptions/exceptions.py
+++ b/cognee/tasks/documents/exceptions/exceptions.py
@@ -30,7 +30,7 @@ class InvalidChunkSizeError(CogneeValidationError):
 class InvalidChunkerError(CogneeValidationError):
     def __init__(self):
         super().__init__(
-            message=f"chunker must be a valid Chunker class.",
+            message="chunker must be a valid Chunker class.",
             name="InvalidChunkerError",
             status_code=status.HTTP_400_BAD_REQUEST,
         )