SurfSense/surfsense_backend/app/schemas/__init__.py

from .auth import (
    LogoutAllResponse,
    LogoutRequest,
    LogoutResponse,
    RefreshTokenRequest,
    RefreshTokenResponse,
)
from .base import IDModel, TimestampModel
from .chunks import ChunkBase, ChunkCreate, ChunkRead, ChunkUpdate
from .documents import (
    DocumentBase,
    DocumentRead,
    DocumentsCreate,
    DocumentStatusBatchResponse,
    DocumentStatusItemRead,
    DocumentStatusSchema,
    DocumentTitleRead,
    DocumentTitleSearchResponse,
    DocumentUpdate,
    DocumentWithChunksRead,
    ExtensionDocumentContent,
    ExtensionDocumentMetadata,
    PaginatedResponse,
)
from .folders import (
    BulkDocumentMove,
    DocumentMove,
    FolderBreadcrumb,
    FolderCreate,
    FolderMove,
    FolderRead,
    FolderReorder,
    FolderUpdate,
)
from .google_drive import DriveItem, GoogleDriveIndexingOptions, GoogleDriveIndexRequest
from .image_generation import (
    ImageGenerationCreate,
    ImageGenerationListRead,
    ImageGenerationRead,
)
from .logs import LogBase, LogCreate, LogFilter, LogRead, LogUpdate
from .model_connections import (
    ConnectionCreate,
    ConnectionRead,
    ConnectionUpdate,
    ModelCreate,
    ModelPreviewRead,
    ModelProviderRead,
    ModelRead,
    ModelRolesRead,
    ModelRolesUpdate,
    ModelsBulkUpdate,
    ModelSelection,
    ModelTestPreview,
    ModelUpdate,
    VerifyConnectionResponse,
)
from .new_chat import (
    ChatMessage,
    NewChatMessageAppend,
    NewChatMessageCreate,
    NewChatMessageRead,
    NewChatRequest,
    NewChatThreadCreate,
    NewChatThreadRead,
    NewChatThreadUpdate,
    NewChatThreadWithMessages,
    ThreadHistoryLoadResponse,
    ThreadListItem,
    ThreadListResponse,
)
from .rbac_schemas import (
    InviteAcceptRequest,
    InviteAcceptResponse,
    InviteCreate,
    InviteInfoResponse,
    InviteRead,
    InviteUpdate,
    MembershipRead,
    MembershipReadWithUser,
    MembershipUpdate,
    PermissionInfo,
    PermissionsListResponse,
    RoleCreate,
    RoleRead,
    RoleUpdate,
    UserSearchSpaceAccess,
)
from .reports import (
    ReportBase,
    ReportContentRead,
    ReportContentUpdate,
    ReportRead,
    ReportVersionInfo,
)
from .search_source_connector import (
    MCPConnectorCreate,
    MCPConnectorRead,
    MCPConnectorUpdate,
    MCPServerConfig,
    SearchSourceConnectorBase,
    SearchSourceConnectorCreate,
    SearchSourceConnectorRead,
    SearchSourceConnectorUpdate,
)
from .search_space import (
    SearchSpaceBase,
    SearchSpaceCreate,
    SearchSpaceRead,
    SearchSpaceUpdate,
    SearchSpaceWithStats,
)
from .stripe import (
    CreateCreditCheckoutSessionRequest,
    CreateCreditCheckoutSessionResponse,
    CreditPurchaseHistoryResponse,
    CreditPurchaseRead,
    CreditStripeStatusResponse,
    PagePurchaseHistoryResponse,
    PagePurchaseRead,
    StripeWebhookResponse,
)
from .users import UserCreate, UserRead, UserUpdate
from .video_presentations import (
    VideoPresentationBase,
    VideoPresentationCreate,
    VideoPresentationRead,
    VideoPresentationUpdate,
)

