Sfoglia il codice sorgente

fix: role,dungeon,godcmd support azure bot

lanvent 2 anni fa
parent
commit
8781b1e976
3 ha cambiato i file con 5 aggiunte e 5 eliminazioni
  1. 1 1
      plugins/dungeon/dungeon.py
  2. 2 2
      plugins/godcmd/godcmd.py
  3. 2 2
      plugins/role/role.py

+ 1 - 1
plugins/dungeon/dungeon.py

@@ -64,7 +64,7 @@ class Dungeon(Plugin):
         if e_context["context"].type != ContextType.TEXT:
             return
         bottype = Bridge().get_bot_type("chat")
-        if bottype not in (const.CHATGPT, const.OPEN_AI):
+        if bottype not in [const.OPEN_AI, const.CHATGPT, const.CHATGPTONAZURE]:
             return
         bot = Bridge().get_bot("chat")
         content = e_context["context"].content[:]

+ 2 - 2
plugins/godcmd/godcmd.py

@@ -265,7 +265,7 @@ class Godcmd(Plugin):
                     except Exception as e:
                         ok, result = False, "你没有设置私有api_key"
                 elif cmd == "reset":
-                    if bottype in (const.CHATGPT, const.OPEN_AI):
+                    if bottype in [const.OPEN_AI, const.CHATGPT, const.CHATGPTONAZURE]:
                         bot.sessions.clear_session(session_id)
                         channel.cancel_session(session_id)
                         ok, result = True, "会话已重置"
@@ -288,7 +288,7 @@ class Godcmd(Plugin):
                             load_config()
                             ok, result = True, "配置已重载"
                         elif cmd == "resetall":
-                            if bottype in (const.CHATGPT, const.OPEN_AI):
+                            if bottype in [const.OPEN_AI, const.CHATGPT, const.CHATGPTONAZURE]:
                                 channel.cancel_all_session()
                                 bot.sessions.clear_all_session()
                                 ok, result = True, "重置所有会话成功"

+ 2 - 2
plugins/role/role.py

@@ -98,8 +98,8 @@ class Role(Plugin):
     def on_handle_context(self, e_context: EventContext):
         if e_context["context"].type != ContextType.TEXT:
             return
-        bottype = Bridge().get_bot_type("chat")
-        if bottype not in (const.CHATGPT, const.OPEN_AI):
+        btype = Bridge().get_bot_type("chat")
+        if btype not in [const.OPEN_AI, const.CHATGPT, const.CHATGPTONAZURE]:
             return
         bot = Bridge().get_bot("chat")
         content = e_context["context"].content[:]