src/supervisor/state.py

# src/supervisor/state.py

from typing import TypedDict, Annotated, List
from langgraph.graph.message import add_messages
from langchain_core.messages import BaseMessage

class AgentState(TypedDict):
    """
    Represents the state of our agent at any given time.
    """
    # The initial question from the user.
    question: str
    
    # The local path to a downloaded file, if any.
    file_path: str | None
    
    # The history of messages in the conversation.
    # `add_messages` is a special LangGraph reducer that appends messages.
    messages: Annotated[list[BaseMessage], add_messages]
    
    # A final, formatted answer for submission.
    final_answer: str | None