__all__ = [
    # Folder schemas
    "BulkDocumentMove",
    # Chat schemas (assistant-ui integration)
    "ChatMessage",
    # Chunk schemas
    "ChunkBase",
    "ChunkCreate",
    "ChunkRead",
    "ChunkUpdate",
    # Model connection schemas
    "ConnectionCreate",
    "ConnectionRead",
    "ConnectionUpdate",
    "CreateCreditCheckoutSessionRequest",
    "CreateCreditCheckoutSessionResponse",
    "CreditPurchaseHistoryResponse",
    "CreditPurchaseRead",
    "CreditStripeStatusResponse",
    # Document schemas
    "DocumentBase",
    "DocumentMove",
    "DocumentRead",
    "DocumentStatusBatchResponse",
    "DocumentStatusItemRead",
    "DocumentStatusSchema",
    "DocumentTitleRead",
    "DocumentTitleSearchResponse",
    "DocumentUpdate",
    "DocumentWithChunksRead",
    "DocumentsCreate",
    # Google Drive schemas
    "DriveItem",
    "ExtensionDocumentContent",
    "ExtensionDocumentMetadata",
    "FolderBreadcrumb",
    "FolderCreate",
    "FolderMove",
    "FolderRead",
    "FolderReorder",
    "FolderUpdate",
    "GoogleDriveIndexRequest",
    "GoogleDriveIndexingOptions",
    # Base schemas
    "IDModel",
    # Image Generation schemas
    "ImageGenerationCreate",
    "ImageGenerationListRead",
    "ImageGenerationRead",
    # RBAC schemas
    "InviteAcceptRequest",
    "InviteAcceptResponse",
    "InviteCreate",
    "InviteInfoResponse",
    "InviteRead",
    "InviteUpdate",
    # Log schemas
    "LogBase",
    "LogCreate",
    "LogFilter",
    "LogRead",
    "LogUpdate",
    # Auth schemas
    "LogoutAllResponse",
    "LogoutRequest",
    "LogoutResponse",
    # Search source connector schemas
    "MCPConnectorCreate",
    "MCPConnectorRead",
    "MCPConnectorUpdate",
    "MCPServerConfig",
    "MembershipRead",
    "MembershipReadWithUser",
    "MembershipUpdate",
    "ModelCreate",
    "ModelPreviewRead",
    "ModelProviderRead",
    "ModelRead",
    "ModelRolesRead",
    "ModelRolesUpdate",
    "ModelSelection",
    "ModelTestPreview",
    "ModelUpdate",
    "ModelsBulkUpdate",
    "NewChatMessageAppend",
    "NewChatMessageCreate",
    "NewChatMessageRead",
    "NewChatRequest",
    "NewChatThreadCreate",
    "NewChatThreadRead",
    "NewChatThreadUpdate",
    "NewChatThreadWithMessages",
    "PagePurchaseHistoryResponse",
    "PagePurchaseRead",
    "PaginatedResponse",
    "PermissionInfo",
    "PermissionsListResponse",
    # Podcast schemas
    "RefreshTokenRequest",
    "RefreshTokenResponse",
    # Report schemas
    "ReportBase",
    "ReportContentRead",
    "ReportContentUpdate",
    "ReportRead",
    "ReportVersionInfo",
    "RoleCreate",
    "RoleRead",
    "RoleUpdate",
    "SearchSourceConnectorBase",
    "SearchSourceConnectorCreate",
    "SearchSourceConnectorRead",
    "SearchSourceConnectorUpdate",
    # Search space schemas
    "SearchSpaceBase",
    "SearchSpaceCreate",
    "SearchSpaceRead",
    "SearchSpaceUpdate",
    "SearchSpaceWithStats",
    "StripeWebhookResponse",
    "ThreadHistoryLoadResponse",
    "ThreadListItem",
    "ThreadListResponse",
    "TimestampModel",
    # User schemas
    "UserCreate",
    "UserRead",
    "UserSearchSpaceAccess",
    "UserUpdate",
    "VerifyConnectionResponse",
    # Video Presentation schemas
    "VideoPresentationBase",
    "VideoPresentationCreate",
    "VideoPresentationRead",
    "VideoPresentationUpdate",
]