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" | |
) | |