agent_plan_solve/graph/state.py

# agent_plan_solve/graph/state.py

from typing import Annotated, TypedDict
import operator
from pydantic import BaseModel, Field

class Plan(BaseModel):
    """Plan to follow in future"""
    steps: list[str] = Field(..., description="Different steps to follow, in sorted order")

class PlanState(TypedDict):
    task: str
    plan: Plan
    past_steps: Annotated[list[str], operator.add]
    final_response: str