label_ai_service/app/core/exceptions.py

from fastapi import Request
from fastapi.responses import JSONResponse


class AIServiceError(Exception):
    status_code: int = 500
    code: str = "INTERNAL_ERROR"

    def __init__(self, message: str) -> None:
        self.message = message
        super().__init__(message)


class UnsupportedFileTypeError(AIServiceError):
    status_code = 400
    code = "UNSUPPORTED_FILE_TYPE"


class VideoTooLargeError(AIServiceError):
    status_code = 400
    code = "VIDEO_TOO_LARGE"


class StorageError(AIServiceError):
    status_code = 502
    code = "STORAGE_ERROR"


class LLMParseError(AIServiceError):
    status_code = 502
    code = "LLM_PARSE_ERROR"


class LLMCallError(AIServiceError):
    status_code = 503
    code = "LLM_CALL_ERROR"


async def ai_service_exception_handler(request: Request, exc: AIServiceError) -> JSONResponse:
    return JSONResponse(
        status_code=exc.status_code,
        content={"code": exc.code, "message": exc.message},
    )


async def unhandled_exception_handler(request: Request, exc: Exception) -> JSONResponse:
    return JSONResponse(
        status_code=500,
        content={"code": "INTERNAL_ERROR", "message": str(exc)},
    )