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 .google_drive import DriveItem, GoogleDriveIndexingOptions, GoogleDriveIndexRequest
from .image_generation import (
    GlobalImageGenConfigRead,
    ImageGenerationConfigCreate,
    ImageGenerationConfigPublic,
    ImageGenerationConfigRead,
    ImageGenerationConfigUpdate,
    ImageGenerationCreate,
    ImageGenerationListRead,
    ImageGenerationRead,
)
from .logs import LogBase, LogCreate, LogFilter, LogRead, LogUpdate
from .new_chat import (
    ChatMessage,
    NewChatMessageAppend,
    NewChatMessageCreate,
    NewChatMessageRead,
    NewChatRequest,
    NewChatThreadCreate,
    NewChatThreadRead,
    NewChatThreadUpdate,
    NewChatThreadWithMessages,
    ThreadHistoryLoadResponse,
    ThreadListItem,
    ThreadListResponse,
)
from .new_llm_config import (
    DefaultSystemInstructionsResponse,
    GlobalNewLLMConfigRead,
    LLMPreferencesRead,
    LLMPreferencesUpdate,
    NewLLMConfigCreate,
    NewLLMConfigPublic,
    NewLLMConfigRead,
    NewLLMConfigUpdate,
)
from .podcasts import PodcastBase, PodcastCreate, PodcastRead, PodcastUpdate
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 .users import UserCreate, UserRead, UserUpdate
from .video_presentations import (
    VideoPresentationBase,
    VideoPresentationCreate,
    VideoPresentationRead,
    VideoPresentationUpdate,
)

__all__ = [
    # Chat schemas (assistant-ui integration)
    "ChatMessage",
    # Chunk schemas
    "ChunkBase",
    "ChunkCreate",
    "ChunkRead",
    "ChunkUpdate",
    "DefaultSystemInstructionsResponse",
    # Document schemas
    "DocumentBase",
    "DocumentRead",
    "DocumentStatusBatchResponse",
    "DocumentStatusItemRead",
    "DocumentStatusSchema",
    "DocumentTitleRead",
    "DocumentTitleSearchResponse",
    "DocumentUpdate",
    "DocumentWithChunksRead",
    "DocumentsCreate",
    # Google Drive schemas
    "DriveItem",
    "ExtensionDocumentContent",
    "ExtensionDocumentMetadata",
    "GlobalImageGenConfigRead",
    "GlobalNewLLMConfigRead",
    "GoogleDriveIndexRequest",
    "GoogleDriveIndexingOptions",
    # Base schemas
    "IDModel",
    # Image Generation Config schemas
    "ImageGenerationConfigCreate",
    "ImageGenerationConfigPublic",
    "ImageGenerationConfigRead",
    "ImageGenerationConfigUpdate",
    # Image Generation schemas
    "ImageGenerationCreate",
    "ImageGenerationListRead",
    "ImageGenerationRead",
    # RBAC schemas
    "InviteAcceptRequest",
    "InviteAcceptResponse",
    "InviteCreate",
    "InviteInfoResponse",
    "InviteRead",
    "InviteUpdate",
    # LLM Preferences schemas
    "LLMPreferencesRead",
    "LLMPreferencesUpdate",
    # Log schemas
    "LogBase",
    "LogCreate",
    "LogFilter",
    "LogRead",
    "LogUpdate",
    # Auth schemas
    "LogoutAllResponse",
    "LogoutRequest",
    "LogoutResponse",
    # Search source connector schemas
    "MCPConnectorCreate",
    "MCPConnectorRead",
    "MCPConnectorUpdate",
    "MCPServerConfig",
    "MembershipRead",
    "MembershipReadWithUser",
    "MembershipUpdate",
    "NewChatMessageAppend",
    "NewChatMessageCreate",
    "NewChatMessageRead",
    "NewChatRequest",
    "NewChatThreadCreate",
    "NewChatThreadRead",
    "NewChatThreadUpdate",
    "NewChatThreadWithMessages",
    # NewLLMConfig schemas
    "NewLLMConfigCreate",
    "NewLLMConfigPublic",
    "NewLLMConfigRead",
    "NewLLMConfigUpdate",
    "PaginatedResponse",
    "PermissionInfo",
    "PermissionsListResponse",
    # Podcast schemas
    "PodcastBase",
    "PodcastCreate",
    "PodcastRead",
    "PodcastUpdate",
    "RefreshTokenRequest",
    "RefreshTokenResponse",
    # Report schemas
    "ReportBase",
    "ReportContentRead",
    "ReportContentUpdate",
    "ReportRead",
    "ReportVersionInfo",
    "RoleCreate",
    "RoleRead",
    "RoleUpdate",
    "SearchSourceConnectorBase",
    "SearchSourceConnectorCreate",
    "SearchSourceConnectorRead",
    "SearchSourceConnectorUpdate",
    # Search space schemas
    "SearchSpaceBase",
    "SearchSpaceCreate",
    "SearchSpaceRead",
    "SearchSpaceUpdate",
    "SearchSpaceWithStats",
    "ThreadHistoryLoadResponse",
    "ThreadListItem",
    "ThreadListResponse",
    "TimestampModel",
    # User schemas
    "UserCreate",
    "UserRead",
    "UserSearchSpaceAccess",
    "UserUpdate",
    # Video Presentation schemas
    "VideoPresentationBase",
    "VideoPresentationCreate",
    "VideoPresentationRead",
    "VideoPresentationUpdate",
]