agent/custom_state.py

from typing import TypedDict, List, Dict, Optional, Any, Annotated
from langgraph.graph.message import add_messages
from langchain_core.messages import AnyMessage

class AssistantState(TypedDict):
    messages: Annotated[list[AnyMessage], add_messages]
    filename : str
    file_extension : str