liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化标签去除，解决搜索的grok标签问题

Browse Source
This commit is contained in:
liuwei
2026-02-05 10:58:25 +08:00
parent 3723434bd6
commit 99d3128802
2 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
plugins/dify/main.py
View File

@@ -18,7 +18,7 @@ from utils.revoke.message_auto_revoke import MessageAutoRevoke
from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
from utils.decorator.points_decorator import plugin_points_cost from utils.decorator.points_decorator import plugin_points_cost
from utils.media_downloader import MediaDownloader from utils.media_downloader import MediaDownloader
from utils.string_utils import remove_trailing_content from utils.string_utils import remove_trailing_content, remove_grok_render_tags
from wechat_ipad import WechatAPIClient from wechat_ipad import WechatAPIClient
import aiohttp import aiohttp
@@ -181,6 +181,7 @@ class DifyPlugin(MessagePluginInterface):
# 去除广告内容 # 去除广告内容
response = remove_trailing_content(response) response = remove_trailing_content(response)
response = remove_grok_render_tags(response)
self.LOG.debug(f"处理后的响应: {response}") self.LOG.debug(f"处理后的响应: {response}")
# 发送回复 # 发送回复

9
utils/string_utils.py
View File

@@ -13,3 +13,12 @@ def remove_trailing_content(text, delimiter='---'):
if index != -1: if index != -1:
return text[:index].strip() return text[:index].strip()
return text return text
import re
def remove_grok_render_tags(text: str) -> str:
if not text:
return text
cleaned = re.sub(r'<\s*grok:[^>]+?>[\s\S]*?<\s*/\s*grok:[^>]+?>', '', text, flags=re.IGNORECASE)
cleaned = re.sub(r'<\s*grok:[^>]+?/?>', '', cleaned, flags=re.IGNORECASE)
return cleaned