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

补充语音单聊前缀判断过滤 (#661)

Browse Source
This commit is contained in:
zhanws
2023-03-29 01:41:36 +08:00
committed by GitHub
parent fd64f88a7e
commit 26540bfb63
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
channel/wechat/wechat_channel.py
View File

@@ -282,7 +282,7 @@ class WechatChannel(Channel):
if reply.type != ReplyType.ERROR and reply.type != ReplyType.INFO: if reply.type != ReplyType.ERROR and reply.type != ReplyType.INFO:
content = reply.content # 语音转文字后，将文字内容作为新的context content = reply.content # 语音转文字后，将文字内容作为新的context
context.type = ContextType.TEXT context.type = ContextType.TEXT
if context["isgroup"]: if context["isgroup"]: # 群聊
# 校验关键字 # 校验关键字
match_prefix = check_prefix(content, conf().get('group_chat_prefix')) match_prefix = check_prefix(content, conf().get('group_chat_prefix'))
match_contain = check_contain(content, conf().get('group_chat_keyword')) match_contain = check_contain(content, conf().get('group_chat_keyword'))
@@ -293,6 +293,13 @@ class WechatChannel(Channel):
else: else:
logger.info("[WX]receive voice, checkprefix didn't match") logger.info("[WX]receive voice, checkprefix didn't match")
return return
else: # 单聊
match_prefix = check_prefix(content, conf().get('single_chat_prefix'))
if match_prefix: # 判断如果匹配到自定义前缀，则返回过滤掉前缀+空格后的内容
content = content.replace(match_prefix, '', 1).strip()
else:
logger.info("[WX]receive voice, checkprefix didn't match")
return
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: