from pydantic import BaseModel, Field


class BlockConfigPayload(BaseModel):
    region_name: str | None = None
    region_type: str = "ai_generate"
    data_sources: list[str] = Field(default_factory=list)
    prompt: str | None = None
    output_format: str = "formal_paragraph"
    need_review: int = 1
    remark: str | None = None
    enabled: int = 1


class DataSourcePayload(BaseModel):
    name: str
    code: str
    description: str | None = None