36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
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)
|
|
|
|
|
|
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="chunker must be a valid Chunker class.",
|
|
name="InvalidChunkerError",
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
)
|