SurfSense/surfsense_backend/app/schemas/podcasts.py

from typing import Any, Literal

from pydantic import BaseModel, ConfigDict

from .base import IDModel, TimestampModel


class PodcastBase(BaseModel):
    title: str
    podcast_transcript: list[Any]
    file_location: str = ""
    search_space_id: int
    chat_state_version: int | None = None


class PodcastCreate(PodcastBase):
    pass


class PodcastUpdate(PodcastBase):
    pass


class PodcastRead(PodcastBase, IDModel, TimestampModel):
    model_config = ConfigDict(from_attributes=True)


class PodcastGenerateRequest(BaseModel):
    type: Literal["DOCUMENT", "CHAT"]
    ids: list[int]
    search_space_id: int
    podcast_title: str | None = None
    user_prompt: str | None = None