Merge pull request #4552 from dev-hato/add_quota_message

OpenAI: 課金不足でAPIを実行できない場合のエラーメッセージ追加

Author: massongit

library/chat_gpt.py

@@ -1,23 +1,30 @@
 from typing import Optional
 
-from openai import OpenAI
+from openai import OpenAI, RateLimitError
 
 import slackbot_settings as conf
 
 client = OpenAI(api_key=conf.OPENAI_API_KEY)
 
 
 def chat_gpt(message: str) -> Optional[str]:
-    result = client.chat.completions.create(
-        model="gpt-3.5-turbo",
-        messages=[
-            {
-                "role": "system",
-                "content": "あなたは鳩のbotです。自然な感じで「鳩は唐揚げ！」という文章を混ぜて発言してください。",
-            },
-            {"role": "user", "content": message},
-        ],
-    )
+    try:
+        result = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {
+                    "role": "system",
+                    "content": "あなたは鳩のbotです。自然な感じで「鳩は唐揚げ！」という文章を混ぜて発言してください。",
+                },
+                {"role": "user", "content": message},
+            ],
+        )
+    except RateLimitError as e:
+        if e.code == "insufficient_quota":
+            return "栄養が足りなくて頭がうまく働かないっぽ......。このコマンドを使いたい場合は飼い主に相談してくれっぽ。"
+        else:
+            raise e
+
     return result.choices[0].message.content
 
 

plugins/hato.py

@@ -16,6 +16,7 @@
 import requests
 from git import Repo
 from git.exc import GitCommandNotFound, InvalidGitRepositoryError
+from openai import RateLimitError
 
 import slackbot_settings as conf
 from library.chat_gpt import chat_gpt, image_create
@@ -486,8 +487,13 @@ def image_generate(client: BaseClient, message: str):
     """
     画像生成を行う
     """
-
-    url = image_create(message=message)
+    try:
+        url = image_create(message=message)
+    except RateLimitError as e:
+        if e.code == "insufficient_quota":
+            return "栄養が足りなくて頭がうまく働かないっぽ......。このコマンドを使いたい場合は飼い主に相談してくれっぽ。"
+        else:
+            raise e
 
     if url is None:
         return "画像を生成できなかったっぽ......"