chore: add use_azure_chatgpt config

bot/bot.py

@@ -0,0 +1,18 @@
+from common.context import Context
+from config import conf
+from common.singleton import singleton
+from common.reply import Reply
+
+
+@singleton
+class Bot:
+    def reply(self, context: Context) -> Reply:
+        use_azure_chatgpt = conf().get("use_azure_chatgpt", False)
+        if use_azure_chatgpt:
+            from bot.azure_chatgpt import AzureChatGPTBot
+
+            return AzureChatGPTBot().reply(context)
+        else:
+            from bot.chatgpt import ChatGPTBot
+
+            return ChatGPTBot().reply(context)

bot/chatgpt.py

@@ -22,7 +22,7 @@ def __init__(self):
             "temperature": conf().get("temperature"),
         }
 
-    def reply(self, context: Context):
+    def reply(self, context: Context) -> Reply:
         query = context.query
         logger.info(f"[ChatGPT] Query={query}")
         if context.type == ContextType.CREATE_IMAGE:
@@ -38,7 +38,7 @@ def reply(self, context: Context):
                 )
             return Reply(ReplyType.TEXT, response["content"])
 
-    def reply_img(self, query):
+    def reply_img(self, query) -> Reply:
         create_image_size = conf().get("create_image_size", "256x256")
         try:
             response = openai.Image.create(prompt=query, n=1, size=create_image_size)

channel/wechat.py

@@ -6,7 +6,7 @@
 from utils.log import logger
 from utils import const
 import os
-from bot.chatgpt import ChatGPTBot
+from bot.bot import Bot
 from common.singleton import singleton
 from config import conf
 from utils.check import check_prefix, is_wx_account
@@ -144,7 +144,7 @@ def handle_reply(self, msg: Message, context: Context):
         if e1.is_bypass:
             return self.send(e1.reply, e1.message)
 
-        reply = ChatGPTBot().reply(e1.context)
+        reply = Bot().reply(e1.context)
 
         e2 = PluginManager().emit(
             Event(

config.template.json

@@ -1,6 +1,7 @@
 {
   "openai_api_key": "YOUR API SECRET KEY",
   "model": "gpt-3.5-turbo",
+  "use_azure_chatgpt": false,
   "azure_deployment_id": "",
   "role_desc": "You are a helpful assistant.",
   "session_expired_duration": 3600,