129 lines
4.3 KiB
Python
129 lines
4.3 KiB
Python
"""L0 基础画像层的 client_meta 透传。
|
||
|
||
线上实测发现：一次写入产出的多条记忆里，l2_fact 能带回 meta，
|
||
l0_basic_info 拿不到——画像走独立的 upsert 路径，透传时漏了。
|
||
这组测试锁住修复后的行为。
|
||
"""
|
||
import asyncio
|
||
|
||
from hy_memory.agent.tools.basic_profile import upsert_basic_profile
|
||
from hy_memory.models.memory import MemoryLayer, MemoryStatus
|
||
|
||
|
||
class FakeVectorStore:
|
||
def __init__(self, nodes=None):
|
||
self.nodes = list(nodes or [])
|
||
self.upserted = []
|
||
self.payload_updates = []
|
||
|
||
async def list_by_user(self, **kwargs):
|
||
return list(self.nodes)
|
||
|
||
async def get_by_id(self, node_id):
|
||
return next((n for n in self.nodes if n.node_id == node_id), None)
|
||
|
||
async def upsert(self, node):
|
||
self.upserted.append(node)
|
||
self.nodes.append(node)
|
||
|
||
async def update_payload(self, node_id, updates):
|
||
self.payload_updates.append((node_id, updates))
|
||
|
||
|
||
class FakeEmbedService:
|
||
async def embed_queued(self, text):
|
||
return [0.1] * 8
|
||
|
||
|
||
FIELDS = ["name", "age", "location"]
|
||
|
||
|
||
def upsert(store, kv, meta=None, user="u1"):
|
||
return asyncio.run(
|
||
upsert_basic_profile(
|
||
user_id=user,
|
||
kv=kv,
|
||
vector_store=store,
|
||
embed_service=FakeEmbedService(),
|
||
allowed_fields=FIELDS,
|
||
client_meta=meta,
|
||
)
|
||
)
|
||
|
||
|
||
class TestBasicProfileClientMeta:
|
||
def test_meta_attached_to_new_node(self):
|
||
store = FakeVectorStore()
|
||
meta = {"source": "codebuddy", "trace_id": "t-1"}
|
||
|
||
r = upsert(store, {"name": "张伟"}, meta=meta)
|
||
|
||
assert r.success and r.changed
|
||
assert len(store.upserted) == 1
|
||
node = store.upserted[0]
|
||
assert node.layer == MemoryLayer.L0_BASIC_INFO
|
||
assert node.custom["meta"] == [meta]
|
||
|
||
def test_basic_info_kv_still_stored_alongside_meta(self):
|
||
"""meta 不能挤掉原有的 basic_info_kv——后者是画像装配的依据。"""
|
||
store = FakeVectorStore()
|
||
|
||
upsert(store, {"name": "张伟", "age": 35}, meta={"k": "v"})
|
||
|
||
custom = store.upserted[0].custom
|
||
assert custom["basic_info_kv"] == {"name": "张伟", "age": 35}
|
||
assert custom["meta"] == [{"k": "v"}]
|
||
|
||
def test_no_meta_key_when_not_provided(self):
|
||
"""不传 meta 时不应凭空造出该键，否则旧数据形态被改。"""
|
||
store = FakeVectorStore()
|
||
|
||
upsert(store, {"name": "张伟"})
|
||
|
||
assert "meta" not in store.upserted[0].custom
|
||
|
||
def test_empty_meta_treated_as_absent(self):
|
||
store = FakeVectorStore()
|
||
|
||
upsert(store, {"name": "张伟"}, meta={})
|
||
|
||
assert "meta" not in store.upserted[0].custom
|
||
|
||
def test_meta_is_copied_not_referenced(self):
|
||
"""存进去后调用方再改自己的 dict，不应影响已落库的内容。"""
|
||
store = FakeVectorStore()
|
||
meta = {"source": "a"}
|
||
|
||
upsert(store, {"name": "张伟"}, meta=meta)
|
||
meta["source"] = "mutated"
|
||
|
||
assert store.upserted[0].custom["meta"] == [{"source": "a"}]
|
||
|
||
def test_new_version_carries_only_current_meta(self):
|
||
"""L0 每次变更新建节点、旧节点转 SUPERSEDED，与 SUPERSEDE 同构：
|
||
新节点只带本次 meta，不继承旧链，避免长链上无限累积。"""
|
||
store = FakeVectorStore()
|
||
|
||
upsert(store, {"name": "张伟"}, meta={"round": 1})
|
||
upsert(store, {"name": "张伟", "age": 35}, meta={"round": 2})
|
||
|
||
assert len(store.upserted) == 2
|
||
assert store.upserted[1].custom["meta"] == [{"round": 2}]
|
||
|
||
# 旧节点被标记取代，其 meta 原样留在链上
|
||
assert store.payload_updates
|
||
old_id, updates = store.payload_updates[-1]
|
||
assert old_id == store.upserted[0].node_id
|
||
assert updates["status"] == MemoryStatus.SUPERSEDED.value
|
||
assert store.upserted[0].custom["meta"] == [{"round": 1}]
|
||
|
||
def test_no_diff_writes_nothing_even_with_meta(self):
|
||
"""内容无变化时本就不落新版本，带了 meta 也不应因此凭空建节点。"""
|
||
store = FakeVectorStore()
|
||
|
||
upsert(store, {"name": "张伟"}, meta={"round": 1})
|
||
r = upsert(store, {"name": "张伟"}, meta={"round": 2})
|
||
|
||
assert r.success and not r.changed
|
||
assert len(store.upserted) == 1
|