feat：添加群印象
This commit is contained in:
SengokuCola
199
src/person_info/group_relationship_manager.py
Normal file
199
src/person_info/group_relationship_manager.py
Normal file
@@ -0,0 +1,199 @@
|
||||
import time
|
||||
import json
|
||||
import re
|
||||
import asyncio
|
||||
from typing import Any, Optional
|
||||
|
||||
from src.common.logger import get_logger
|
||||
from src.config.config import global_config, model_config
|
||||
from src.llm_models.utils_model import LLMRequest
|
||||
from src.chat.utils.chat_message_builder import (
|
||||
get_raw_msg_by_timestamp_with_chat_inclusive,
|
||||
build_readable_messages,
|
||||
)
|
||||
from src.person_info.group_info import get_group_info_manager
|
||||
from src.plugin_system.apis.message_api import get_message_api
|
||||
from json_repair import repair_json
|
||||
|
||||
|
||||
logger = get_logger("group_relationship_manager")
|
||||
|
||||
|
||||
class GroupRelationshipManager:
|
||||
def __init__(self):
|
||||
self.group_llm = LLMRequest(
|
||||
model_set=model_config.model_task_config.utils, request_type="group.relationship"
|
||||
)
|
||||
self.last_group_impression_time = 0.0
|
||||
self.last_group_impression_message_count = 0
|
||||
|
||||
async def build_relation(self, chat_id: str, platform: str, group_number: str | int) -> None:
|
||||
"""构建群关系，类似 relationship_builder.build_relation() 的调用方式"""
|
||||
current_time = time.time()
|
||||
talk_frequency = global_config.chat.get_current_talk_frequency(chat_id)
|
||||
|
||||
# 计算间隔时间，基于活跃度动态调整：最小10分钟，最大30分钟
|
||||
interval_seconds = max(600, int(1800 / max(0.5, talk_frequency)))
|
||||
|
||||
# 统计新消息数量
|
||||
message_api = get_message_api()
|
||||
new_messages_since_last_impression = message_api.count_new_messages(
|
||||
chat_id=chat_id,
|
||||
start_time=self.last_group_impression_time,
|
||||
end_time=current_time,
|
||||
filter_mai=True,
|
||||
filter_command=True,
|
||||
)
|
||||
|
||||
# 触发条件：时间间隔 OR 消息数量阈值
|
||||
if (current_time - self.last_group_impression_time >= interval_seconds) or \
|
||||
(new_messages_since_last_impression >= 100):
|
||||
logger.info(f"[{chat_id}] 触发群印象构建 (时间间隔: {current_time - self.last_group_impression_time:.0f}s, 消息数: {new_messages_since_last_impression})")
|
||||
|
||||
# 异步执行群印象构建
|
||||
asyncio.create_task(
|
||||
self.build_group_impression(
|
||||
chat_id=chat_id,
|
||||
platform=platform,
|
||||
group_number=group_number,
|
||||
lookback_hours=12,
|
||||
max_messages=300
|
||||
)
|
||||
)
|
||||
|
||||
self.last_group_impression_time = current_time
|
||||
self.last_group_impression_message_count = 0
|
||||
else:
|
||||
# 更新消息计数
|
||||
self.last_group_impression_message_count = new_messages_since_last_impression
|
||||
logger.debug(f"[{chat_id}] 群印象构建等待中 (时间: {current_time - self.last_group_impression_time:.0f}s/{interval_seconds}s, 消息: {new_messages_since_last_impression}/100)")
|
||||
|
||||
async def build_group_impression(
|
||||
self,
|
||||
chat_id: str,
|
||||
platform: str,
|
||||
group_number: str | int,
|
||||
lookback_hours: int = 24,
|
||||
max_messages: int = 300,
|
||||
) -> Optional[str]:
|
||||
"""基于最近聊天记录构建群印象并存储
|
||||
返回生成的topic
|
||||
"""
|
||||
now = time.time()
|
||||
start_ts = now - lookback_hours * 3600
|
||||
|
||||
# 拉取最近消息（包含边界）
|
||||
messages = get_raw_msg_by_timestamp_with_chat_inclusive(chat_id, start_ts, now)
|
||||
if not messages:
|
||||
logger.info(f"[{chat_id}] 无近期消息，跳过群印象构建")
|
||||
return None
|
||||
|
||||
# 限制数量，优先最新
|
||||
messages = sorted(messages, key=lambda m: m.get("time", 0))[-max_messages:]
|
||||
|
||||
# 构建可读文本
|
||||
readable = build_readable_messages(
|
||||
messages=messages, replace_bot_name=True, timestamp_mode="normal_no_YMD", truncate=True
|
||||
)
|
||||
if not readable:
|
||||
logger.info(f"[{chat_id}] 构建可读消息文本为空，跳过")
|
||||
return None
|
||||
|
||||
# 确保群存在
|
||||
group_info_manager = get_group_info_manager()
|
||||
group_id = await group_info_manager.get_or_create_group(platform, group_number)
|
||||
|
||||
group_name = await group_info_manager.get_value(group_id, "group_name") or str(group_number)
|
||||
alias_str = ", ".join(global_config.bot.alias_names)
|
||||
|
||||
prompt = f"""
|
||||
你的名字是{global_config.bot.nickname}，{global_config.bot.nickname}的别名是{alias_str}。
|
||||
你现在在群「{group_name}」（平台：{platform}）中。
|
||||
请你根据以下群内最近的聊天记录，总结这个群给你的印象。
|
||||
|
||||
要求：
|
||||
- 关注群的氛围（友好/活跃/娱乐/学习/严肃等）、常见话题、互动风格、活跃时段或频率、是否有显著文化/梗。
|
||||
- 用白话表达，避免夸张或浮夸的词汇；语气自然、接地气。
|
||||
- 不要暴露任何个人隐私信息。
|
||||
- 请严格按照json格式输出，不要有其他多余内容：
|
||||
{{
|
||||
"impression": "不超过200字的群印象长描述，白话、自然",
|
||||
"topic": "一句话概括群主要聊什么，白话",
|
||||
"style": "一句话描述大家的说话风格，白话"
|
||||
}}
|
||||
|
||||
群内聊天（节选）：
|
||||
{readable}
|
||||
"""
|
||||
# 生成印象
|
||||
content, _ = await self.group_llm.generate_response_async(prompt=prompt)
|
||||
raw_text = (content or "").strip()
|
||||
|
||||
def _strip_code_fences(text: str) -> str:
|
||||
if text.startswith("```") and text.endswith("```"):
|
||||
# 去除首尾围栏
|
||||
return re.sub(r"^```[a-zA-Z0-9_\-]*\n|\n```$", "", text, flags=re.S)
|
||||
# 提取围栏中的主体
|
||||
match = re.search(r"```[a-zA-Z0-9_\-]*\n([\s\S]*?)\n```", text)
|
||||
return match.group(1) if match else text
|
||||
|
||||
parsed_text = _strip_code_fences(raw_text)
|
||||
|
||||
long_impression: str = ""
|
||||
topic_val: Any = ""
|
||||
style_val: Any = ""
|
||||
|
||||
# 参考关系模块：先repair_json再loads，兼容返回列表/字典/字符串
|
||||
try:
|
||||
fixed = repair_json(parsed_text)
|
||||
data = json.loads(fixed) if isinstance(fixed, str) else fixed
|
||||
if isinstance(data, list) and data and isinstance(data[0], dict):
|
||||
data = data[0]
|
||||
if isinstance(data, dict):
|
||||
long_impression = str(data.get("impression") or "").strip()
|
||||
topic_val = data.get("topic", "")
|
||||
style_val = data.get("style", "")
|
||||
else:
|
||||
# 不是字典，直接作为文本
|
||||
text_fallback = str(data)
|
||||
long_impression = text_fallback[:400].strip()
|
||||
topic_val = ""
|
||||
style_val = ""
|
||||
except Exception:
|
||||
long_impression = parsed_text[:400].strip()
|
||||
topic_val = ""
|
||||
style_val = ""
|
||||
|
||||
# 兜底
|
||||
if not long_impression and not topic_val and not style_val:
|
||||
logger.info(f"[{chat_id}] LLM未产生有效群印象，跳过")
|
||||
return None
|
||||
|
||||
# 写入数据库
|
||||
await group_info_manager.update_one_field(group_id, "group_impression", long_impression)
|
||||
# 将 topic/style 写入 group_info JSON
|
||||
try:
|
||||
current_group_info = await group_info_manager.get_value(group_id, "group_info") or {}
|
||||
if not isinstance(current_group_info, dict):
|
||||
current_group_info = {}
|
||||
except Exception:
|
||||
current_group_info = {}
|
||||
if topic_val != "":
|
||||
current_group_info["topic"] = topic_val
|
||||
if style_val != "":
|
||||
current_group_info["style"] = style_val
|
||||
await group_info_manager.update_one_field(group_id, "group_info", current_group_info)
|
||||
await group_info_manager.update_one_field(group_id, "last_active", now)
|
||||
|
||||
logger.info(f"[{chat_id}] 群印象更新完成: topic={topic_val} style={style_val}")
|
||||
return str(topic_val) if topic_val else ""
|
||||
|
||||
|
||||
group_relationship_manager: Optional[GroupRelationshipManager] = None
|
||||
|
||||
|
||||
def get_group_relationship_manager() -> GroupRelationshipManager:
|
||||
global group_relationship_manager
|
||||
if group_relationship_manager is None:
|
||||
group_relationship_manager = GroupRelationshipManager()
|
||||
return group_relationship_manager
|
||||
Reference in New Issue
Block a user