| |
| from typing import Dict, List, Optional |
| from chat_state import ChatState |
|
|
|
|
| class ChatManager: |
| def __init__(self, system_prompt: str): |
| self.system_prompt = system_prompt |
|
|
| def create_new_chat(self) -> List[Dict[str, str]]: |
| return [{"role": "system", "content": self.system_prompt}] |
|
|
| def save_chat( |
| self, |
| temp_chat: List[Dict[str, str]], |
| chat_history: Dict[str, List[Dict[str, str]]], |
| ) -> str: |
| if temp_chat and len(temp_chat) > 1: |
| new_chat_id = str(len(chat_history) + 1) |
| chat_history[new_chat_id] = temp_chat |
| return new_chat_id |
| return None |
|
|
| def delete_chat(self, chat_id: str, chat_state: ChatState) -> None: |
| if chat_id in chat_state.chat_history: |
| del chat_state.chat_history[chat_id] |
| if chat_state.current_chat_id == chat_id: |
| chat_state.current_chat_id = None |
| chat_state.delete_chat_id = None |
|
|
