Spaces:
Build error
Build error
| from typing import Dict, List | |
| # chat_manager.py | |
| class ChatManager: | |
| def __init__(self): | |
| self.user_chats: Dict[str, List[Dict]] = {} | |
| def initialize_chat(self, sender_id: str) -> None: | |
| if sender_id not in self.user_chats: | |
| self.user_chats[sender_id] = [ | |
| {"role": "user", "parts": "This is the chat history so far"} | |
| ] | |
| def append_message(self, sender_id: str, role: str, content: str) -> None: | |
| if content: | |
| self.user_chats[sender_id].append({"role": role, "parts": content}) | |
| def get_chat_history(self, sender_id: str) -> List[Dict]: | |
| return self.user_chats.get(sender_id, []) |