xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-07 14:26:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: replace message_chain.has usage

Browse Source
This commit is contained in:
Junyan Qin
2025-08-25 23:22:36 +08:00
parent ea6ce2f552
commit 56183867a7
2 changed files with 27 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
pkg/pipeline/resprule/rules/atbot.py
View File

@@ -16,17 +16,26 @@ class AtBotRule(rule_model.GroupRespondRule):
rule_dict: dict,
query: pipeline_query.Query,
) -> entities.RuleJudgeResult:
if message_chain.has(platform_message.At(query.adapter.bot_account_id)) and rule_dict['at']:
message_chain.remove(platform_message.At(query.adapter.bot_account_id))
def remove_at(message_chain: platform_message.MessageChain):
for component in message_chain.root:
if isinstance(component, platform_message.At) and component.target == query.adapter.bot_account_id:
message_chain.remove(component)
break
if message_chain.has(
platform_message.At(query.adapter.bot_account_id)
): # 回复消息时会at两次，检查并删除重复的
message_chain.remove(platform_message.At(query.adapter.bot_account_id))
remove_at(message_chain)
remove_at(message_chain) # 回复消息时会at两次，检查并删除重复的
return entities.RuleJudgeResult(
matching=True,
replacement=message_chain,
)
# if message_chain.has(platform_message.At(query.adapter.bot_account_id)) and rule_dict['at']:
# message_chain.remove(platform_message.At(query.adapter.bot_account_id))
# if message_chain.has(
# platform_message.At(query.adapter.bot_account_id)
# ): # 回复消息时会at两次，检查并删除重复的
# message_chain.remove(platform_message.At(query.adapter.bot_account_id))
# return entities.RuleJudgeResult(
# matching=True,
# replacement=message_chain,
# )
return entities.RuleJudgeResult(matching=False, replacement=message_chain)

9
pkg/platform/sources/discord.py
View File

@@ -1034,7 +1034,14 @@ class DiscordAdapter(abstract_platform_adapter.AbstractMessagePlatformAdapter):
if quote_origin:
args['reference'] = message_source.source_platform_object
if message.has(platform_message.At):
has_at = False
for component in message.root:
if isinstance(component, platform_message.At):
has_at = True
break
if has_at:
args['mention_author'] = True
await message_source.source_platform_object.channel.send(**args)