lzf_0626
84bf51c0d0
## 现象
通用自然语言查询里问「查询15911的净值」，返回的是**别的产品**的数据：
```
问：查询15911的净值 → 返回 511810 货币ETF南方 的净值
问：查询基金代码15911最近30天净值 → 同上
```
而且 `status = success`、`message = 查询成功` —— **不报错，只是数据是错的**。
## 根因
`financial_nl2sql_service._filters()` 只认 6 位数字：
```python
re.findall(r"(?<!\d)\d{6}(?!\d)", question) # 15911 只有 5 位 ⇒ 被静默丢弃
```
过滤条件为空 ⇒ 生成的 SQL **没有产品过滤**（`WHERE 1=1 ... GROUP BY ... LIMIT 50`）
⇒ 返回一批无关于问题的产品净值。现库实测：**6 位代码 25 个、5 位代码 1 个**
（后者就是演示产品 `15911`）—— 所以这个缺陷精确地打在了演示产品上。
`tests/unit/service/test_financial_nl2sql_golden_cases.py` 里 30 条黄金用例**全是 6 位代码**
（159511/588890/159948），因此从未暴露。
## 修法
`app/service/financial_nl2sql_service.py`：
- 产品代码模式放宽为 **5–6 位**（`(?<!\d)\d{5,6}(?!\d)`）；
- 5 位数字更容易与"金额/数量/天数"撞车（6 位不会），因此加单位守卫：
紧跟 `元/万元/万/亿/份/股/手/天/日/周/个月/年/次/笔` 的一律**不当**产品代码
（例：「申购金额50000元」里的 50000）；
- 抽成 `_product_codes()` 便于单测。
## 验证
- `query("查询15911的净值")` → SQL 含 `p.product_code = :filter_0`、参数 `15911`、
返回行 `{"product_code": "15911", "product_name": "15911 自定义产品", "nav": "1.074596"}`；
三种问法（"查询15911的净值"/"查询基金代码 15911 最近 30 天的净值"/"15911 的最新净值是多少"）全部命中 15911；
- 新增 5 条单测：5 位代码成为过滤条件、6 位仍正常、金额不当代码、
年份/天数不被误认、多代码并存；
- `pytest tests/unit tests/contract` → **1458 passed, 2 skipped, 0 failed**；
- `ruff` 干净。
## 说明（未改，供你决定）
问「最新净值」时返回的是该产品的 50 行净值明细（SQL 无 `ORDER BY`），
不是"最新的那一条"。要的话我可以补成按日期倒序、单点取值。
131 lines
4.8 KiB
Python
131 lines
4.8 KiB
Python
from datetime import date
|
||
|
||
import pytest
|
||
|
||
from app.core.contracts import RequestContext
|
||
from app.core.nl2sql_contracts import FinancialNL2SQLInput
|
||
from app.service.financial_nl2sql_service import FinancialNL2SQLService
|
||
|
||
|
||
def context(**updates):
|
||
base = RequestContext(
|
||
user_id="1",
|
||
trace_id="trace-nl2sql",
|
||
roles=("advisor",),
|
||
permissions=("financial:nl2sql:read",),
|
||
data_scope="all",
|
||
)
|
||
return base.model_copy(update=updates)
|
||
|
||
|
||
def test_query_rows_convert_date_values_to_json_strings() -> None:
|
||
row = FinancialNL2SQLService._jsonable({"nav_date": date(2026, 9, 10)})
|
||
|
||
assert row == {"nav_date": "2026-09-10"}
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_generates_read_only_market_sql_with_product_filter() -> None:
|
||
result = await FinancialNL2SQLService().query(
|
||
FinancialNL2SQLInput(question="查询159511近30天行情收盘价", dry_run=True),
|
||
context(),
|
||
)
|
||
assert result["status"] == "ready"
|
||
assert result["sql"].startswith("SELECT ")
|
||
assert "fin_market_price" in result["sql"]
|
||
assert "DROP" not in result["sql"]
|
||
assert result["parameters"]["filter_0"] == "159511"
|
||
assert result["audit"]["permission_check"]["status"] == "passed"
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_low_confidence_question_requires_confirmation() -> None:
|
||
result = await FinancialNL2SQLService().query(
|
||
FinancialNL2SQLInput(question="帮我看看这个情况", dry_run=True),
|
||
context(),
|
||
)
|
||
assert result["status"] == "need_confirmation"
|
||
assert result["query_plan"]["confidence"] < 0.85
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_as_of_query_rejects_current_snapshot_tables() -> None:
|
||
result = await FinancialNL2SQLService().query(
|
||
FinancialNL2SQLInput(question="截至某日客户当前持仓市值", dry_run=True),
|
||
context(),
|
||
)
|
||
assert result["status"] == "rejected"
|
||
assert "历史版本" in result["message"]
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_customer_scope_is_injected_for_non_all_scope() -> None:
|
||
result = await FinancialNL2SQLService().query(
|
||
FinancialNL2SQLInput(question="查询客户账户余额", dry_run=True),
|
||
context(data_scope="own_customers", customer_ids=("7", "8")),
|
||
)
|
||
assert "customer_id IN" in result["sql"]
|
||
assert result["parameters"]["scope_customer_0"] == 7
|
||
assert result["parameters"]["scope_customer_1"] == 8
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_cash_ledger_is_in_nl2sql_scope() -> None:
|
||
result = await FinancialNL2SQLService().query(
|
||
FinancialNL2SQLInput(question="查询近7天资金流水变化", dry_run=True),
|
||
context(),
|
||
)
|
||
assert result["status"] == "ready"
|
||
assert "fin_cash_ledger" in result["sql"]
|
||
assert result["query_plan"]["intent"] == "cash_ledger_query"
|
||
|
||
|
||
# --- 产品代码提取：5 位与 6 位并存（2026-09-14 修） --------------------------------
|
||
#
|
||
# 原先只认 `\d{6}`，5 位代码（演示产品 `15911`）被静默丢弃 ⇒ 查询**不带产品过滤**，
|
||
# 却仍返回 `status=success`：实测问 15911 的净值拿到的是 511810 的数据。
|
||
# 「看起来成功、数据却是错的」比报错更危险，所以这几条要钉住。
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_five_digit_product_code_is_used_as_filter() -> None:
|
||
result = await FinancialNL2SQLService().query(
|
||
FinancialNL2SQLInput(question="查询15911的净值", dry_run=True),
|
||
context(),
|
||
)
|
||
|
||
assert result["status"] == "ready"
|
||
assert result["parameters"]["filter_0"] == "15911"
|
||
# 关键：SQL 里必须**真的带上产品过滤**（列名走别名 `p.`），否则就是"成功但查错数据"。
|
||
assert "p.product_code = :filter_0" in result["sql"]
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_six_digit_product_code_still_works() -> None:
|
||
result = await FinancialNL2SQLService().query(
|
||
FinancialNL2SQLInput(question="查询510300的净值", dry_run=True),
|
||
context(),
|
||
)
|
||
|
||
assert result["parameters"]["filter_0"] == "510300"
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_amount_like_five_digit_number_is_not_treated_as_product_code() -> None:
|
||
"""「申购金额50000元」里的 50000 是金额，不是产品代码 —— 它后面紧跟单位。"""
|
||
result = await FinancialNL2SQLService().query(
|
||
FinancialNL2SQLInput(question="查询申购金额50000元的记录", dry_run=True),
|
||
context(),
|
||
)
|
||
|
||
assert "filter_0" not in result.get("parameters", {})
|
||
|
||
|
||
def test_product_codes_ignores_years_and_short_numbers() -> None:
|
||
from app.service.financial_nl2sql_service import _product_codes
|
||
|
||
assert _product_codes("查询2026年的净值") == [] # 4 位年份
|
||
assert _product_codes("查询近30天净值") == [] # 天数
|
||
assert _product_codes("查询15911的净值") == ["15911"]
|
||
assert _product_codes("对比15911和510300") == ["15911", "510300"]
|