103 lines
5.2 KiB
Python
103 lines
5.2 KiB
Python
|
"""记忆召回演示（只读）：**同一个客户，换不同身份去读，看谁能读到、谁读不到。**
|
|||
|
|
|
|||
|
|
## 为什么需要它
|
|||
|
|
|
|||
|
|
"记忆系统的权限"这套东西光讲没用 —— 演示时最好当场看到：
|
|||
|
|
用**客户本人**的身份能读到自己，用**别的客户**身份读到的是空的，
|
|||
|
|
用**有权限且有归属**的风控身份能读到，用**没权限**的运营身份读不到（而且日志会点名原因）。
|
|||
|
|
|
|||
|
|
这个脚本就是干这个的：按平台**真实**的身份解析（`IdentityService`，读的是库里的
|
|||
|
|
角色／权限／客户归属）构造上下文，再走**产线同一条召回链路**
|
|||
|
|
（`PlatformGovernance` + `build_memory_recall_service`，含 Redis 热缓存与 Milvus 语义通道），
|
|||
|
|
逐个身份打印召回结果与降级原因。全部只读，不写任何数据。
|
|||
|
|
|
|||
|
|
用法::
|
|||
|
|
|
|||
|
|
python tools/memory_recall_demo.py # 默认看客户 9001
|
|||
|
|
python tools/memory_recall_demo.py --customer 10001
|
|||
|
|
python tools/memory_recall_demo.py --identity 9002 # 只看某个身份
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import asyncio
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
from app.core.contracts import RequestContext
|
|||
|
|
from app.core.memory_scope import REQUIRED_EMPLOYEE_PERMISSION, customer_memory_scope
|
|||
|
|
from app.service.agent.bootstrap import build_memory_recall_service
|
|||
|
|
from app.service.agent.governance import PlatformGovernance
|
|||
|
|
from app.service.identity_service import IdentityService
|
|||
|
|
|
|||
|
|
if hasattr(sys.stdout, "reconfigure"):
|
|||
|
|
sys.stdout.reconfigure(errors="replace")
|
|||
|
|
|
|||
|
|
#: 演示身份：(显示名, user_id, 说明)。真实的角色/权限/归属都从库里查，不在这里编。
|
|||
|
|
DEMO_IDENTITIES: tuple[tuple[str, str, str], ...] = (
|
|||
|
|
("客户本人（9001 / cust_t）", "9001", "客户身份：只读自己"),
|
|||
|
|
("另一个客户（12001）", "12001", "客户身份：读的是**他自己**名下，不是 9001"),
|
|||
|
|
("风控专员（9002 / risk_t）", "9002", "员工：有 memory:read:customer 且 9001 已分配给他"),
|
|||
|
|
("投顾（9020 / advisor_t）", "9020", "员工：同上，看归属表里有没有"),
|
|||
|
|
("运营（9006 / offsite_t）", "9006", "员工：**没有** memory:read:customer → 失败关闭"),
|
|||
|
|
("管理员（9003 / admin_t）", "9003", "员工：有权限码，但要看归属表"),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _reason(context: RequestContext, has_scope: bool) -> str:
|
|||
|
|
"""把"为什么 0 条"讲成人话（判定本身仍由 `customer_memory_scope` 做，这里只解释）。"""
|
|||
|
|
if not has_scope:
|
|||
|
|
if REQUIRED_EMPLOYEE_PERMISSION not in context.permissions:
|
|||
|
|
return (f"该身份缺少 {REQUIRED_EMPLOYEE_PERMISSION} 能力码 → **失败关闭**"
|
|||
|
|
"（归属关系只解决'谁负责谁'，不构成读他人客户数据的授权）")
|
|||
|
|
return ("该身份在 `sys_customer_assignment` 里**没有生效的归属客户** → **失败关闭**"
|
|||
|
|
"（这不是记忆坏了，是没授权；要读指定客户可先维护分配关系）")
|
|||
|
|
return "可读范围内这些客户确实没有 active 的长期记忆（或都过期了）"
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def show(customer_id: int, only: str | None) -> None:
|
|||
|
|
governance = PlatformGovernance(recall_factory=build_memory_recall_service)
|
|||
|
|
identity = IdentityService()
|
|||
|
|
|
|||
|
|
for label, user_id, note in DEMO_IDENTITIES:
|
|||
|
|
if only and only != user_id:
|
|||
|
|
continue
|
|||
|
|
context = await identity.resolve(
|
|||
|
|
RequestContext(user_id=user_id, trace_id=f"memory-recall-demo-{user_id}")
|
|||
|
|
)
|
|||
|
|
items = await governance.recall(context)
|
|||
|
|
scope = customer_memory_scope(context)
|
|||
|
|
print("=" * 78)
|
|||
|
|
print(f"{label} （{note}）")
|
|||
|
|
print(f" 角色={list(context.roles)} 数据范围={context.data_scope} "
|
|||
|
|
f"归属客户={list(context.customer_ids)}")
|
|||
|
|
print(f" memory:read:customer = "
|
|||
|
|
f"{REQUIRED_EMPLOYEE_PERMISSION in context.permissions}")
|
|||
|
|
print(f" 可读客户范围 = {list(scope) or '（空）'}")
|
|||
|
|
if not items:
|
|||
|
|
print(f" 召回：**0 条** —— 原因：{_reason(context, bool(scope))}")
|
|||
|
|
continue
|
|||
|
|
print(f" 召回：{len(items)} 条")
|
|||
|
|
# `RecalledMemory` 就是**送进模型上下文**的那一份（只有 uuid/客户号/正文），
|
|||
|
|
# 置信度与来源在更下层的 `RecallItem` 里，这里刻意不展开 —— 演示时讲清这一点即可。
|
|||
|
|
for item in items:
|
|||
|
|
print(f" · 客户 {item.customer_id}：{item.content[:60]} "
|
|||
|
|
f"（uuid={item.memory_uuid[:8]}…）")
|
|||
|
|
print("=" * 78)
|
|||
|
|
print(f"提示：这里看的是**召回**（把记忆塞进模型上下文）。"
|
|||
|
|
f"客户 {customer_id} 的记忆本体在 memory_unit 表里，用 "
|
|||
|
|
f"`python tools/probe_memory_state.py {customer_id}` 看。")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> int:
|
|||
|
|
parser = argparse.ArgumentParser(description="记忆召回按身份的演示（只读）")
|
|||
|
|
parser.add_argument("--customer", type=int, default=9001, help="被观察的客户号（默认 9001）")
|
|||
|
|
parser.add_argument("--identity", default=None, help="只看某一个身份（user_id）")
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
asyncio.run(show(args.customer, args.identity))
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
raise SystemExit(main())
|