src/state.py

from typing_extensions import TypedDict, List, Dict, Any
from langgraph.graph.message import add_messages
# Make your state comprehensive enough to track all the important information, but avoid bloating it with unnecessary details.

class AgentState(TypedDict):
    messages: List[Dict[str, Any]]
    question: str
    thought: str
    action: str
    action_input: str
    observation: str 
    final_answer: str
    tools: str
    tool_names: str