chat_state.py

# chat_state.py
from dataclasses import dataclass
from typing import Dict, List, Optional


@dataclass
class ChatState:
    chat_history: Dict[str, List[Dict[str, str]]]
    current_chat_id: Optional[str]
    temp_chat: Optional[List[Dict[str, str]]]
    delete_chat_id: Optional[str]

    @classmethod
    def initialize(cls) -> "ChatState":
        return cls(
            chat_history={}, current_chat_id=None, temp_chat=None, delete_chat_id=None
        )