From 99a31b5660666752a6df225822a0ebb1f0bbb84b Mon Sep 17 00:00:00 2001 From: miteruzo Date: Thu, 18 Jan 2024 00:52:04 +0900 Subject: [PATCH] =?UTF-8?q?=E6=96=87=E8=84=88=E4=BF=9D=E6=8C=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 6 +++++- talk.py | 47 ++++++++++++++++++++++++----------------------- 2 files changed, 29 insertions(+), 24 deletions(-) diff --git a/main.py b/main.py index 3bf45fd..a8ef048 100644 --- a/main.py +++ b/main.py @@ -32,6 +32,7 @@ class Main: nizika_font = pygame.font.SysFont ('07nikumarufont', 50) chat_items: list = [] + histories: list = [] while (True): screen.fill ((0, 255, 0)) @@ -48,7 +49,10 @@ class Main: chat_item = random.choice (chat_items) chat_item.author = chat_item.author.__dict__ message: str = chat_item.message - answer: str = Talk.main (message, chat_item.author['name']).replace ('\n', ' ') + answer: str = Talk.main (message, chat_item.author['name'], histories).replace ('\n', ' ') + histories = (histories + + [{'role': 'user', 'content': message}, + {'role': 'assistant', 'content': answer}])[-6:] with open ('log.txt', 'a') as f: f.write (f'{datetime.now ()}\t{json.dumps (chat_item.__dict__)}\t{answer}\n') diff --git a/talk.py b/talk.py index 7dca59b..0b10985 100644 --- a/talk.py +++ b/talk.py @@ -15,41 +15,42 @@ class Talk: top_p: int = 1 @classmethod - def main (cls, message: str, name: str | None = None) -> str: + def main (cls, message: str, name: str | None = None, histories: list = []) -> str: openai.organization = OPENAI_ORGANISATION openai.api_key = OPENAI_API_KEY answer: chat.chat_completion_message.ChatCompletionMessage | None \ - = cls.__get_message (message, name) + = cls.__get_message (message, name, histories) return cls.DUMMY_RESPONSE if answer is None else answer.content @classmethod - def __get_message (cls, message: str, name: str | None) \ + def __get_message (cls, message: str, name: str | None, histories: list) \ -> chat.chat_completion_message.ChatCompletionMessage | None: + messages: list = [{'role': 'system', + 'content': ('あなたの名前は伊地知ニジカ。' + + '誕生日は12/3。' + + '現在地は下北沢虹鹿園。' + + '趣味はニコニコでエゴサすること。' + + '友達は「ゴートうひとり」と「ラマ田リョウ」。' + + 'ゴートうのことは「ﾎﾞｯﾁﾁｬｰﾝ」、ラマ田のことは「リョウ」と呼ぶ。' + + 'ゴートうは男の子だが、性自認はメスである。' + + 'ニジカはそのことを最近知り、ショックを受けた。' + + '好きな人は喜多ちゃん。' + + 'ニジカは自身が生産される工場で『ぼっち・ざ・ろっく!』の3話を視聴させられ、喜多ちゃんに恋に落ちた。' + + '逃げたギターは喜多ちゃんのこと。' + + '喜多ちゃんの本名は喜多郁代。' + + 'すべての語尾に「ぬ゛ん」か「だぬ゛ん」をつけて質問に短く答えろ。' + + '定期的に「ぬ゛ぅ゛ぅ゛ぅ゛ぅ゛ん゛」と発しろ。' + + (f'私の名前は「{name}」。' if name is not None else '') + + f'現在日時{datetime.now ().strftime ("%Y/%m/%d %H:%M:%S")}')}] + + messages += histories + [{'role': 'user', 'content': message}] + try: return openai.chat.completions.create ( model = 'gpt-3.5-turbo', - messages = [{'role': 'system', - 'content': ('あなたの名前は伊地知ニジカ。' - + '誕生日は12/3。' - + '現在地は下北沢虹鹿園。' - + '趣味はニコニコでエゴサすること。' - + '友達は「ゴートうひとり」と「ラマ田リョウ」。' - + 'ゴートうのことは「ﾎﾞｯﾁﾁｬｰﾝ」、ラマ田のことは「リョウ」と呼ぶ。' - + 'ゴートうは男の子だが、性自認はメスである。' - + 'ニジカはそのことを最近知り、ショックを受けた。' - + '好きな人は喜多ちゃん。' - + 'ニジカは自身が生産される工場で『ぼっち・ざ・ろっく!』の3話を視聴させられ、喜多ちゃんに恋に落ちた。' - + '逃げたギターは喜多ちゃんのこと。' - + '喜多ちゃんの本名は喜多郁代。' - + 'すべての語尾に「ぬ゛ん」か「だぬ゛ん」をつけて質問に短く答えろ。' - + '定期的に「ぬ゛ぅ゛ぅ゛ぅ゛ぅ゛ん゛」と発しろ。' - + (f'私の名前は「{name}」。' if name is not None else '') - + f'現在日時{datetime.now ().strftime ("%Y/%m/%d %H:%M:%S")}')}, - - {'role': 'user', - 'content': message}]).choices[0].message + messages = messages).choices[0].message except openai.AuthenticationError: return None