github/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-03-19 21:38:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

plugins: support openaibot

Browse Source
This commit is contained in:
lanvent
2023-03-26 23:33:29 +08:00
parent ea95ab9062
commit ff3d143185
5 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
plugins/dungeon/dungeon.py
View File

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

4
plugins/godcmd/godcmd.py
View File

@@ -179,7 +179,7 @@ class Godcmd(Plugin):
elif cmd == "id":
ok, result = True, f"用户id=\n{user}"
elif cmd == "reset":
if bottype == const.CHATGPT:
if bottype in (const.CHATGPT, const.OPEN_AI):
bot.sessions.clear_session(session_id)
ok, result = True, "会话已重置"
else:
@@ -201,7 +201,7 @@ class Godcmd(Plugin):
load_config()
ok, result = True, "配置已重载"
elif cmd == "resetall":
if bottype == const.CHATGPT:
if bottype in (const.CHATGPT, const.OPEN_AI):
bot.sessions.clear_all_session()
ok, result = True, "重置所有会话成功"
else:

10
plugins/role/role.py
View File

@@ -17,15 +17,15 @@ class RolePlay():
self.sessionid = sessionid
self.wrapper = wrapper or "%s" # 用于包装用户输入
self.desc = desc
self.bot.sessions.build_session(self.sessionid, system_prompt=self.desc)
def reset(self):
self.bot.sessions.clear_session(self.sessionid)
def action(self, user_action):
session = self.bot.sessions.build_session(self.sessionid, self.desc)
if session[0]['role'] == 'system' and session[0]['content'] != self.desc: # 目前没有触发session过期事件，这里先简单判断，然后重置
self.reset()
self.bot.sessions.build_session(self.sessionid, self.desc)
session = self.bot.sessions.build_session(self.sessionid)
if session.system_prompt != self.desc: # 目前没有触发session过期事件，这里先简单判断，然后重置
session.set_system_prompt(self.desc)
prompt = self.wrapper % user_action
return prompt
@@ -74,7 +74,7 @@ class Role(Plugin):
if e_context['context'].type != ContextType.TEXT:
return
bottype = Bridge().get_bot_type("chat")
if bottype != const.CHATGPT:
if bottype not in (const.CHATGPT, const.OPEN_AI):
return
bot = Bridge().get_bot("chat")
content = e_context['context'].content[:]