"""
Coze AI 服务模块
"""

from .client import get_coze_client, get_auth_manager, get_bot_config, get_workspace_id
from .service import get_coze_service, CozeService
from .models import (
    SessionType,
    MessageRole,
    ContentType,
    StreamEventType,
    CozeSession,
    CozeMessage,
    StreamEvent,
    CreateSessionRequest,
    CreateSessionResponse,
    SendMessageRequest,
    EndSessionRequest,
    EndSessionResponse,
)
from .exceptions import (
    CozeException,
    CozeAuthError,
    CozeAPIError,
    CozeRateLimitError,
    CozeTimeoutError,
    CozeStreamError,
    map_coze_error_to_exception,
)

__all__ = [
    # Client
    "get_coze_client",
    "get_auth_manager",
    "get_bot_config",
    "get_workspace_id",
    # Service
    "get_coze_service",
    "CozeService",
    # Models
    "SessionType",
    "MessageRole",
    "ContentType",
    "StreamEventType",
    "CozeSession",
    "CozeMessage",
    "StreamEvent",
    "CreateSessionRequest",
    "CreateSessionResponse",
    "SendMessageRequest",
    "EndSessionRequest",
    "EndSessionResponse",
    # Exceptions
    "CozeException",
    "CozeAuthError",
    "CozeAPIError",
    "CozeRateLimitError",
    "CozeTimeoutError",
    "CozeStreamError",
    "map_coze_error_to_exception",
]
