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
    user_input: str
    audio_path: str
    transcript: str
    audio_transcript: str
    wiki_query: str
    wiki_result: str
    task_id: str
    tool_counter: int