from pydantic import BaseModel, Field
from typing import List


class PlanStep(BaseModel):
    """A step for the planning to be followed"""

    agent: str
    description: str


class Plan(BaseModel):
    """Plan to follow in future"""

    steps: List[PlanStep] = Field(
        description="different steps to follow, should be in sorted order"
    )