Customer_Support/models.py

from typing import TypedDict, Dict

class State(TypedDict):
    """State for the customer support workflow."""
    query: str
    category: str
    sentiment: str
    response: str