from __future__ import annotations

from pydantic import BaseModel, Field


class SummaryRequest(BaseModel):
    transcript: str = Field(..., min_length=1)
    llm_model: str
    prompt: str = Field("Summarize the transcript below.")
    generate_title: bool = Field(default=True)


class SpeakerNameDetectionRequest(BaseModel):
    utterances: list = Field(..., min_length=1)
    llm_model: str