72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
"""知识库对话 Tool（T21-5 · FLOW §2「milvus_tool：产品规则 RAG + source_refs」）。
|
||
|
||
与 core_tools / risk.chat_tools 同构的 Tool 定义层：由 tool_service.run_tool
|
||
经统一注册表分发。search_knowledge 是公开知识检索（产品手册/交易规则）——
|
||
**不做客户归属校验**（ToolSpec.skip_access_check=True）：
|
||
|
||
- 知识内容为公开产品信息，无客户数据，归属校验无对象；
|
||
- 开放范围（customer + advisor，用户拍板 2026-09-07）由意图层控制：
|
||
_INTENT_KEYWORDS 仅 customer/advisor 组配置 kb 关键词，risk/analyst
|
||
不命中 → Tool 不会被触发；即使被直接调用，返回的也只是公开知识。
|
||
|
||
func 签名对齐 runner 约定 ``func(customer_id, core_ro, risk_repo, **params)``；
|
||
query 为唯一业务参数（tool_node 从 user_message 构造 tool_input 注入，
|
||
一期不来自 LLM 输出）。RAG 异常（EmbeddingError 等）原样上抛 → run_tool
|
||
统一转 TOOL_ERROR 留痕，不静默降级空结果（防 LLM 编造回答）。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Any, Callable
|
||
|
||
from app.service import rag_service
|
||
|
||
# 对话 Tool 检索条数（与 rag_service.DEFAULT_TOP_K 一致；独立常量便于对话口径单独调整）
|
||
KB_TOP_K = 3
|
||
|
||
|
||
def search_knowledge(
|
||
query: str = "",
|
||
customer_id: str = "",
|
||
core_ro=None,
|
||
risk_repo=None,
|
||
) -> dict[str, Any]:
|
||
"""产品知识检索（TopK chunks + 溯源清单）。
|
||
|
||
customer_id/core_ro/risk_repo 为 runner 恒传参数，本 Tool 不使用
|
||
（公开知识，无归属语义）；保留形参以满足统一签名。
|
||
"""
|
||
out = rag_service.search_knowledge(query, top_k=KB_TOP_K)
|
||
return {
|
||
"hit_count": len(out["results"]),
|
||
"results": out["results"],
|
||
"source_refs": out["source_refs"],
|
||
}
|
||
|
||
|
||
class KBToolSpec(dict):
|
||
"""知识库 Tool 注册表条目（与 core_tools.ToolSpec / RiskToolSpec 同构；
|
||
新增 skip_access_check：True → run_tool 跳过客户归属校验）。"""
|
||
|
||
|
||
KB_TOOL_REGISTRY: dict[str, KBToolSpec] = {
|
||
"search_knowledge": KBToolSpec(
|
||
func=search_knowledge,
|
||
description="检索产品知识库（基金产品手册、费率、申赎规则、风险说明；返回结果附溯源）",
|
||
requires_customer=False,
|
||
param_whitelist=("query",),
|
||
int_bounds={},
|
||
skip_access_check=True,
|
||
),
|
||
}
|
||
|
||
|
||
def get_kb_tool(name: str) -> KBToolSpec | None:
|
||
"""白名单查找（未知 Tool 返回 None）。"""
|
||
return KB_TOOL_REGISTRY.get(name)
|
||
|
||
|
||
def kb_tool_func(name: str) -> Callable[..., dict[str, Any]] | None:
|
||
spec = KB_TOOL_REGISTRY.get(name)
|
||
return spec["func"] if spec else None
|