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

increase group_chat_keyword and group_name_keyword_white_list into config.json to support fuzzy matching on group chat.

Browse Source
This commit is contained in:
evolay
2023-02-02 13:56:44 +08:00
parent 8b2a9ee130
commit 2aa48af661
3 changed files with 20 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
channel/wechat/wechat_channel.py
View File

@@ -89,8 +89,8 @@ class WechatChannel(Channel):
content = content_list[1]
config = conf()
match_prefix = msg['IsAt'] or self.check_prefix(origin_content, config.get('group_chat_prefix'))
if (group_name in config.get('group_name_white_list') or 'ALL_GROUP' in config.get('group_name_white_list')) and match_prefix:
match_prefix = msg['IsAt'] or self.check_prefix(origin_content, config.get('group_chat_prefix')) or self.check_contain(origin_content, config.get('group_chat_keyword'))
if (group_name in config.get('group_name_white_list') or 'ALL_GROUP' in config.get('group_name_white_list') or self.fuzzy_match_group_name(group_name, config.get('group_name_keyword_white_list'))) and match_prefix:
img_match_prefix = self.check_prefix(content, conf().get('image_create_prefix'))
if img_match_prefix:
content = content.split(img_match_prefix, 1)[1].strip()
@@ -152,3 +152,15 @@ class WechatChannel(Channel):
if content.startswith(prefix):
return prefix
return None
def check_contain(self, content, keyword_list):
for ky in keyword_list:
if content.find(ky) != -1:
return ky
return None
def fuzzy_match_group_name(self, group_name, keyword_list):
for ky in keyword_list:
if group_name.find(ky) != -1:
return True
return False