| | |
| | 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 |
| |
|
