from pydantic import BaseModel, Field


class RewriteRequest(BaseModel):
    source_text: str = Field(..., min_length=20)
    title_hint: str = ""
    tone: str = "专业、可信、可读性强"
    audience: str = "公众号读者"
    keep_points: str = ""
    avoid_words: str = ""


class RewriteResponse(BaseModel):
    title: str
    summary: str
    body_markdown: str


class WechatPublishRequest(BaseModel):
    title: str
    summary: str = ""
    body_markdown: str
    author: str = ""


class IMPublishRequest(BaseModel):
    title: str
    body_markdown: str


class PublishResponse(BaseModel):
    ok: bool
    detail: str
    data: dict | None = None