gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

better：调整印象生产时间

Browse Source
This commit is contained in:
SengokuCola
2025-06-03 20:33:28 +08:00
parent e2d478441d
commit 12f9ef0527
2 changed files with 53 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
src/person_info/impression_update_task.py
View File

@@ -11,34 +11,31 @@ import time
import random
from collections import defaultdict
logger = get_logger("impression_update_task")
logger = get_logger("relation")
class ImpressionUpdateTask(AsyncTask):
def __init__(self):
super().__init__(
task_name="impression_update",
wait_before_start=2, # 启动后等待10秒
run_interval=20 # 每1分钟运行一次
wait_before_start=10, # 启动后等待10秒
run_interval=600 # 每1分钟运行一次
)
async def run(self):
try:
logger.info("开始执行印象更新任务")
# 获取最近10分钟的消息
current_time = int(time.time())
start_time = current_time - 6000 # 10分钟前
logger.debug(f"获取时间范围: {start_time} -> {current_time}")
start_time = current_time - 600 # 10分钟前
# 获取所有消息
messages = get_raw_msg_by_timestamp(
timestamp_start=start_time,
timestamp_end=current_time,
limit=200
limit=150
)
if not messages:
logger.info("没有找到需要处理的消息")
# logger.info("没有找到需要处理的消息")
return
logger.info(f"获取到 {len(messages)} 条消息")
@@ -52,7 +49,7 @@ class ImpressionUpdateTask(AsyncTask):
# 处理每个聊天组
for chat_id, msgs in chat_messages.items():
logger.info(f"处理聊天组 {chat_id}, 消息数: {len(msgs)}")
# logger.info(f"处理聊天组 {chat_id}, 消息数: {len(msgs)}")
# 获取chat_stream
chat_stream = chat_manager.get_stream(chat_id)
@@ -60,21 +57,17 @@ class ImpressionUpdateTask(AsyncTask):
logger.warning(f"未找到聊天组 {chat_id} 的chat_stream，跳过处理")
continue
# 统计用户发言权重
user_weights = defaultdict(lambda: {"weight": 0, "messages": [], "middle_time": 0})
# 找到bot的消息
bot_messages = [msg for msg in msgs if msg["user_nickname"] == global_config.bot.nickname]
logger.debug(f"找到 {len(bot_messages)} 条bot消息")
for bot_msg in bot_messages:
# 获取bot消息前后的消息
bot_time = bot_msg["time"]
context_messages = [msg for msg in msgs if abs(msg["time"] - bot_time) <= 600] # 前后10分钟
logger.debug(f"Bot消息 {bot_time} 的上下文消息数: {len(context_messages)}")
# 计算权重
for msg in context_messages:
# 统计用户发言权重
user_weights = defaultdict(lambda: {"weight": 0, "messages": [], "middle_time": 0})
if not bot_messages:
# 如果没有bot消息，所有消息权重都为1
logger.info("没有找到bot消息，所有消息权重设为1")
for msg in msgs:
if msg["user_nickname"] == global_config.bot.nickname:
continue
@@ -83,15 +76,35 @@ class ImpressionUpdateTask(AsyncTask):
logger.warning(f"未找到用户 {msg['user_nickname']} 的person_id")
continue
# 在bot消息附近的发言权重加倍
if abs(msg["time"] - bot_time) <= 300: # 前后5分钟
user_weights[person_id]["weight"] += 2
logger.debug(f"用户 {msg['user_nickname']} 在bot消息附近发言，权重+2")
else:
user_weights[person_id]["weight"] += 1
logger.debug(f"用户 {msg['user_nickname']} 发言，权重+1")
user_weights[person_id]["weight"] += 1
user_weights[person_id]["messages"].append(msg)
else:
# 有bot消息时的原有逻辑
for bot_msg in bot_messages:
# 获取bot消息前后的消息
bot_time = bot_msg["time"]
context_messages = [msg for msg in msgs if abs(msg["time"] - bot_time) <= 600] # 前后10分钟
logger.debug(f"Bot消息 {bot_time} 的上下文消息数: {len(context_messages)}")
# 计算权重
for msg in context_messages:
if msg["user_nickname"] == global_config.bot.nickname:
continue
person_id = person_info_manager.get_person_id(msg["chat_info_platform"], msg["user_id"])
if not person_id:
logger.warning(f"未找到用户 {msg['user_nickname']} 的person_id")
continue
# 在bot消息附近的发言权重加倍
if abs(msg["time"] - bot_time) <= 120: # 前后2分钟
user_weights[person_id]["weight"] += 2
logger.debug(f"用户 {msg['user_nickname']} 在bot消息附近发言，权重+2")
else:
user_weights[person_id]["weight"] += 1
logger.debug(f"用户 {msg['user_nickname']} 发言，权重+1")
user_weights[person_id]["messages"].append(msg)
# 计算每个用户的中间时间
for person_id, data in user_weights.items():
@@ -108,7 +121,7 @@ class ImpressionUpdateTask(AsyncTask):
reverse=True
)
logger.info(f"用户权重排序: {[(msg[1]['messages'][0]['user_nickname'], msg[1]['weight']) for msg in sorted_users]}")
logger.debug(f"用户权重排序: {[(msg[1]['messages'][0]['user_nickname'], msg[1]['weight']) for msg in sorted_users]}")
# 随机选择三个用户
selected_users = []
@@ -121,10 +134,10 @@ class ImpressionUpdateTask(AsyncTask):
k=3
)
selected_users = [sorted_users[i] for i in selected_indices]
logger.info(f"随机选择用户: {[msg[1]['messages'][0]['user_nickname'] for msg in selected_users]}")
logger.info(f"开始进一步了解这些用户: {[msg[1]['messages'][0]['user_nickname'] for msg in selected_users]}")
else:
selected_users = sorted_users
logger.info(f"用户数不足3个，全部选择: {[msg[1]['messages'][0]['user_nickname'] for msg in selected_users]}")
logger.info(f"开始进一步了解用户: {[msg[1]['messages'][0]['user_nickname'] for msg in selected_users]}")
# 更新选中用户的印象
for person_id, data in selected_users:
@@ -136,9 +149,8 @@ class ImpressionUpdateTask(AsyncTask):
reason="",
timestamp=data["middle_time"]
)
logger.info(f"用户 {user_nickname} 的印象更新完成")
logger.info("印象更新任务执行完成")
logger.debug("印象更新任务执行完成")
except Exception as e:
logger.exception(f"更新印象任务失败: {str(e)}")