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

chore: strip content

Browse Source
This commit is contained in:
lanvent
2023-04-16 00:47:32 +08:00
parent 78332d882b
commit 720de9d73f
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
bot/session_manager.py
View File

@@ -50,6 +50,9 @@ class SessionManager(object):
如果session_id不在sessions中，创建一个新的session并添加到sessions中 如果session_id不在sessions中，创建一个新的session并添加到sessions中
如果system_prompt不会空，会更新session的system_prompt并重置session 如果system_prompt不会空，会更新session的system_prompt并重置session
''' '''
if session_id is None:
return self.sessioncls(session_id, system_prompt, **self.session_args)
if session_id not in self.sessions: if session_id not in self.sessions:
self.sessions[session_id] = self.sessioncls(session_id, system_prompt, **self.session_args) self.sessions[session_id] = self.sessioncls(session_id, system_prompt, **self.session_args)
elif system_prompt is not None: # 如果有新的system_prompt，更新并重置session elif system_prompt is not None: # 如果有新的system_prompt，更新并重置session

4
channel/chat_channel.py
View File

@@ -111,11 +111,11 @@ class ChatChannel(Channel):
img_match_prefix = check_prefix(content, conf().get('image_create_prefix')) img_match_prefix = check_prefix(content, conf().get('image_create_prefix'))
if img_match_prefix: if img_match_prefix:
content = content.replace(img_match_prefix, '', 1).strip() content = content.replace(img_match_prefix, '', 1)
context.type = ContextType.IMAGE_CREATE context.type = ContextType.IMAGE_CREATE
else: else:
context.type = ContextType.TEXT context.type = ContextType.TEXT
context.content = content context.content = content.strip()
if 'desire_rtype' not in context and conf().get('always_reply_voice') and ReplyType.VOICE not in self.NOT_SUPPORT_REPLYTYPE: if 'desire_rtype' not in context and conf().get('always_reply_voice') and ReplyType.VOICE not in self.NOT_SUPPORT_REPLYTYPE:
context['desire_rtype'] = ReplyType.VOICE context['desire_rtype'] = ReplyType.VOICE
elif context.type == ContextType.VOICE: elif context.type == ContextType.VOICE: