按业务方确定的取向实现:金融场景确定性优先,能溯源到公司资料的才答,答不了就 引导客户拨打客服热线,绝不用模型猜答案。端到端验收 8/8 通过。 新增: - app/service/knowledge_search_service.py:知识检索。未复用记忆的 VectorMemoryAdapter 是因为它只返回 (memory_uuid, score),会丢掉知识块的标题与正文,而客服回答必须能把 原文与出处一起交付。检索失败一律返回 degraded 而不抛异常,由 Agent 走兜底。 - app/service/knowledge_tool.py + app/core/knowledge_contracts.py:只读工具 search_knowledge。 走 ToolExecutor 而不是让 Agent 直接持有检索服务,是为了让白名单、权限、审计、超时 都归基座统一管理;工具只读也符合 ToolRegistry 的硬约束。复用既有权限码 knowledge:reference:read(customer 角色已具备),不新增权限点。 - app/service/agent/implementations/customer_service.py:Agent 本体,刻意保持薄—— 意图分发 + 四条出口(faq/产品/政策直返、闲聊走模型、其余与异常引导人工)。 直接返回知识原文而不经模型改写,答案的字面内容全部来自公司已发布资料。 - tools/publish_customer_service_config.py:发布意图工具白名单。 - tools/customer_service_check.py:端到端验收(8 个用例,含越界请求与知识库外问题)。 装配: - bootstrap 新增 get_knowledge_search_service 工厂,注册 search_knowledge 工具与 customer_service Agent。 - runtime_config_service 新增 load_active_prompt:提示词绑定 release_id,按当前生效 版本读取,未发布时回落代码默认值。闲聊话术因此可审核、可回滚,不必改代码发版。 过程中发现并处理的三个问题: 1. 自造 source_references 被基座合规闸门拒绝。governance.review_output 只接受 「本次召回的记忆」与「本次成功调用的工具」两类引用(用于防止伪造来源), knowledge 类型会被判非法并使整个 run 失败。处理方式是**不放开那道校验**, 而把知识出处(文件标题与内部编号)写进正文,source_references 交给基座自动附加。 2. 发布配置是整版本替换语义:新版本会清空旧版本的全部配置项。若只发客服白名单, 示例 Agent 的 fund_query_demo:fund_quote 会被静默清空。故发布脚本先读取当前生效 版本的全部配置项并原样继承,再追加新增项。 3. 验收脚本自身两处自伤:打印 emoji 触发 GBK UnicodeEncodeError、以及读错结果字段 (RunQueryService 返回的答案键是 content 不是 text)。 已知缺口(未修,已记录): - CoreResult.transfer_required 未持久化:conversation_message 不存该标记, API 读不到"本次是否引导了人工"。当前靠正文里的固定话术判断。 - 知识块引用(source_type=knowledge)尚未启用,需先让 ToolExecutor 把工具返回的 doc_id 登记为本次可引用来源。 验证:ruff 通过、mypy 107 文件无错、unit+contract 447 passed; tools/customer_service_check.py 8/8 通过(含越界请求、投诉、知识库外问题三类 必须引导人工的场景,以及 7 个零容忍负面词零命中)。
50 lines
2.1 KiB
Python
50 lines
2.1 KiB
Python
"""知识检索工具:注册给 Agent 的**只读**公共工具。
|
|
|
|
为什么把知识检索做成工具,而不是让 Agent 直接持有检索服务:走 `ToolExecutor` 就同时
|
|
得到四件由基座保证的事——工具白名单(发布配置可收窄、缺配置即失败关闭)、权限校验、
|
|
调用审计(`agent.tool_executed`)、超时保护。Agent 拿到的 `source_references` 也由基座
|
|
统一附加,业务代码不能伪造来源引用。
|
|
|
|
工具只读是硬约束(`ToolRegistry.register` 会拒绝 `read_only=False`),本工具确实只查库。
|
|
"""
|
|
|
|
from typing import Any
|
|
|
|
from app.core.contracts import RequestContext
|
|
from app.core.knowledge_contracts import KnowledgeSearchInput
|
|
|
|
|
|
async def knowledge_search_tool(
|
|
arguments: KnowledgeSearchInput, context: RequestContext
|
|
) -> dict[str, Any]:
|
|
"""检索三个知识集合,返回命中原文与来源信息。
|
|
|
|
检索链路(向量化 / Milvus)任一环节失败都**不抛异常**,而是以 `degraded=True` 返回:
|
|
客服 Agent 据此走「引导客户致电人工客服」,而不是把基础设施故障暴露成客户可见的错误。
|
|
"""
|
|
del context # 检索本身不区分身份;权限与白名单已在 ToolExecutor 中校验
|
|
# 延迟导入:bootstrap 会导入本模块完成工具注册,模块级导入会形成循环依赖。
|
|
from app.service.agent.bootstrap import get_knowledge_search_service
|
|
|
|
outcome = await get_knowledge_search_service().search(
|
|
arguments.query,
|
|
collections=(arguments.collection,) if arguments.collection else None,
|
|
top_k=arguments.top_k,
|
|
)
|
|
return {
|
|
"degraded": outcome.degraded,
|
|
"reason": outcome.reason,
|
|
"hits": [
|
|
{
|
|
"doc_id": hit.doc_id,
|
|
"title": hit.title,
|
|
"content": hit.content,
|
|
"score": round(hit.score, 4),
|
|
"source_file": hit.source_file,
|
|
"doc_no": hit.doc_no,
|
|
"visibility": hit.visibility,
|
|
}
|
|
for hit in outcome.hits
|
|
],
|
|
}
|