feat(cross_context): 支持私聊和群聊混合的上下文共享

本次更新扩展了跨上下文共享功能,使其不再局限于群聊之间,现在可以支持群聊与私聊的混合共享。

主要变更:
- 将 `cross_context.groups` 的 `chat_ids` 配置格式从一维字符串列表(仅群号)更新为二维列表 `[["type", "id"], ...]`,其中 `type` 可以是 "group" 或 "private"。
- 更新了 `cross_context_api` 以正确处理新的配置格式,使其能够识别并获取私聊和群聊的消息历史。
- 相应地更新了配置文件模板 `bot_config_template.toml` 以反映新的配置格式和功能。

BREAKING CHANGE: `cross_context.groups` 的 `chat_ids` 配置格式已更改。旧的一维群号列表格式不再受支持,用户需要按照 `[["type", "id"], ...]` 的新格式更新其配置文件才能使用此功能。
This commit is contained in:
minecraft1024a
2025-09-05 19:30:06 +08:00
parent 5bae2fa8f8
commit e3d1209d14
3 changed files with 55 additions and 38 deletions

View File

@@ -652,7 +652,9 @@ class ContextGroup(ValidatedConfigBase):
"""上下文共享组配置"""
name: str = Field(..., description="共享组的名称")
chat_ids: List[str] = Field(..., description="属于该组的聊天ID列表")
chat_ids: List[List[str]] = Field(
..., description='属于该组的聊天ID列表格式为 [["type", "chat_id"], ...],例如 [["group", "123456"], ["private", "789012"]]'
)
class CrossContextConfig(ValidatedConfigBase):

View File

