File size: 460 Bytes
23b8b85 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# 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
)
|