Spaces:
Sleeping
Sleeping
File size: 745 Bytes
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 = "User"
BOT_NAME = "Tanjiro"
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 = [] |