agent_state.py

from pydantic import BaseModel, Field
from typing import Optional

class AgentState(BaseModel):
    audio_path: str
    transcribed_text: Optional[str] = None
    news_report: Optional[str] = None
    feedback: Optional[str] = Field(default=None, description="Feedback from the human for revision")
    approved: bool = Field(default=False, description="Has the human approved the summary?")