fix:修复一处小bug,并且把print改为logger,减少review血压

This commit is contained in:
HYY
2025-03-12 22:01:05 +08:00
parent e50c73932b
commit 8ee53fb312
2 changed files with 10 additions and 9 deletions

View File

@@ -135,7 +135,7 @@ class ChatBot:
for pattern in global_config.ban_msgs_regex: for pattern in global_config.ban_msgs_regex:
if re.search(pattern, message.raw_message): if re.search(pattern, message.raw_message):
logger.info( logger.info(
f"[{chat.group_info.group_name if chat.group_info.group_id else '私聊'}]{message.user_nickname}:{message.raw_message}" f"[{chat.group_info.group_name if chat.group_info.group_id else '私聊'}]{userinfo.user_nickname}:{message.raw_message}"
) )
logger.info(f"[正则表达式过滤]消息匹配到{pattern}filtered") logger.info(f"[正则表达式过滤]消息匹配到{pattern}filtered")
return return
@@ -239,10 +239,10 @@ class ChatBot:
is_head=not mark_head, is_head=not mark_head,
is_emoji=False, is_emoji=False,
) )
print(f"bot_message: {bot_message}") logger.debug(f"bot_message: {bot_message}")
if not mark_head: if not mark_head:
mark_head = True mark_head = True
print(f"添加消息到message_set: {bot_message}") logger.debug(f"添加消息到message_set: {bot_message}")
message_set.add_message(bot_message) message_set.add_message(bot_message)
# message_set 可以直接加入 message_manager # message_set 可以直接加入 message_manager

View File

@@ -2,6 +2,7 @@ import asyncio
import random import random
import time import time
from typing import Dict from typing import Dict
from loguru import logger
from .config import global_config from .config import global_config
@@ -27,7 +28,7 @@ class WillingManager:
"""定期衰减回复意愿""" """定期衰减回复意愿"""
while True: while True:
await asyncio.sleep(5) await asyncio.sleep(5)
for chat_id in list(self.chat_reply_willing.keys()): for chat_id in self.chat_reply_willing:
is_high_mode = self.chat_high_willing_mode.get(chat_id, False) is_high_mode = self.chat_high_willing_mode.get(chat_id, False)
if is_high_mode: if is_high_mode:
# 高回复意愿期内轻微衰减 # 高回复意愿期内轻微衰减
@@ -42,7 +43,7 @@ class WillingManager:
current_time = time.time() current_time = time.time()
await asyncio.sleep(10) # 每10秒检查一次 await asyncio.sleep(10) # 每10秒检查一次
for chat_id in list(self.chat_high_willing_mode.keys()): for chat_id in self.chat_high_willing_mode:
last_change_time = self.chat_last_mode_change.get(chat_id, 0) last_change_time = self.chat_last_mode_change.get(chat_id, 0)
is_high_mode = self.chat_high_willing_mode.get(chat_id, False) is_high_mode = self.chat_high_willing_mode.get(chat_id, False)
@@ -146,7 +147,7 @@ class WillingManager:
is_follow_up_question = True is_follow_up_question = True
in_conversation_context = True in_conversation_context = True
self.chat_conversation_context[chat_id] = True self.chat_conversation_context[chat_id] = True
print(f"检测到追问 (同一用户), 提高回复意愿") logger.debug(f"检测到追问 (同一用户), 提高回复意愿")
current_willing += 0.3 current_willing += 0.3
# 特殊情况处理 # 特殊情况处理
@@ -154,11 +155,11 @@ class WillingManager:
current_willing += 0.5 current_willing += 0.5
in_conversation_context = True in_conversation_context = True
self.chat_conversation_context[chat_id] = True self.chat_conversation_context[chat_id] = True
print(f"被提及, 当前意愿: {current_willing}") logger.debug(f"被提及, 当前意愿: {current_willing}")
if is_emoji: if is_emoji:
current_willing *= 0.1 current_willing *= 0.1
print(f"表情包, 当前意愿: {current_willing}") logger.debug(f"表情包, 当前意愿: {current_willing}")
# 根据话题兴趣度适当调整 # 根据话题兴趣度适当调整
if interested_rate > 0.5: if interested_rate > 0.5:
@@ -170,7 +171,7 @@ class WillingManager:
if in_conversation_context: if in_conversation_context:
# 在对话上下文中,降低基础回复概率 # 在对话上下文中,降低基础回复概率
base_probability = 0.5 if is_high_mode else 0.25 base_probability = 0.5 if is_high_mode else 0.25
print(f"处于对话上下文中,基础回复概率: {base_probability}") logger.debug(f"处于对话上下文中,基础回复概率: {base_probability}")
elif is_high_mode: elif is_high_mode:
# 高回复周期4-8句话有50%的概率会回复一次 # 高回复周期4-8句话有50%的概率会回复一次
base_probability = 0.50 if 4 <= msg_count <= 8 else 0.2 base_probability = 0.50 if 4 <= msg_count <= 8 else 0.2