kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-06-02 00:57:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support plugins

Browse Source
This commit is contained in:
lanvent
2023-03-12 11:53:06 +08:00
parent 9e07703eb1
commit 0fcf0824dc
8 changed files with 220 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
bot/chatgpt/chat_gpt_bot.py
View File

@@ -60,12 +60,13 @@ class ChatGPTBot(Bot):
ok, retstring = self.create_img(query, 0)
reply = None
if ok:
reply = {'type': 'IMAGE', 'content': retstring}
reply = {'type': 'IMAGE_URL', 'content': retstring}
else:
reply = {'type': 'ERROR', 'content': retstring}
return reply
else:
reply= {'type':'ERROR', 'content':'Bot不支持处理{}类型的消息'.format(context['type'])}
return reply
def reply_text(self, session, session_id, retry_count=0) -> dict:
'''
@@ -139,7 +140,11 @@ class ChatGPTBot(Bot):
class SessionManager(object):
def __init__(self):
self.sessions = {}
if conf().get('expires_in_seconds'):
sessions = ExpiredDict(conf().get('expires_in_seconds'))
else:
sessions = dict()
self.sessions = sessions
def build_session_query(self, query, session_id):
'''