src/genai/analytics_chatbot/utils/state.py

from pydantic import BaseModel, ConfigDict , Field
from typing import  Annotated , TypedDict
from langgraph.graph.message import add_messages

class State(TypedDict):
    messages: Annotated[list, add_messages]
    possible_endpoints:list
    endpoint: str
    method: str
    needed_parameters: dict
    response:dict
    error_message:str
    latest_message:str
    parameters_values:str
    backup_data:str
    query_type: str