ChatGPT対応

.env.example

@@ -1,4 +1,5 @@
 DATABASE_URL=postgres://postgres:password@postgres:5432/
+OPENAI_API_KEY=
 SLACK_API_TOKEN=
 SLACK_SIGNING_SECRET=
 YAHOO_API_TOKEN=

Pipfile

@@ -31,3 +31,4 @@ slackclient = "==2.9.4"
 gitpython = "==3.1.31"
 pandas = "==1.5.3"
 matplotlib = "==3.7.0"
+openai = "==0.27.0"

library/chat_gpt.py

@@ -0,0 +1,15 @@
+import openai
+
+import slackbot_settings as conf
+
+
+def chat_gpt(message: str) -> str:
+    openai.api_key = conf.OPENAI_API_KEY
+    result = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": "自然な感じで「鳩は唐揚げ！」という文章を混ぜて発言してください。"},
+            {"role": "user", "content": message},
+        ],
+    )
+    return result.get("choices")[0].message.content

plugins/analyze.py

@@ -35,6 +35,7 @@ def analyze_message(message: str) -> Callable[[BaseClient], None]:
         "version": lambda m: hato.version,
         "にゃーん": lambda m: hato.yoshiyoshi,
         "おみくじ": lambda m: hato.omikuji,
+        "chat": lambda m: partial(hato.chat, message=m[len("chat") :].strip()),
     }
 
     for key, method in conditions.items():

plugins/hato.py

@@ -18,6 +18,7 @@
 from git.exc import GitCommandNotFound, InvalidGitRepositoryError
 
 import slackbot_settings as conf
+from library.chat_gpt import chat_gpt
 from library.clientclass import BaseClient
 from library.earthquake import generate_map_img, get_quake_list
 from library.geo import get_geo_data
@@ -466,3 +467,12 @@ def omikuji():
     """
 
     return omikuji_draw(omikuji_results)[1].message
+
+
+@action("chat")
+def chat(message: str):
+    """
+    chat-gptに聞く
+    """
+
+    return chat_gpt(message=message)

slackbot_settings.py

@@ -34,6 +34,9 @@
 DEFAULT_REPLY = "使い方がわからない時は `help` とメンションするっぽ!"
 PLUGINS = ["plugins"]
 
+# ChatGPT用の設定
+OPENAI_API_KEY = str(os.environ["OPENAI_API_KEY"])
+
 GIT_COMMIT_HASH = os.environ.get("GIT_COMMIT_HASH")
 
 VERSION = "2.4.4"