from typing_extensions import TypedDict
from typing import Annotated
from langgraph.graph.message import add_messages

class AgentState(TypedDict, total=False):
    messages: Annotated[list, add_messages]
    web_search_query: str
    ocr_path: str
    excel_path: str
    excel_sheet_name: str
    web_search_result: str
    ocr_result: str
    excel_result: str
    final_answer: str