# SPDX-FileCopyrightText: 2025 Weibo, Inc.
#
# SPDX-License-Identifier: Apache-2.0

"""
Shared SQLAlchemy database models for Wegent project.

These models are used by both Backend and chat_shell services.
"""

from .base import Base
from .enums import (
    ContextStatus,
    ContextType,
    InjectionMode,
    QueueMessagePriority,
    QueueMessageStatus,
    QueueVisibility,
    SenderType,
    SubtaskRole,
    SubtaskStatus,
    TriggerMode,
)
from .kind import Kind
from .skill_binary import SkillBinary
from .subtask import Subtask
from .subtask_context import SubtaskContext
from .user import User
from .work_queue import (
    QueueMessage,
    RecentContact,
)

__all__ = [
    # Base
    "Base",
    # Enums
    "SubtaskStatus",
    "SubtaskRole",
    "SenderType",
    "ContextType",
    "ContextStatus",
    "InjectionMode",
    # Work Queue Enums
    "QueueVisibility",
    "QueueMessageStatus",
    "QueueMessagePriority",
    "TriggerMode",
    # Models
    "Subtask",
    "SubtaskContext",
    "User",
    "Kind",
    "SkillBinary",
    # Work Queue Models
    "QueueMessage",
    "RecentContact",
]
