feat: adds configattributeerror

This commit is contained in:
hajdul88 2025-08-13 12:22:03 +02:00
parent ebd4403c2f
commit 657c775cbe
3 changed files with 26 additions and 4 deletions

View file

@ -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):

View file

@ -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
)

View file

@ -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)