文脈保持

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

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