Blog-Writing-Agent / src /models /Plan_model.py
VashuTheGreat2's picture
Upload folder using huggingface_hub
63de3ab verified
raw
history blame contribute delete
489 Bytes
from pydantic import BaseModel,Field
from typing import List,Literal
from src.models.Task_models import Task
class Plan(BaseModel):
blog_title: str
audience: str = Field(..., description="Who this blog is for.")
tone: str = Field(..., description="Writing tone (e.g., practical, crisp).")
blog_kind: Literal["explainer", "tutorial", "news_roundup", "comparison", "system_design"] = "explainer"
constraints: List[str] = Field(default_factory=list)
tasks: List[Task]