15d436b3a1
- 将 A_Memorix 从旧 submodule / 插件形态迁入主线源码，主体落到 src/A_memorix - 调整主程序接入方式，使 A_Memorix 作为源码内长期记忆子系统运行 - 回收父项目插件体系中针对 A_Memorix 的特判，减少对 plugin 通用层的侵入 - 将长期记忆配置、运行时、自检、导入、调优等能力收口到 memory 路由与主线服务层 - 重做长期记忆控制台与图谱页面，按 MaiBot 现有 dashboard 风格接入 - 补充实体关系图与证据视图双视图能力，支持查看节点、关系、段落及其证据链路 - 新增长期记忆配置编辑器与 memory-api，支持主线内配置管理 - 补齐删除管理能力：删除预览、混合删除、来源批量删除、删除操作恢复 - 优化删除预览与删除操作详情的前端展示，支持分页、检索，并以实体名/关系内容/段落摘要替代单纯 hash 展示 - 修复图谱与控制台相关前端问题，包括证据视图切换、查询触发时机、删除弹层空值保护等 - 新增或更新 A_Memorix 相关测试、WebUI 路由测试、前端 vitest 测试与辅助验证脚本 - 移除旧 plugins/A_memorix、.gitmodules 及相关历史维护文档
125 lines
4.1 KiB
Python
125 lines
4.1 KiB
Python
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from src.chat.brain_chat.PFC import pfc_KnowledgeFetcher as knowledge_module
|
|
from src.services.memory_service import MemoryHit, MemorySearchResult
|
|
|
|
|
|
def test_knowledge_fetcher_resolves_private_memory_context(monkeypatch):
|
|
monkeypatch.setattr(
|
|
knowledge_module,
|
|
"_chat_manager",
|
|
SimpleNamespace(get_session_by_session_id=lambda session_id: SimpleNamespace(platform="qq", user_id="42", group_id="")),
|
|
)
|
|
monkeypatch.setattr(
|
|
knowledge_module,
|
|
"resolve_person_id_for_memory",
|
|
lambda *, person_name, platform, user_id: f"{person_name}:{platform}:{user_id}",
|
|
)
|
|
|
|
fetcher = knowledge_module.KnowledgeFetcher(private_name="Alice", stream_id="stream-1")
|
|
|
|
assert fetcher._resolve_private_memory_context() == {
|
|
"chat_id": "stream-1",
|
|
"person_id": "Alice:qq:42",
|
|
"user_id": "42",
|
|
"group_id": "",
|
|
}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_knowledge_fetcher_memory_get_knowledge_uses_memory_service(monkeypatch):
|
|
monkeypatch.setattr(
|
|
knowledge_module,
|
|
"_chat_manager",
|
|
SimpleNamespace(get_session_by_session_id=lambda session_id: SimpleNamespace(platform="qq", user_id="42", group_id="")),
|
|
)
|
|
monkeypatch.setattr(
|
|
knowledge_module,
|
|
"resolve_person_id_for_memory",
|
|
lambda *, person_name, platform, user_id: f"{person_name}:{platform}:{user_id}",
|
|
)
|
|
|
|
calls = []
|
|
|
|
async def fake_search(query: str, **kwargs):
|
|
calls.append((query, kwargs))
|
|
return MemorySearchResult(summary="", hits=[MemoryHit(content="她喜欢猫", source="person_fact:qq:42")], filtered=False)
|
|
|
|
monkeypatch.setattr(knowledge_module.memory_service, "search", fake_search)
|
|
|
|
fetcher = knowledge_module.KnowledgeFetcher(private_name="Alice", stream_id="stream-1")
|
|
result = await fetcher._memory_get_knowledge("她喜欢什么")
|
|
|
|
assert "1. 她喜欢猫" in result
|
|
assert calls == [
|
|
(
|
|
"她喜欢什么",
|
|
{
|
|
"limit": 5,
|
|
"mode": "search",
|
|
"chat_id": "stream-1",
|
|
"person_id": "Alice:qq:42",
|
|
"user_id": "42",
|
|
"group_id": "",
|
|
"respect_filter": True,
|
|
},
|
|
)
|
|
]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_knowledge_fetcher_falls_back_to_chat_scope_when_person_scope_misses(monkeypatch):
|
|
monkeypatch.setattr(
|
|
knowledge_module,
|
|
"_chat_manager",
|
|
SimpleNamespace(get_session_by_session_id=lambda session_id: SimpleNamespace(platform="qq", user_id="42", group_id="")),
|
|
)
|
|
monkeypatch.setattr(
|
|
knowledge_module,
|
|
"resolve_person_id_for_memory",
|
|
lambda *, person_name, platform, user_id: "person-1",
|
|
)
|
|
|
|
calls = []
|
|
|
|
async def fake_search(query: str, **kwargs):
|
|
calls.append((query, kwargs))
|
|
if kwargs.get("person_id"):
|
|
return MemorySearchResult(summary="", hits=[], filtered=False)
|
|
return MemorySearchResult(summary="", hits=[MemoryHit(content="她计划去杭州音乐节", source="chat_summary:stream-1")], filtered=False)
|
|
|
|
monkeypatch.setattr(knowledge_module.memory_service, "search", fake_search)
|
|
|
|
fetcher = knowledge_module.KnowledgeFetcher(private_name="Alice", stream_id="stream-1")
|
|
result = await fetcher._memory_get_knowledge("Alice 最近在忙什么")
|
|
|
|
assert "杭州音乐节" in result
|
|
assert calls == [
|
|
(
|
|
"Alice 最近在忙什么",
|
|
{
|
|
"limit": 5,
|
|
"mode": "search",
|
|
"chat_id": "stream-1",
|
|
"person_id": "person-1",
|
|
"user_id": "42",
|
|
"group_id": "",
|
|
"respect_filter": True,
|
|
},
|
|
),
|
|
(
|
|
"Alice 最近在忙什么",
|
|
{
|
|
"limit": 5,
|
|
"mode": "search",
|
|
"chat_id": "stream-1",
|
|
"person_id": "",
|
|
"user_id": "42",
|
|
"group_id": "",
|
|
"respect_filter": True,
|
|
},
|
|
),
|
|
]
|