SurfSense/surfsense_backend/app/schemas/surfsense_docs.py

"""
Schemas for Surfsense documentation.
"""

from datetime import datetime

from pydantic import BaseModel, ConfigDict


class SurfsenseDocsChunkRead(BaseModel):
    """Schema for a Surfsense docs chunk."""

    id: int
    content: str

    model_config = ConfigDict(from_attributes=True)


class SurfsenseDocsDocumentRead(BaseModel):
    """Schema for a Surfsense docs document (without chunks)."""

    id: int
    title: str
    source: str
    content: str
    created_at: datetime | None = None
    updated_at: datetime | None = None

    model_config = ConfigDict(from_attributes=True)


class SurfsenseDocsDocumentWithChunksRead(BaseModel):
    """Schema for a Surfsense docs document with its chunks."""

    id: int
    title: str
    source: str
    content: str
    chunks: list[SurfsenseDocsChunkRead]

    model_config = ConfigDict(from_attributes=True)