张胜宇
e239eb778b
1) 客服 Agent 四份交付文档 + 构建脚手架：品牌由包装占位 XX科技 / 旧名 南方财富 统一为南方基金（热线 400-889-8899 / 官网 nffund.com），系统名改为「智能服务系统」； 同步追加 §0.4 修订记录行，工程记录行保留原占位字面以支撑硬编码扫描验收。 2) 开发文档：清理 28 份已作废/残留文档（14 份移出归档 + 14 份仓库副本）， 新增《文档规整方案与开发前待决事项-2026-09-17》。 3) 客服agent 四份交付文档首次纳入本分支。
71 lines
3.8 KiB
Python
71 lines
3.8 KiB
Python
from datetime import UTC, datetime
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
from sqlalchemy import delete, text
|
|
|
|
from app.core.contracts import AgentDefinition, AgentRequest, CoreResult, RequestContext
|
|
from app.infrastructure.db import SessionFactory
|
|
from app.model.memory import MemoryUnit
|
|
from app.service.agent.base import BaseAgent
|
|
from app.service.agent.factory import AgentFactory
|
|
|
|
# 依赖真实 MySQL：必须打 integration marker，否则按 marker 过滤时会漏测这批用例。
|
|
pytestmark = pytest.mark.integration
|
|
|
|
|
|
async def test_real_memory_and_reviewed_negative_rule_are_used():
|
|
customer_id = uuid4().int % 10**15 + 10**15
|
|
agent_type = f"test_{uuid4().hex[:20]}"
|
|
memory_uuid = str(uuid4())
|
|
rule_code = f"test-{uuid4()}"
|
|
now = datetime.now(UTC).replace(tzinfo=None)
|
|
async with SessionFactory() as session, session.begin():
|
|
for user_id in (customer_id, customer_id + 1):
|
|
await session.execute(text("""
|
|
INSERT INTO sys_user
|
|
(id,user_no,username,password_hash,user_type,professional_investor_status,
|
|
fund_account_status,status,created_at,updated_at)
|
|
VALUES (:id,:name,:name,'test-only','员工','未申请','未开户','正常',UTC_TIMESTAMP(),
|
|
UTC_TIMESTAMP())
|
|
"""), {"id": user_id, "name": f"gov-{user_id}"})
|
|
session.add(MemoryUnit(
|
|
memory_uuid=memory_uuid, customer_id=customer_id, memory_key="test_preference",
|
|
content="测试低风险偏好", memory_type="fact", source_type="conversation",
|
|
source_confidence=0.8, confidence=0.8, status="active", valid_from=now,
|
|
version=1, created_at=now, updated_at=now,
|
|
))
|
|
await session.execute(text("""
|
|
INSERT INTO agent_negative_word
|
|
(rule_code,word_pattern,match_type,category,severity,applicable_agents,status,
|
|
created_by,reviewer_id,reviewed_at)
|
|
VALUES (:code,'测试禁止词','contains','test','block',JSON_ARRAY(:agent),'active',:c,:r,
|
|
UTC_TIMESTAMP())
|
|
"""), {"code": rule_code, "agent": agent_type, "c": customer_id, "r": customer_id + 1})
|
|
try:
|
|
class TestAgent(BaseAgent):
|
|
async def handle(self, request, context):
|
|
assert self.config is not None and self.config.config_version
|
|
assert self.memories[0].memory_uuid == memory_uuid
|
|
assert self.memories[0].customer_id == str(customer_id)
|
|
return CoreResult(text="测试禁止词")
|
|
|
|
definition = AgentDefinition(agent_type=agent_type, version="1",
|
|
allowed_roles=("customer",), allowed_portals=("api",))
|
|
factory = AgentFactory()
|
|
factory.register(definition, lambda _: TestAgent(definition))
|
|
context = RequestContext(user_id=str(customer_id), trace_id="test",
|
|
roles=("customer",), permissions=("agent:run",))
|
|
result = [event async for event in factory.create(agent_type, context).execute(
|
|
AgentRequest(agent_type=agent_type, message="test", session_id="test",
|
|
idempotency_key=str(uuid4())), context, "test-run")]
|
|
assert result[-1].payload["result"]["result"]["transfer_required"] is True
|
|
assert "测试禁止词" not in result[-1].payload["result"]["result"]["text"]
|
|
finally:
|
|
async with SessionFactory() as session, session.begin():
|
|
await session.execute(delete(MemoryUnit).where(MemoryUnit.memory_uuid == memory_uuid))
|
|
await session.execute(text("DELETE FROM agent_negative_word WHERE rule_code=:code"),
|
|
{"code": rule_code})
|
|
await session.execute(text("DELETE FROM sys_user WHERE id IN (:c,:r)"),
|
|
{"c": customer_id, "r": customer_id + 1})
|