freedakgmail
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
"""NLQ 结构化检索集成测试（需 PostgreSQL）。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from app.engines import scan
|
|
from app.nlq import service as nlq
|
|
from app.scenarios.split_contract import ContractRecord
|
|
|
|
|
|
def _seed(session):
|
|
contracts = [ContractRecord(f"C{i}", f"CUST{i}", 850000) for i in range(8)]
|
|
scan.run_split_contract_scan(
|
|
session, contracts, approval_threshold=1_000_000, shared_controller=True
|
|
)
|
|
session.flush()
|
|
|
|
|
|
def test_nlq_retrieves_split_clues(session):
|
|
_seed(session)
|
|
ans = nlq.ask("列出高置信的政企拆单线索", session=session)
|
|
assert ans.provider == "datahub"
|
|
assert ans.egress is False
|
|
assert "政企拆单" in ans.answer
|
|
assert "共检索到" in ans.answer
|
|
|
|
|
|
def test_nlq_no_match(session):
|
|
ans = nlq.ask("列出养卡骗补线索", session=session)
|
|
assert ans.egress is False
|
|
assert "未检索到" in ans.answer or "共检索到" in ans.answer
|
|
|
|
|
|
def test_nlq_open_question_falls_back_to_llm(session):
|
|
# 不含检索关键词 → 走 LLM（mock）
|
|
ans = nlq.ask("你好，请介绍一下你的能力", session=session)
|
|
assert ans.provider in ("mock", "datahub")
|
|
assert ans.egress is False
|