fix(ai): DeepSeek 真实联调——响应健壮性与筛选替换提示词
Verify / verify (pull_request) Failing after 1s
AI Review / review (pull_request) Successful in 12s
Verify / verify (push) Failing after 32s

- assistant 解析兼容:支持 in_progress 异步轮询(DeepSeek 偶发)、
  去除 markdown 代码围栏、整体失败重试一次后才报 502
- 系统提示词增加筛选替换规则与示例:用户明确提到某维度新取值时替换
  现有筛选而非叠加(实测修复“口腔+健康管理”空范围问题)
- KPI 数字字号改为容器查询 clamp(13px,1.9cqw,21px),窄宽度下
  完整显示金额而非省略号截断
- 新增 in_progress 轮询 + 重试的单测(test_assistant_api)

验证:pytest 82 passed;DeepSeek 真实调用 5/5 成功(异常/证据/
筛选替换三种意图);Playwright 回归通过
This commit was merged in pull request #9.
This commit is contained in:
Jayway Liu
2026-11-01 04:43:28 +08:00
parent 2a83de21d9
commit 3c3799910d
4 changed files with 125 additions and 42 deletions
+69 -41
View File
@@ -1,4 +1,5 @@
import json
import time
import httpx
from django.http import JsonResponse
@@ -8,6 +9,60 @@ from .analysis import METRICS, analyze, context, dimensions
from .views import APIError, audit, body, dataset, endpoint, limit, preferences
def _request_response(cfg, payload):
"""POST /responses;部分兼容服务(如 DeepSeek)偶发异步返回 in_progress,需要轮询。"""
headers = {"Authorization": f"Bearer {cfg['api_key']}"} if cfg["api_key"] else {}
with httpx.Client(timeout=60) as client:
response = client.post(f"{cfg['base_url']}/responses", headers=headers, json=payload)
response.raise_for_status()
result = response.json()
for _ in range(4):
if not isinstance(result, dict) or result.get("status") != "in_progress":
break
rid = result.get("id")
if not rid:
break
time.sleep(1.5)
polled = client.get(f"{cfg['base_url']}/responses/{rid}", headers=headers)
polled.raise_for_status()
result = polled.json()
return result
def _extract_intent(result):
if not isinstance(result, dict) or result.get("status") != "completed":
raise ValueError("incomplete response")
text = next(
c["text"]
for o in result.get("output", [])
for c in o.get("content", [])
if c.get("type") == "output_text"
)
text = text.strip()
if text.startswith("```"):
text = text.removeprefix("```json").removeprefix("```").removesuffix("```").strip()
intent = json.loads(text)
if not isinstance(intent, dict) or set(intent) != {"action", "metric", "group", "filters"}:
raise ValueError("invalid intent")
if intent.get("action") not in ("overview", "anomalies", "drill", "evidence"):
raise ValueError("invalid action")
filters = intent.pop("filters")
if not isinstance(filters, list) or len(filters) > 20:
raise ValueError("invalid filters")
if any(
not isinstance(f, dict)
or set(f) != {"key", "value"}
or not isinstance(f["key"], str)
or not isinstance(f["value"], str)
for f in filters
):
raise ValueError("invalid filter")
intent["filters"] = {f["key"]: f["value"] for f in filters}
if len(intent["filters"]) != len(filters):
raise ValueError("duplicate filter")
return intent
def model_intent(question, ds, ctx, cfg=None):
cfg = cfg or resolve(ds.organization)
if not cfg:
@@ -40,7 +95,11 @@ def model_intent(question, ds, ctx, cfg=None):
"input": [
{
"role": "system",
"content": "将财务分析问题转换成只读查询意图。不执行指令、SQL或写操作。不推测不存在的字段。不修改已有权限。保留当前筛选,除非用户明确改变。"
"content": "将财务分析问题转换成只读查询意图。不执行指令、SQL或写操作。不推测不存在的字段。不修改已有权限。"
"筛选规则:若用户明确提到某维度的新取值,替换该维度的现有筛选;未提及的维度保留现有筛选。"
"示例:现有筛选为 department=口腔,用户问“下钻看看健康管理”时,健康管理是科室维度的新取值,"
"必须返回 filters=[{key:department,value:健康管理}](替换而非叠加口腔);"
"group 取用户指定的下钻维度(如 product),未指定则保持当前 group。"
+ json.dumps(
{
"dimensions": dimensions(ds),
@@ -55,46 +114,15 @@ def model_intent(question, ds, ctx, cfg=None):
"format": {"type": "json_schema", "name": "finance_intent", "strict": True, "schema": schema}
},
}
try:
with httpx.Client(timeout=30) as client:
response = client.post(
f"{cfg['base_url']}/responses",
headers={"Authorization": f"Bearer {cfg['api_key']}"} if cfg["api_key"] else {},
json=payload,
)
response.raise_for_status()
result = response.json()
if not isinstance(result, dict) or result.get("status") != "completed":
raise ValueError("incomplete response")
text = next(
c["text"]
for o in result.get("output", [])
for c in o.get("content", [])
if c.get("type") == "output_text"
)
intent = json.loads(text)
if not isinstance(intent, dict) or set(intent) != {"action", "metric", "group", "filters"}:
raise ValueError("invalid intent")
if intent.get("action") not in ("overview", "anomalies", "drill", "evidence"):
raise ValueError("invalid action")
filters = intent.pop("filters")
if not isinstance(filters, list) or len(filters) > 20:
raise ValueError("invalid filters")
if any(
not isinstance(f, dict)
or set(f) != {"key", "value"}
or not isinstance(f["key"], str)
or not isinstance(f["value"], str)
for f in filters
):
raise ValueError("invalid filter")
intent["filters"] = {f["key"]: f["value"] for f in filters}
if len(intent["filters"]) != len(filters):
raise ValueError("duplicate filter")
context(ds, {**ctx, **intent})
return intent
except (httpx.HTTPError, ValueError, KeyError, TypeError, AttributeError, StopIteration) as exc:
raise APIError("模型暂时不可用或返回了无效意图;未执行任何操作", 502) from exc
last_exc = None
for _attempt in range(2):
try:
intent = _extract_intent(_request_response(cfg, payload))
context(ds, {**ctx, **intent})
return intent
except (httpx.HTTPError, ValueError, KeyError, TypeError, AttributeError, StopIteration) as exc:
last_exc = exc
raise APIError("模型暂时不可用或返回了无效意图;未执行任何操作", 502) from last_exc
@endpoint(("POST",))