@@ -16,34 +16,45 @@ from src.chat.message_receive.chat_stream import get_chat_manager, ChatStream
logger = get_logger("cross_context_api")
def get_context_groups(chat_id: str) -> Optional[List[str]]:
def get_context_groups(chat_id: str) -> Optional[List[List[str]]]:
"""
获取当前聊所在的共享组的其他聊ID
获取当前聊所在的共享组的其他聊ID
"""
current_stream = get_chat_manager().get_stream(chat_id)
if not current_stream or not current_stream.group_info:
if not current_stream:
return None
try:
current_chat_raw_id = current_stream.group_info.group_id
except Exception as e:
logger.error(f"获取群聊ID失败: {e}")
return None
is_group = current_stream.group_info is not None
current_chat_raw_id = (
current_stream.group_info.group_id if is_group else current_stream.user_info.user_id
)
current_type = "group" if is_group else "private"
for group in global_config.cross_context.groups:
if str(current_chat_raw_id) in group.chat_ids:
return [chat_id for chat_id in group.chat_ids if chat_id != str(current_chat_raw_id)]
# 检查当前聊天的ID和类型是否在组的chat_ids
if [current_type, str(current_chat_raw_id)] in group.chat_ids:
# 返回组内其他聊天的 [type, id] 列表
return [
chat_info
for chat_info in group.chat_ids
if chat_info != [current_type, str(current_chat_raw_id)]
]
return None
async def build_cross_context_normal(chat_stream: ChatStream, other_chat_raw_ids: List[str]) -> str:
async def build_cross_context_normal(
chat_stream: ChatStream, other_chat_infos: List[List[str]]
) -> str:
"""
构建跨群聊上下文 (Normal模式)
构建跨群聊/私聊上下文 (Normal模式)
"""
cross_context_messages = []
for chat_raw_id in other_chat_raw_ids:
stream_id = get_chat_manager().get_stream_id(chat_stream.platform, chat_raw_id, is_group=True)
for chat_type, chat_raw_id in other_chat_infos:
is_group = chat_type == "group"
stream_id = get_chat_manager().get_stream_id(
chat_stream.platform, chat_raw_id, is_group=is_group
)
if not stream_id:
continue
@@ -54,33 +65,38 @@ async def build_cross_context_normal(chat_stream: ChatStream, other_chat_raw_ids
limit=5, # 可配置
)
if messages:
chat_name = get_chat_manager().get_stream_name(stream_id) or stream_id
formatted_messages, _ = build_readable_messages_with_id(messages, timestamp_mode="relative")
chat_name = get_chat_manager().get_stream_name(stream_id) or chat_raw_id
formatted_messages, _ = build_readable_messages_with_id(
messages, timestamp_mode="relative"
)
cross_context_messages.append(f'[以下是来自"{chat_name}"的近期消息]\n{formatted_messages}')
except Exception as e:
logger.error(f"获取{chat_raw_id}的消息失败: {e}")
logger.error(f"获取聊{chat_raw_id} 的消息失败: {e}")
continue
if not cross_context_messages:
return ""
return "# 跨上下文参考\n" + "\n\n".join(cross_context_messages) + "\n"
return "# 跨上下文参考\n" + "\n\n".join(cross_context_messages) + "\n"
async def build_cross_context_s4u(
chat_stream: ChatStream, other_chat_raw_ids: List[str], target_user_info: Optional[Dict[str, Any]]
chat_stream: ChatStream,
other_chat_infos: List[List[str]],
target_user_info: Optional[Dict[str, Any]],
) -> str:
"""
构建跨群聊上下文 (S4U模式)
构建跨群聊/私聊上下文 (S4U模式)
"""
cross_context_messages = []
if target_user_info:
user_id = target_user_info.get("user_id")
if user_id:
for chat_raw_id in other_chat_raw_ids:
for chat_type, chat_raw_id in other_chat_infos:
is_group = chat_type == "group"
stream_id = get_chat_manager().get_stream_id(
chat_stream.platform, chat_raw_id, is_group=True
chat_stream.platform, chat_raw_id, is_group=is_group
)
if not stream_id:
continue
@@ -91,12 +107,10 @@ async def build_cross_context_s4u(
timestamp=time.time(),
limit=20, # 获取更多消息以供筛选
)
user_messages = [msg for msg in messages if msg.get("user_id") == user_id][
-5:
] # 筛选并取最近5条
user_messages = [msg for msg in messages if msg.get("user_id") == user_id][-5:]
if user_messages:
chat_name = get_chat_manager().get_stream_name(stream_id) or stream_id
chat_name = get_chat_manager().get_stream_name(stream_id) or chat_raw_id
user_name = (
target_user_info.get("person_name")
or target_user_info.get("user_nickname")
@@ -109,10 +123,10 @@ async def build_cross_context_s4u(
f'[以下是"{user_name}""{chat_name}"的近期发言]\n{formatted_messages}'
)
except Exception as e:
logger.error(f"获取用户{user_id}在群聊{chat_raw_id}的消息失败: {e}")
logger.error(f"获取用户 {user_id} 在聊天 {chat_raw_id} 的消息失败: {e}")
continue
if not cross_context_messages:
return ""
return "# 跨上下文参考\n" + "\n\n".join(cross_context_messages) + "\n"
return "# 跨上下文参考\n" + "\n\n".join(cross_context_messages) + "\n"

View File

@@ -473,19 +473,20 @@ pre_sleep_notification_groups = []
# 用于生成睡前消息的提示。AI会根据这个提示生成一句晚安问候。
pre_sleep_prompt = "我准备睡觉了,请生成一句简短自然的晚安问候。"
[cross_context] # 跨群聊上下文共享配置
[cross_context] # 跨群聊/私聊上下文共享配置
# 这是总开关,用于一键启用或禁用此功能
enable = false
enable = true
# 在这里定义您的“共享组”
# 只有在同一个组内的聊才会共享上下文
# 注意:这里的chat_ids需要填写群号
# 只有在同一个组内的聊才会共享上下文
# 格式:chat_ids = [["type", "id"], ["type", "id"], ...]
# type 可选 "group" 或 "private"
[[cross_context.groups]]
name = "项目A技术讨论组"
chat_ids = [
"111111", # 假设这是“开发群”的ID
"222222" # 假设这是“产品群”的ID
["group", "169850076"], # 假设这是“开发群”的ID
["group", "1025509724"], # 假设这是“产品群”的ID
["private", "123456789"] # 假设这是某个用户的私聊
]
[maizone_intercom]
# QQ空间互通组配置
# 启用后,发布说说时会读取指定互通组的上下文
@@ -495,6 +496,6 @@ enable = false
[[maizone_intercom.groups]]
name = "Maizone默认互通组"
chat_ids = [
"111111", # 示例群聊1
"222222" # 示例聊2
["group", "111111"], # 示例群聊1
["private", "222222"] # 示例聊2
]