iukea/open-notebooklm

1"""2schema.py3"""4 5from typing import Literal, List6 7from pydantic import BaseModel, Field8 9 10class DialogueItem(BaseModel):11    """A single dialogue item."""12 13    speaker: Literal["Host (Jane)", "Guest"]14    text: str15 16 17class ShortDialogue(BaseModel):18    """The dialogue between the host and guest."""19 20    scratchpad: str21    name_of_guest: str22    dialogue: List[DialogueItem] = Field(23        ..., description="A list of dialogue items, typically between 11 to 17 items"24    )25 26 27class MediumDialogue(BaseModel):28    """The dialogue between the host and guest."""29 30    scratchpad: str31    name_of_guest: str32    dialogue: List[DialogueItem] = Field(33        ..., description="A list of dialogue items, typically between 19 to 29 items"34    )35