xuthus/LittlePaimon
1
0
Fork 0
mirror of https://github.com/xuthus83/LittlePaimon.git synced 2025-04-12 23:29:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 群聊学习修复连续主动发言概率失效的问题，延长主动发言判断间隔

Browse Source
This commit is contained in:
CMHopeSunshine 2023-01-14 23:51:51 +08:00
parent 1dc87aab97
commit adbfe9436a
2 changed files with 43 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
LittlePaimon/plugins/Learning_Chat/__init__.py
View File

@ -60,7 +60,7 @@ async def _(event: GroupMessageEvent, answers=Arg('answers')):
logger.info('群聊学习', f'{NICKNAME}向群<m>{event.group_id}</m>的回复<m>"{answer}"</m>发送<r>失败，可能处于风控中</r>') logger.info('群聊学习', f'{NICKNAME}向群<m>{event.group_id}</m>的回复<m>"{answer}"</m>发送<r>失败，可能处于风控中</r>')
@scheduler.scheduled_job('interval', minutes=1, misfire_grace_time=5) @scheduler.scheduled_job('interval', minutes=3, misfire_grace_time=5)
async def speak_up(): async def speak_up():
if not config_manager.config.total_enable: if not config_manager.config.total_enable:
return return

20
LittlePaimon/plugins/Learning_Chat/handler.py
View File

@ -389,7 +389,8 @@ class LearningChat:
# 如果最后一条消息是自己发的，则不主动发言 # 如果最后一条消息是自己发的，则不主动发言
if last_reply := await ChatMessage.filter(group_id=group_id, user_id=self_id).first(): if last_reply := await ChatMessage.filter(group_id=group_id, user_id=self_id).first():
if last_reply.time >= messages[0].time: if last_reply.time >= messages[0].time:
logger.debug('群聊学习', f'主动发言：群<m>{group_id}</m>最后一条消息是{NICKNAME}发的{last_reply.message}，不发言') logger.debug('群聊学习',
f'主动发言：群<m>{group_id}</m>最后一条消息是{NICKNAME}发的{last_reply.message}，不发言')
continue continue
elif cur_time - last_reply.time < config.speak_min_interval: elif cur_time - last_reply.time < config.speak_min_interval:
logger.debug('群聊学习', f'主动发言：群<m>{group_id}</m>上次主动发言时间小于主动发言最小间隔，不发言') logger.debug('群聊学习', f'主动发言：群<m>{group_id}</m>上次主动发言时间小于主动发言最小间隔，不发言')
@ -401,7 +402,8 @@ class LearningChat:
silent_time = cur_time - messages[0].time silent_time = cur_time - messages[0].time
threshold = avg_interval * config.speak_threshold threshold = avg_interval * config.speak_threshold
if silent_time < threshold: if silent_time < threshold:
logger.debug('群聊学习', f'主动发言：群<m>{group_id}</m>已沉默时间({silent_time})小于阈值({int(threshold)})，不发言') logger.debug('群聊学习',
f'主动发言：群<m>{group_id}</m>已沉默时间({silent_time})小于阈值({int(threshold)})，不发言')
continue continue
if contexts := await ChatContext.filter(count__gte=config.answer_threshold).all(): if contexts := await ChatContext.filter(count__gte=config.answer_threshold).all():
@ -411,12 +413,13 @@ class LearningChat:
random.shuffle(contexts) random.shuffle(contexts)
for context in contexts: for context in contexts:
if (not speak_list or random.random() < config.speak_continuously_probability) and len( if (not speak_list or random.random() < config.speak_continuously_probability) and len(
speak_list) < config.speak_continuously_max_len and ( speak_list) < config.speak_continuously_max_len:
answers := await ChatAnswer.filter(context=context, if answers := await ChatAnswer.filter(context=context,
group_id=group_id, group_id=group_id,
count__gte=config.answer_threshold)): count__gte=config.answer_threshold):
answer = random.choices(answers, answer = random.choices(answers,
weights=[answer.count + 1 if answer.time >= today_time else answer.count weights=[
answer.count + 1 if answer.time >= today_time else answer.count
for answer in answers])[0] for answer in answers])[0]
message = random.choice(answer.messages) message = random.choice(answer.messages)
if len(message) < 2: if len(message) < 2:
@ -429,7 +432,8 @@ class LearningChat:
follow_answers = answer follow_answers = answer
while random.random() < config.speak_continuously_probability and len( while random.random() < config.speak_continuously_probability and len(
speak_list) < config.speak_continuously_max_len: speak_list) < config.speak_continuously_max_len:
if (follow_context := await ChatContext.filter(keywords=follow_answers.keywords).first()) and ( if (follow_context := await ChatContext.filter(
keywords=follow_answers.keywords).first()) and (
follow_answers := await ChatAnswer.filter( follow_answers := await ChatAnswer.filter(
group_id=group_id, group_id=group_id,
context=follow_context, context=follow_context,
@ -446,7 +450,7 @@ class LearningChat:
speak_list.append(message) speak_list.append(message)
else: else:
break break
elif len(speak_list) >= config.speak_continuously_max_len: else:
break break
if speak_list: if speak_list:
if random.random() < config.speak_poke_probability: if random.random() < config.speak_poke_probability: