File size: 752 Bytes
aab927d
 
579780d
 
aab927d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
class ContextManager:
    # ์ „์—ญ ์ƒ์ˆ˜ ์„ค์ •
    USER_NAME = "์‚ฌ์šฉ์ž"
    BOT_NAME = "ํƒ„์ง€๋กœ"

    def __init__(self):
        self.user_name = self.USER_NAME
        self.bot_name = self.BOT_NAME
        self.history = []

    def getUserName(self) -> str:
        return self.user_name
    
    def getBotName(self) -> str:
        return self.bot_name
    
    def getHistory(self) -> str:
        return self.history
    
    def setHistory(self, new_history: list):
        self.history = new_history
    
    # ๋Œ€ํ™” ๊ธฐ๋ก์„ history์— ์ถ”๊ฐ€
    def addHistory(self, role: str, text: str):
        self.history.append({"role": role, "text": text})

    # ๋Œ€ํ™” ๊ธฐ๋ก ์ดˆ๊ธฐํ™”
    def clearHistory(self):
        self.history = []