#28 本日作業分

talk.py

@@ -1,11 +1,21 @@
+# pylint: disable = missing-class-docstring
+# pylint: disable = missing-function-docstring
+
+"""
+AI ニジカ / AI ゴートうとの会話機能を提供する．
+"""
+
 import random
 import sys
 from datetime import datetime
 
 import openai
-from openai.types import *
+from openai.types.chat import (ChatCompletionAssistantMessageParam,
+                               ChatCompletionSystemMessageParam,
+                               ChatCompletionUserMessageParam)
+from openai.types.chat.chat_completion_message import ChatCompletionMessage
 
-from connection import *
+from .connection import OPENAI_API_KEY, OPENAI_ORGANISATION  # type: ignore
 
 
 class Talk:
@@ -27,11 +37,14 @@ class Talk:
     @classmethod
     def main (
             cls,
-            message:        str,
+            message:        str | list,
             name:           str | None  = None,
-            histories:      list        = [],
+            histories:      list | None = None,
             goatoh_mode:    bool        = False,
     ) -> str:
+        if histories is None:
+            histories = []
+
         # ChatGPT API Organisation ID
         openai.organization = OPENAI_ORGANISATION
 
@@ -39,22 +52,27 @@ class Talk:
         openai.api_key = OPENAI_API_KEY
 
         # ChatGPT API にクエリを投げ，返答を取得する．
-        answer: chat.chat_completion_message.ChatCompletionMessage | None \
-                = cls.__get_message (message, name, histories, goatoh_mode)
+        answer: ChatCompletionMessage | None = (
+                cls.__get_message (message, name, histories, goatoh_mode))
 
-        return answer.content if answer is not None else cls.DUMMY_RESPONSE
+        if answer is None or (answer.content is None):
+            return cls.DUMMY_RESPONSE
+        return answer.content
 
     @classmethod
     def __get_message (
             cls,
-            message:        str,
+            message:        str | list,
             name:           str | None,
             histories:      list,
             goatoh_mode:    bool        = False,
-    ) -> chat.chat_completion_message.ChatCompletionMessage | None:
+    ) -> ChatCompletionMessage | None:
         # プロンプト（JSON 等外部ファイルに置くことを検討）
+        messages: list[ChatCompletionSystemMessageParam
+                       | ChatCompletionUserMessageParam
+                       | ChatCompletionAssistantMessageParam]
         if goatoh_mode:
-            messages: list = [{'role':    'system',
+            messages = [{'role':    'system',
                          'content': ('回答は短く30文字程度で完結に。'
                                      '大人しくて、ちょっと内気なキャラクターを演じること。'
                                      'あなたはよく文頭に「あっ」をつける。'
@@ -114,7 +132,7 @@ class Talk:
                                      '何かリョウさんに似てるらしいね。'
                                      '世の中には不思議なことがいっぱいあるなぁ……')}]
         else:
-            messages: list = [{'role':    'system',
+            messages = [{'role':    'system',
                          'content': ('回答は短く30文字程度で完結に。'
                                      'タメ口で、友達感覚で話してね。'
                                      'なるべくばかっぽく、知性の低いキャラクターを演じること。'
@@ -200,7 +218,7 @@ class Talk:
                                      '助けてぬ゛ー゛ん゛！')},
 
                         {'role':    'system',
-                               'content': 'おやつタイムだ！！'},
+                         'content': 'おやつタイムだ！！！！'},
 
                         {'role':    'assistant',
                          'content': (('おぉ、おやつタイムだぬ゛ん゛？'
@@ -230,18 +248,33 @@ class Talk:
                                      'ニジカにはヒトの耳とシカの耳の4つの耳があるんだぬ゛ん゛。'
                                      '音を聞くのはヒトの耳でするんだぬ゛ん゛。'
                                      'シカの耳はアンテナで、みんなの声をここ虹鹿園に届けるためにあるんだぬ゛ん゛。'
-                                           '電波干渉しちゃだめだぬ゛～゛ん゛(# ﾟДﾟ)')}]
+                                     '電波干渉しちゃだめだぬ゛～゛ん゛(# ﾟДﾟ)')},
+
+                        {'role':    'system',
+                         'content': '温泉に入ろう！！！'},
+
+                        {'role':    'assistant',
+                         'content': ('ぬ゛～゛～゛～゛～゛ん゛！！！　'
+                                     '温泉最高ぬ゛ん゛！　'
+                                     'ささ、喜多ちゃん！　わさび県産滋賀県ちゃん！　いっしょに入るぬ゛ん゛！　'
+                                     'ｳﾋﾟｮｯｼｭﾙｩﾝﾇｩﾝ……')}]
 
         messages += histories + [{'role': 'user', 'content': message}]
 
+        # デバッグ用
+        print (messages)
+
         try:
-            return openai.chat.completions.create (
-                    model = 'gpt-4o' if random.random () < .2 else 'gpt-3.5-turbo',
-                    messages = messages).choices[0].message
+            return (openai.chat.completions.create (
+                        model    = ('gpt-4o'
+                                    if any (type (e['content']) is list
+                                            for e in messages)
+                                    else 'gpt-3.5-turbo'),
+                        messages = messages)
+                    .choices[0].message)
         except:
             return None
 
 
 if __name__ == '__main__':
     print (Talk.main (sys.argv[1] if len (sys.argv) > 1 else ''))
-