Files
group_fqcd_jr/app/service/knowledge_tool.py
T
张胜宇 e239eb778b docs: 品牌全量口径统一为「南方基金」+ 作废文档清理
1) 客服 Agent 四份交付文档 + 构建脚手架:品牌由包装占位 XX科技 / 旧名 南方财富
   统一为南方基金(热线 400-889-8899 / 官网 nffund.com),系统名改为「智能服务系统」;
   同步追加 §0.4 修订记录行,工程记录行保留原占位字面以支撑硬编码扫描验收。
2) 开发文档:清理 28 份已作废/残留文档(14 份移出归档 + 14 份仓库副本),
   新增《文档规整方案与开发前待决事项-2026-09-17》。
3) 客服agent 四份交付文档首次纳入本分支。
2026-09-17 15:15:22 +08:00

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
],
}