backend/db_models/__init__.py

"""
Models package for AegisLM SaaS Backend.

Database models for users, evaluations, and API keys.
"""

from .user import User
from .evaluation import Evaluation, EvaluationStatus
from .api_key import ApiKey
from .role import Role, Permission

# Import Base from database module
from core.database import Base

# Import AI models if available
try:
    from ai.models.attack_models import Attack
    __all__ = [
        "User",
        "Evaluation", 
        "EvaluationStatus",
        "ApiKey",
        "Role",
        "Permission",
        "Attack"
    ]
except ImportError:
    __all__ = [
        "User",
        "Evaluation", 
        "EvaluationStatus",
        "ApiKey",
        "Role",
        "Permission"
    ]