#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:
@@ -29,9 +39,12 @@ class Talk:
             cls,
             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,10 +52,12 @@ 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 (
@@ -51,10 +66,13 @@ class Talk:
             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文字程度で完結に。'
                                      'タメ口で、友達感覚で話してね。'
                                      'なるべくばかっぽく、知性の低いキャラクターを演じること。'
@@ -260,4 +278,3 @@ class Talk:
 
 if __name__ == '__main__':
     print (Talk.main (sys.argv[1] if len (sys.argv) > 1 else ''))
-