mirror of
https://github.com/ThearchyHelios/AgentLab.git
synced 2026-11-15 07:02:00 +00:00
ThearchyHelios
2f90f0eb3a
- QueryCore 固定执行（K09）：按已发布的不可变口径跑一次，交回类型化标量、发布/manifest 引用与逐查询证据 - 依赖闭包（K08）：按选中的指标算出要拷的取数节点、要的入口字段，接进 derive / compose / metrics - SQL 模板参数完整性（K01）：试跑、组合问数、派生图、运行时取数共用一套语义，缺参数在渲染前就拒 - 受限字段变更服务（K03）与口径领域边界（K01）：改动打在整图深拷贝上，身份与证据归属不可改 - 版本化受控模板绑定（K04）：业务口径按显式绑定落到 SQL，对不上就报 unsupported / needs_update - 验证记录（K05）与最小审阅：样例匹配不等于正式批准，审批人只从服务端身份取 - 发布依据清单（K07）：发布时按不可变图物化内容寻址清单，正式用途按它判 - 增量重导入（K11）与 B/L/R 三方合并、受限自然语言修改提案（K12）：只产候选，落图仍走既有写路径 - 接口与前端：口径草稿页接上字段变更、绑定状态、验证与重导入，新增 caliber-drafts / studio 检查脚本 验证：前端 tsc -b 通过。
849 lines
47 KiB
Python
849 lines
47 KiB
Python
"""业务文档 + 数据目录 → 一份一次到位的口径草稿：定义、取数 SQL、入口字段、指标表达式（docs/design 做法 8 的主路径）。
|
||
|
||
**为什么这次要模型写 SQL。** 文档里只有业务说法，没有表名。但只抽定义，人还得自己去问一次数、再把运行 id 接回来，等于把
|
||
「口径怎么写」拆成两段，谁都说不清草稿什么时候算数。主路径改成一次产出可试跑的图：模型读文档、对照数据目录写一条只读
|
||
SELECT，服务端把 SQL、入口字段、口径卡、报告串成图，后台跑一次，草稿页直接给「定义 + 数 + 原文引用」。
|
||
|
||
**为什么目录是唯一依据。** 文档里没有表名列名，模型只能从调用方给的目录里挑。所以没有数据源、或数据源还没探查表结构时，
|
||
调用方必须拒收（api/ledger 的 422）：让模型凭空猜表，猜出来的图跑不了，还不如不做。normalize_plan 再核对一遍 SQL 里出现的
|
||
表都在目录里，防止模型从示例、同义词里编出表名。
|
||
|
||
**为什么 SQL 只收单条只读 SELECT。** 草稿会在后台自动跑一次。多语句、DML/DDL、注释后藏的第二条都可能在试跑时改库；
|
||
这里按 sqlglot 解析后的语句数判定，不靠字符串里有没有分号（字符串字面量里的分号不算）。注释里的第二条照样会被解析成
|
||
第二条语句，所以也拒。
|
||
|
||
**为什么表达式必须是 cell(nodes.fetch, 0, '列名')。** 试跑和之后按口径出具都从口径卡读值。一次 plan 只起草一组，这组的
|
||
取数节点固定叫 fetch（多组拼图时由 build_graph 改成 fetch_1、fetch_2…），单行结果取第 0 行的某一列；写成别的样子，卡上
|
||
就没有值，图能建但出不了数。
|
||
|
||
**为什么 build_graph 收多组 blocks。** 一张草稿图原来只能挂一条 SQL，业务文档几十条指标得分几十次起草、几十张图。取数
|
||
节点多几个、口径卡读全部，试跑一次就跑完；入口字段按 name 合并，必填取并集、类型按一条固定的合并规则定（见 _merged_type，
|
||
结果与各组先后无关），默认值冲突就清掉默认值要求人填。不抛错、也不悄悄取某一组：不一致的地方记进 entry_notes 摆给起草的
|
||
人看（早先这里抛 ValueError，一份 34 条指标的草稿因为一个字段的 required 不同就整份作废）。
|
||
|
||
**为什么引用核对还是照 docdraft 的老规矩。** 定义的事实来源还是文档原文。模型转述得再像，也要能在原文里逐字定位；一条
|
||
引用都核不上的指标整条丢弃，写进 problems 交给人看。这里只做校验，不改 docdraft 的规则。
|
||
|
||
**为什么不落库、不导 app.api。** 这个模块是纯逻辑加一次模型调用：从哪读文档、允许哪些表、结果存成什么，都由调用方决定。
|
||
测试注入假模型，不会发真实请求。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import re
|
||
from collections.abc import Mapping, Sequence
|
||
from typing import Any
|
||
|
||
from app.ledger import docdraft
|
||
|
||
#: 一次发给模型的材料上限（按 chunk 边界截断；表目录先截到这个数以内）
|
||
MAX_INPUT_CHARS = docdraft.MAX_INPUT_CHARS
|
||
MAX_TABLES_CHARS = 24000
|
||
#: 取数 SQL 的长度上限（字符）。再长的查询不是草稿该承载的
|
||
MAX_SQL_CHARS = 20000
|
||
#: 这一版提示词的版本（改了 _SYSTEM 就 +1）。写进草稿的提示记录，历史结果才知道是按哪一版起草的。
|
||
PROMPT_VERSION = 1
|
||
|
||
#: 入口字段能写的类型（apps/spec.FORM_TYPES 的子集；日期用 string + description 说清）
|
||
ENTRY_TYPES = ("string", "number", "integer", "boolean")
|
||
#: 指标里固定出现的字段（顺序照接口契约）。grain、time_field、numerator、denominator、dimensions、filters 是文档起草
|
||
#: 交回来的业务描述（docdraft.METRIC_FIELDS 里也有），校验阶段必须原样留着，草稿页要逐行展示
|
||
METRIC_FIELDS = ("id", "name", "synonyms", "unit", "decimals", "expression", "grain", "time_field",
|
||
"numerator", "denominator", "dimensions", "filters", "definition", "open_questions")
|
||
|
||
#: 入口字段最多几个（时间范围占两个）。再多说明把「每次都不变」的条件也做成了参数，
|
||
#: 人每跑一次都要填一遍，口径就没法用了
|
||
MAX_ENTRY_FIELDS = 4
|
||
#: 输入里能出现的入口占位符：{{ input.x }}（引号内外都认）
|
||
_INPUT = re.compile(r"\{\{\s*input\.([A-Za-z_][\w-]*)\s*\}\}")
|
||
#: 口径卡的取值表达式，列名可以用单引号或双引号
|
||
_EXPRESSION = re.compile(r"^cell\(\s*nodes\.fetch\s*,\s*0\s*,\s*['\"]([^'\"]+)['\"]\s*\)$")
|
||
#: 指标表达式里的取数节点引用；后面紧跟字母数字下划线的不算（nodes.fetch_2 已经带下标，不改）
|
||
_FETCH_REF = re.compile(r"nodes\.fetch(?![A-Za-z0-9_])")
|
||
#: 只读查询的根节点：普通 SELECT，或各分支都是 SELECT 的 UNION
|
||
_WRITE_NODES = ("Insert", "Update", "Delete", "Create", "Drop", "Alter", "Merge", "TruncateTable", "Command")
|
||
|
||
_SYSTEM = """你是业务口径与取数模板的起草助手。读给定的业务文档和数据目录，一次产出一份可试跑的口径。
|
||
|
||
硬性约束：
|
||
1. 只输出一个 JSON 对象，不要写解释，不要写 Markdown 代码块之外的话。
|
||
2. 每组的 sql 是取数模板，只能是**一条只读 SELECT**：不要多语句、不要 INSERT/UPDATE/DELETE/DDL，也不要在注释后面再藏一条。
|
||
表和列只能来自「数据目录」，表名、列名照抄，不许猜、不许编。
|
||
3. sql 里需要按入口筛选时写 '{{ input.字段名 }}'（字符串）或 {{ input.字段名 }}（数），并在该组的 entry_fields 里声明同名
|
||
字段；不要把日期、门店之类的字面量写死。
|
||
4. 每条指标都要有 expression，固定写成 cell(nodes.fetch, 0, '列名')，列名是这组 sql 最外层 SELECT 的输出列名。
|
||
5. 每条指标的定义必须带 citations：引用逐字抄原文，chunk 写材料里方括号内的 id。核对不上、或原文里没有依据的定义
|
||
不要写进指标，写进 open_questions。
|
||
6. 拿不准的不要猜，写进 open_questions，用一句问得清的话。不要编造文档里没有的数字、单位、生效日期。
|
||
7. **一次尽量多交几组。** 每组是能共用同一条 SQL 的指标，最多 6 条；不同粒度、不同表、不同时间字段的指标不要硬凑
|
||
进同一组。文档里其余的指标不要硬凑，把名称写进 uncovered（字符串数组），下一轮再起草。human 里会列出「已经起草
|
||
过的指标」：这些不要重复起草，剩下的未起草指标都写进 uncovered。
|
||
8. **sql 要短。** 能用 GROUP BY 和聚合表达的，不要为每个属性枚举一长串 CASE WHEN 列；输出额度有限，SQL 写长了，
|
||
后面的指标就写不出来了。
|
||
9. 写不下时宁可少写几组、把剩下的写进 uncovered，**也不要让 JSON 半截断掉**。
|
||
10. **同一件事只做一个入口。** 工厂、产线这类维度全图只给一个字段，不要为不同的表各造一个
|
||
（factory_code / gc_code / scx_code 说的是同一件事）；跨表的编码差异在 sql 里用 JOIN 或子查询映射，
|
||
不要让填参数的人重复填等价编码。
|
||
11. **编码和主键不当必填入口。** object_id、line_code 这类字段只有业务上确实每次按它跑时才做入口，
|
||
而且必须是可选（留空不过滤）；人填不出来才能跑的入口，等于这份口径跑不了。
|
||
12. **可选维度要写成「空值不过滤」**：`AND ('{{ input.line_code }}' = '' OR t.line_code = '{{ input.line_code }}')`。
|
||
不这么写，留空会把结果筛成空。
|
||
13. **入口最多 4 个**，时间范围占两个。再多就把不常变的条件写死在 sql 里（见第 14 条）。
|
||
14. **不常变的条件写进 sql，不做入口**：固定筛选段里给出的取值（工厂、产线、属性组、状态）直接写成常量。
|
||
|
||
输出格式：
|
||
{
|
||
"groups": [
|
||
{
|
||
"entry_fields": [
|
||
{"name": "start", "type": "string", "required": true, "default": "", "description": "开始日期，格式 YYYY-MM-DD"}
|
||
],
|
||
"sql": "SELECT SUM(amount) AS gmv FROM orders WHERE paid_at >= '{{ input.start }}'",
|
||
"metrics": [
|
||
{
|
||
"id": "m1",
|
||
"name": "指标名称",
|
||
"synonyms": ["同义词"],
|
||
"unit": "单位（文档没写就留空）",
|
||
"decimals": 2,
|
||
"expression": "cell(nodes.fetch, 0, 'gmv')",
|
||
"definition": {
|
||
"text": "口径说明",
|
||
"scope": "适用范围",
|
||
"exclusions": ["不包含的情况"],
|
||
"owner": "文档写明的负责人或部门",
|
||
"effective_from": "生效日期（文档原文写法）",
|
||
"citations": [{"chunk": "c1", "quote": "逐字原文"}]
|
||
},
|
||
"open_questions": ["还没说清、需要业务确认的问题"]
|
||
}
|
||
]
|
||
}
|
||
],
|
||
"open_questions": ["跨指标、以及整条 SQL 还没把握的问题"],
|
||
"uncovered": ["这次没有起草的指标名称（不同粒度、不同表、或写不下的）"]
|
||
}
|
||
"""
|
||
|
||
|
||
def _collapse(text: str) -> str:
|
||
return re.sub(r"\s+", " ", text).strip()
|
||
|
||
|
||
def _order(chunks: Mapping[str, dict]) -> list[str]:
|
||
def key(chunk_id: str) -> tuple[int, str]:
|
||
chunk = chunks[chunk_id] if isinstance(chunks[chunk_id], Mapping) else {}
|
||
ordinal = chunk.get("ordinal")
|
||
return (ordinal if isinstance(ordinal, int) else 0, chunk_id)
|
||
|
||
return sorted(chunks, key=key)
|
||
|
||
|
||
def _doc_line(doc: Any) -> str:
|
||
if not isinstance(doc, Mapping):
|
||
return str(doc)
|
||
doc_id = doc.get("document_id") or doc.get("id") or ""
|
||
title = doc.get("title") or doc.get("filename") or doc.get("name") or doc_id
|
||
return f"- {doc_id}：{title}"
|
||
|
||
|
||
def _chunk_block(chunk_id: str, chunk: Any) -> str:
|
||
chunk = chunk if isinstance(chunk, Mapping) else {}
|
||
ordinal = chunk.get("ordinal")
|
||
where = f" · 第 {ordinal} 段" if isinstance(ordinal, int) else ""
|
||
document = chunk.get("document_id") or ""
|
||
return f"[{chunk_id}]（文档 {document}{where}）\n{chunk.get('content') or ''}"
|
||
|
||
|
||
def _human(docs: list[dict], chunks: Mapping[str, dict], guidance: str, datasource: dict | None,
|
||
tables_text: str, covered: Sequence[str] = (), fixed: str = "") -> str:
|
||
parts = ["文档清单：", *(_doc_line(doc) for doc in docs), "",
|
||
"逐段原文（引用时 chunk 写方括号里的 id，quote 逐字抄）："]
|
||
parts.extend(_chunk_block(chunk_id, chunks[chunk_id]) for chunk_id in _order(chunks))
|
||
parts.append("")
|
||
if datasource:
|
||
bits = [str(datasource.get(key)) for key in ("name", "type", "dialect", "description")
|
||
if datasource.get(key)]
|
||
parts.append("数据源：{}".format(" · ".join(bits)))
|
||
parts.append("数据目录（sql 只能用这里的表和列，表名列名照抄）：")
|
||
parts.append(tables_text.strip() or "（没有给出表结构，不能写 sql）")
|
||
parts.append("")
|
||
parts.append("已经起草过的指标（不要重复起草）：")
|
||
names = [name.strip() for name in covered if isinstance(name, str) and name.strip()]
|
||
parts.append("、".join(names) if names else "（无）")
|
||
parts.append("")
|
||
parts.append("固定条件（这些写死在 sql 里，**不要**做成入口字段）：{}".format(
|
||
fixed.strip() if fixed and fixed.strip() else "（无）"))
|
||
parts.append("")
|
||
parts.append("用户补充要求：{}".format(guidance.strip() if guidance and guidance.strip() else "（无）"))
|
||
return "\n".join(parts)
|
||
|
||
|
||
def prompt(docs: list[dict], chunks: dict[str, dict], guidance: str = "", datasource: dict | None = None,
|
||
tables_text: str = "", covered: Sequence[str] = (), fixed: str = "") -> list[Any]:
|
||
"""构造 system + human：文档原文 + 数据目录 + 已经起草过的指标 + 固定条件。目录是 sql 的唯一依据。"""
|
||
from langchain_core.messages import HumanMessage, SystemMessage
|
||
|
||
return [SystemMessage(content=_SYSTEM),
|
||
HumanMessage(content=_human(docs, chunks, guidance, datasource, tables_text, covered, fixed))]
|
||
|
||
|
||
def _chars(messages: list[Any]) -> int:
|
||
return sum(len(str(getattr(message, "content", "") or "")) for message in messages)
|
||
|
||
|
||
def fit_material(docs: list[dict], chunks: dict[str, dict], guidance: str,
|
||
datasource: dict | None, tables_text: str,
|
||
covered: Sequence[str] = (), fixed: str = "") -> tuple[dict[str, dict], str, bool, bool]:
|
||
"""把材料压到 MAX_INPUT_CHARS 以内：先截表目录尾部，再从后往前按 chunk 边界丢，最后截仅剩的那一段。
|
||
返回 (真正进提示的 chunk, 真正进提示的目录, doc_truncated, catalog_truncated)。
|
||
|
||
表目录先截：SQL 必须写在目录里的表上，目录比文档的细枝末节更硬；被截掉的是表详情，表名索引还在最前面。
|
||
|
||
截断来源分开报：doc_truncated 是文档段落被丢/被截，catalog_truncated 是表目录被截。调用方要能说清「指标没覆盖
|
||
完，是因为原文没给全，还是因为目录没给全」——改文档、改目录是两件不同的事。
|
||
|
||
**返回值就是「模型实际看到的材料」。** 调用方拿它去核对引用（不能引用被丢掉的段落）、记进提示记录；也可以自己
|
||
先算一次、把结果传回 plan / plan_groups / repair_group，免得同一份材料在两处各算一遍、算得不一样。
|
||
"""
|
||
kept = dict(chunks)
|
||
tables = tables_text
|
||
doc_truncated = False
|
||
catalog_truncated = False
|
||
if len(tables) > MAX_TABLES_CHARS:
|
||
tables = tables[:MAX_TABLES_CHARS]
|
||
catalog_truncated = True
|
||
|
||
def size() -> int:
|
||
return _chars(prompt(docs, kept, guidance, datasource, tables, covered, fixed))
|
||
|
||
order = _order(kept)
|
||
while size() > MAX_INPUT_CHARS and len(order) > 1:
|
||
kept.pop(order.pop())
|
||
doc_truncated = True
|
||
if size() > MAX_INPUT_CHARS and order:
|
||
chunk_id = order[0]
|
||
chunk = dict(kept[chunk_id]) if isinstance(kept[chunk_id], Mapping) else {}
|
||
content = str(chunk.get("content") or "")
|
||
excess = size() - MAX_INPUT_CHARS
|
||
chunk["content"] = content[:max(0, len(content) - excess)]
|
||
kept[chunk_id] = chunk
|
||
doc_truncated = True
|
||
if size() > MAX_INPUT_CHARS:
|
||
excess = size() - MAX_INPUT_CHARS
|
||
tables = tables[:max(0, len(tables) - excess)]
|
||
catalog_truncated = True
|
||
return kept, tables, doc_truncated, catalog_truncated
|
||
|
||
|
||
def _salvage_value(text: str, key: str) -> Any:
|
||
"""从被截断的回复里抢回一个完整的顶层字段（"sql": "…" 这样的值）。值本身不完整就返回 None。"""
|
||
at = text.find(f'"{key}"')
|
||
if at < 0:
|
||
return None
|
||
colon = text.find(":", at + len(key) + 2)
|
||
if colon < 0:
|
||
return None
|
||
start = colon + 1
|
||
while start < len(text) and text[start] in " \t\r\n":
|
||
start += 1
|
||
try:
|
||
value, _ = json.JSONDecoder().raw_decode(text, start)
|
||
except (json.JSONDecodeError, RecursionError):
|
||
return None
|
||
return value
|
||
|
||
|
||
def _salvage_reply(text: str) -> dict:
|
||
"""JSON 被截断时，把还能完整读出来的部分抢回来。
|
||
|
||
指标逐条抢（docdraft 按平衡括号找完整的指标对象），sql 和入口字段按完整的 JSON 值抢。抢不回来的留空，
|
||
由调用方照实说「这次起草不完整」，而不是把半截回复当成没有指标。
|
||
"""
|
||
return {
|
||
"metrics": docdraft.salvage_metrics(text),
|
||
"sql": _salvage_value(text, "sql") or "",
|
||
"entry_fields": _salvage_value(text, "entry_fields") or [],
|
||
"open_questions": _salvage_value(text, "open_questions") or [],
|
||
"uncovered": _salvage_value(text, "uncovered") or [],
|
||
}
|
||
|
||
|
||
async def plan(session: Any, *, docs: list[dict], chunks: dict[str, dict], guidance: str = "",
|
||
datasource: dict | None = None, tables_text: str = "", model: Any = None,
|
||
covered: Sequence[str] = (), fixed: str = "",
|
||
fitted: tuple[dict[str, dict], str, bool, bool] | None = None) -> dict:
|
||
"""一次模型调用，返回 {entry_fields, sql, metrics, open_questions, model_id, doc_truncated,
|
||
catalog_truncated, truncated, visible_chunk_ids, prompt_version, parse_version}（未校验）。
|
||
|
||
covered 是前几轮已经起草过的指标名，写进提示词让模型不要重复；fixed 是「每次都不变的条件」（工厂、产线、
|
||
属性组），提示词要求它写死在 sql 里、不做成入口字段。调用方按 uncovered 续跑，直到 covered 不再增长。
|
||
截断情况与 visible_chunk_ids 说的是同一份材料：**实际进了这次提示的**那几段，被 fit_material 丢掉的段落不在里面，
|
||
引用它也核不上。fitted 是调用方先算好的材料（fit_material 的返回值），给了就用它，不给这里自己算。
|
||
truncated 仍是两者的「或」，老前端只认这一个键。不落库。model 可注入（测试用假对象：有 async ainvoke(messages)，
|
||
回复有 .content）；不给就按设置解析，模型不可用时抛 docdraft.DraftUnavailable。SQL 的对错、引用能不能核对，交给
|
||
normalize_plan。
|
||
"""
|
||
from app.engine.state import message_text
|
||
|
||
if model is None:
|
||
from app.data.recipe_ai import resolve_draft_model
|
||
|
||
model, model_id, availability = await resolve_draft_model(session)
|
||
if model is None or not getattr(availability, "available", False):
|
||
raise docdraft.DraftUnavailable(getattr(availability, "reason", "") or "AI 起草不可用")
|
||
else:
|
||
named = getattr(model, "model_name", None) or getattr(model, "model", None)
|
||
model_id = named if isinstance(named, str) else ""
|
||
|
||
if fitted is None:
|
||
fitted = fit_material(docs, chunks, guidance, datasource, tables_text, covered, fixed)
|
||
kept, tables, doc_truncated, catalog_truncated = fitted
|
||
reply = await model.ainvoke(prompt(docs, kept, guidance, datasource, tables, covered, fixed))
|
||
text = message_text(reply)
|
||
salvaged = False
|
||
try:
|
||
data = docdraft.parse_reply(text)
|
||
except ValueError:
|
||
# 回复被截断（SQL 写长了、指标太多）：能抢回多少算多少，别整份当成「没有指标」
|
||
data, salvaged = _salvage_reply(text), True
|
||
metrics = data.get("metrics")
|
||
questions = data.get("open_questions")
|
||
uncovered = data.get("uncovered")
|
||
metrics_list = metrics if isinstance(metrics, list) else []
|
||
diagnostics = ""
|
||
if salvaged:
|
||
diagnostics = f"回复被截断，从完整片段里抢救出 {len(metrics_list)} 条指标"
|
||
if not metrics_list:
|
||
keys = "、".join(str(k) for k in data) or "无"
|
||
shape = {"dict": "对象", "list": "数组", "str": "字符串", "NoneType": "空"}.get(
|
||
type(metrics).__name__, type(metrics).__name__)
|
||
diagnostics = (f"模型回复里没有可读的指标数组（metrics 是{shape}；顶层键：{keys}）"
|
||
+ (f"；{diagnostics}" if diagnostics else ""))
|
||
return {
|
||
"entry_fields": data.get("entry_fields") if isinstance(data.get("entry_fields"), list) else [],
|
||
"sql": data.get("sql") if isinstance(data.get("sql"), str) else "",
|
||
"metrics": metrics_list,
|
||
"open_questions": [q for q in questions if isinstance(q, str) and q.strip()]
|
||
if isinstance(questions, list) else [],
|
||
"uncovered": [q.strip() for q in uncovered if isinstance(q, str) and q.strip()]
|
||
if isinstance(uncovered, list) else [],
|
||
"salvaged": salvaged,
|
||
"diagnostics": diagnostics,
|
||
"model_id": model_id,
|
||
"visible_chunk_ids": _order(kept),
|
||
"prompt_version": PROMPT_VERSION,
|
||
"parse_version": docdraft.PARSE_VERSION,
|
||
"doc_truncated": doc_truncated,
|
||
"catalog_truncated": catalog_truncated,
|
||
"truncated": doc_truncated or catalog_truncated,
|
||
}
|
||
|
||
|
||
# ==========================================================================
|
||
# 校验
|
||
# ==========================================================================
|
||
|
||
|
||
def _reply_groups(data: Mapping) -> list[dict]:
|
||
"""从模型回复里取分组：有 groups 数组就逐项收；旧形状（顶层 sql + metrics）当成只有一组。"""
|
||
raw = data.get("groups")
|
||
if isinstance(raw, list):
|
||
return [dict(group) for group in raw if isinstance(group, Mapping)]
|
||
return [dict(data)] if (data.get("sql") or data.get("metrics")) else []
|
||
|
||
|
||
async def plan_groups(session: Any, *, docs: list[dict], chunks: dict[str, dict], guidance: str = "",
|
||
datasource: dict | None = None, tables_text: str = "", model: Any = None,
|
||
covered: Sequence[str] = (), fixed: str = "",
|
||
fitted: tuple[dict[str, dict], str, bool, bool] | None = None) -> dict:
|
||
"""一次模型调用，交回**多组**草案（未校验）：{groups, open_questions, uncovered, model_id,
|
||
visible_chunk_ids, prompt_version, parse_version, ...}。
|
||
|
||
和 plan() 的区别只在输出形状：模型一次尽量交几组「能共用一条 SQL 的指标」，剩下的写 uncovered；调用方不必为一组
|
||
指标各调一次模型。每组的结构仍是 {sql, entry_fields, metrics}，交给 normalize_plan 逐组校验；一组不合格只丢那一组，
|
||
不影响别的组。旧形状（顶层 sql + metrics）也收，当成只有一组——模型不按新格式回时不用整份作废。
|
||
|
||
fitted 是调用方先算好的材料（fit_material 的返回值）。给了就用它，调用方由此知道自己看到的 visible_chunk_ids
|
||
与实际提示一致，不必再算一遍；不给这里自己算（其它入口照旧）。
|
||
"""
|
||
from app.engine.state import message_text
|
||
|
||
if model is None:
|
||
from app.data.recipe_ai import resolve_draft_model
|
||
|
||
model, model_id, availability = await resolve_draft_model(session)
|
||
if model is None or not getattr(availability, "available", False):
|
||
raise docdraft.DraftUnavailable(getattr(availability, "reason", "") or "AI 起草不可用")
|
||
else:
|
||
named = getattr(model, "model_name", None) or getattr(model, "model", None)
|
||
model_id = named if isinstance(named, str) else ""
|
||
|
||
if fitted is None:
|
||
fitted = fit_material(docs, chunks, guidance, datasource, tables_text, covered, fixed)
|
||
kept, tables, doc_truncated, catalog_truncated = fitted
|
||
reply = await model.ainvoke(prompt(docs, kept, guidance, datasource, tables, covered, fixed))
|
||
text = message_text(reply)
|
||
salvaged = False
|
||
try:
|
||
data = docdraft.parse_reply(text)
|
||
except ValueError:
|
||
salvaged_data = _salvage_reply(text)
|
||
if salvaged_data.get("sql") or salvaged_data.get("metrics"):
|
||
data = salvaged_data
|
||
else:
|
||
data = {"groups": [], "open_questions": [], "uncovered": []}
|
||
salvaged = True
|
||
groups = _reply_groups(data)
|
||
questions = data.get("open_questions")
|
||
uncovered = data.get("uncovered")
|
||
diagnostics = ""
|
||
if not groups:
|
||
keys = "、".join(str(key) for key in data) or "无"
|
||
diagnostics = f"模型回复里没有可读的分组（顶层键：{keys}）"
|
||
if salvaged:
|
||
diagnostics += "；回复被截断，没有抢救出完整的一组"
|
||
return {
|
||
"groups": groups,
|
||
"open_questions": [q for q in questions if isinstance(q, str) and q.strip()]
|
||
if isinstance(questions, list) else [],
|
||
"uncovered": [q.strip() for q in uncovered if isinstance(q, str) and q.strip()]
|
||
if isinstance(uncovered, list) else [],
|
||
"salvaged": salvaged,
|
||
"diagnostics": diagnostics,
|
||
"model_id": model_id,
|
||
"visible_chunk_ids": _order(kept),
|
||
"prompt_version": PROMPT_VERSION,
|
||
"parse_version": docdraft.PARSE_VERSION,
|
||
"doc_truncated": doc_truncated,
|
||
"catalog_truncated": catalog_truncated,
|
||
"truncated": doc_truncated or catalog_truncated,
|
||
}
|
||
|
||
|
||
async def repair_group(session: Any, *, group: Mapping, problems: list[dict], docs: list[dict],
|
||
chunks: dict[str, dict], guidance: str = "", datasource: dict | None = None,
|
||
tables_text: str = "", model: Any = None, fixed: str = "",
|
||
fitted: tuple[dict[str, dict], str, bool, bool] | None = None) -> dict:
|
||
"""只重写一组：把这一组原来的指标和没通过校验的原因交回模型，返回修好的原始组（未校验）。
|
||
|
||
修不出来返回 {}，由调用方照实记失败，不让整份草稿跟着作废。修的时候只给这一组的指标名和问题，不让模型重排别的组。
|
||
fitted 是调用方先算好的材料（fit_material 的返回值，与本次提示一致）。给了就用它，引用核对与本轮其它调用用同一份
|
||
可见材料；不给这里自己算。
|
||
"""
|
||
from app.engine.state import message_text
|
||
|
||
if model is None:
|
||
from app.data.recipe_ai import resolve_draft_model
|
||
|
||
model, _model_id, availability = await resolve_draft_model(session)
|
||
if model is None or not getattr(availability, "available", False):
|
||
raise docdraft.DraftUnavailable(getattr(availability, "reason", "") or "AI 起草不可用")
|
||
if fitted is None:
|
||
fitted = fit_material(docs, chunks, guidance, datasource, tables_text, (), fixed)
|
||
kept, tables = fitted[0], fitted[1]
|
||
names = [str(metric.get("name") or metric.get("id")) for metric in group.get("metrics") or []
|
||
if isinstance(metric, Mapping) and (metric.get("name") or metric.get("id"))]
|
||
reasons = "；".join(f"{problem.get('name') or '问题'}：{problem.get('reason') or ''}"
|
||
for problem in problems if problem.get("reason"))
|
||
repair = ("上一版这一组没有通过校验：" + (reasons or "原因不明") + "。"
|
||
+ (f"这一组原来的指标：{'、'.join(names)}。" if names else "")
|
||
+ "只重写这一组：保留原来的指标名，改正 sql / expression / entry_fields；不要重复别的组，也不要新增指标。")
|
||
full_guidance = f"{guidance}\n\n{repair}" if guidance.strip() else repair
|
||
reply = await model.ainvoke(prompt(docs, kept, full_guidance, datasource, tables, fixed=fixed))
|
||
text = message_text(reply)
|
||
try:
|
||
data = docdraft.parse_reply(text)
|
||
except ValueError:
|
||
data = _salvage_reply(text)
|
||
groups = _reply_groups(data)
|
||
return groups[0] if groups else {}
|
||
|
||
|
||
def _entry_type(value: Any) -> str:
|
||
"""入口字段声明的类型规范成 ENTRY_TYPES 里的一个词：认不出的按文字（apps.forms.input_schema 也按同一套认）。
|
||
|
||
起草时（_entry_fields）和合并各组声明时（_merged_type）用同一份判断：这套写法从哪个入口进来，认出来的都是同一个词。"""
|
||
return value if value in ENTRY_TYPES else "string"
|
||
|
||
|
||
def _entry_fields(value: Any, problems: list[dict]) -> list[dict]:
|
||
"""入口字段归一化：名字必填、不重名；类型落回能认的几种，认不出的按文字。"""
|
||
out: list[dict] = []
|
||
seen: set[str] = set()
|
||
for position, item in enumerate(value if isinstance(value, list) else [], start=1):
|
||
if not isinstance(item, Mapping):
|
||
problems.append({"name": f"第 {position} 个入口字段", "reason": "不是一个对象"})
|
||
continue
|
||
name = item.get("name")
|
||
if not isinstance(name, str) or not name.strip():
|
||
problems.append({"name": f"第 {position} 个入口字段", "reason": "缺少字段名"})
|
||
continue
|
||
name = name.strip()
|
||
if name in seen:
|
||
problems.append({"name": name, "reason": "入口字段重名"})
|
||
continue
|
||
seen.add(name)
|
||
kind = _entry_type(item.get("type"))
|
||
default = item.get("default")
|
||
if not isinstance(default, (str, int, float, bool)):
|
||
default = None
|
||
description = item.get("description")
|
||
out.append({"name": name, "type": kind, "required": bool(item.get("required")),
|
||
"default": default,
|
||
"description": description.strip() if isinstance(description, str) else ""})
|
||
return out
|
||
|
||
|
||
def _sql_problem(sql: str, allowed_tables: set[str]) -> str | None:
|
||
"""单条只读 SELECT、表都在目录里；不满足时返回一句给人看的原因。"""
|
||
if len(sql) > MAX_SQL_CHARS:
|
||
return f"SQL 超过 {MAX_SQL_CHARS} 字，草稿不承载这么长的查询"
|
||
body = _INPUT.sub("?", sql)
|
||
try:
|
||
import sqlglot
|
||
from sqlglot import exp
|
||
except ImportError: # pragma: no cover - sqlglot 是现有依赖，装不上时照实拒
|
||
return "没有 SQL 解析器，不能校验这条查询"
|
||
try:
|
||
statements = [item for item in sqlglot.parse(body) if item is not None]
|
||
except Exception as e: # noqa: BLE001 - 语法错、方言特性：都按解析不了
|
||
return f"SQL 解析不了：{e}"
|
||
if len(statements) != 1:
|
||
return "只能是一条语句：多语句、或注释后藏着第二条语句的，一律不收"
|
||
tree = statements[0]
|
||
if not isinstance(tree, (exp.Select, exp.Union)):
|
||
return "只能是只读 SELECT"
|
||
if isinstance(tree, exp.Union) and not all(isinstance(part, exp.Select) for part in tree.find_all(exp.Select)):
|
||
return "只能是只读 SELECT"
|
||
if any(tree.find(getattr(exp, name)) is not None for name in _WRITE_NODES):
|
||
return "只能是只读 SELECT，不能有写操作或建表、删表这类语句"
|
||
if isinstance(tree, exp.Select) and tree.args.get("into"):
|
||
return "只能是只读 SELECT，不能把结果写进另一张表"
|
||
cte_names = {str(cte.alias or "").casefold() for cte in tree.find_all(exp.CTE)}
|
||
allowed = {str(name).casefold() for name in allowed_tables}
|
||
used: list[str] = []
|
||
for table in tree.find_all(exp.Table):
|
||
name = str(table.name or "")
|
||
short = name.rsplit(".", 1)[-1]
|
||
if not name or name.casefold() in cte_names or short.casefold() in cte_names:
|
||
continue
|
||
db = str(table.db or "")
|
||
keys = {name.casefold(), short.casefold()}
|
||
if db:
|
||
keys.add(f"{db}.{name}".casefold())
|
||
if not (keys & allowed):
|
||
used.append(name)
|
||
if used:
|
||
return f"SQL 用了数据目录里没有的表：{'、'.join(dict.fromkeys(used))}"
|
||
if not any(True for table in tree.find_all(exp.Table)
|
||
if str(table.name or "").casefold() not in cte_names):
|
||
return "SQL 没有引用任何表：请对照数据目录写取数"
|
||
return None
|
||
|
||
|
||
def _text(value: Any) -> str:
|
||
"""文档起草交回的文字字段：去掉首尾空白；不是字符串的当没有。"""
|
||
return value.strip() if isinstance(value, str) else ""
|
||
|
||
|
||
def _texts(value: Any) -> list[str]:
|
||
"""文档起草交回的字符串列表：只收非空字符串（去首尾空白），别的类型和空项都丢掉。"""
|
||
if not isinstance(value, list):
|
||
return []
|
||
return [item.strip() for item in value if isinstance(item, str) and item.strip()]
|
||
|
||
|
||
def _metric(item: Mapping, position: int, problems: list[dict]) -> dict | None:
|
||
"""一条已经过 docdraft 引用核对的指标：再查 expression，归一成契约里的字段。
|
||
|
||
grain / time_field / numerator / denominator / dimensions / filters 是文档起草里的业务描述。前端草稿页已经在画这几行，
|
||
这里不能像以前那样重建指标时丢掉；空值不写成空字符串（没有这个键就没有内容），dimensions / filters 只收非空字符串。
|
||
"""
|
||
name = str(item.get("name") or "").strip()
|
||
expression = item.get("expression")
|
||
if not isinstance(expression, str) or not _EXPRESSION.match(expression.strip()):
|
||
problems.append({"name": name or f"第 {position} 条",
|
||
"reason": "expression 必须是 cell(nodes.fetch, 0, '列名')，列名用 sql 的输出列名"})
|
||
return None
|
||
definition = item.get("definition") if isinstance(item.get("definition"), Mapping) else {}
|
||
citations = definition.get("citations") if isinstance(definition.get("citations"), list) else []
|
||
decimals = item.get("decimals")
|
||
if isinstance(decimals, bool) or not isinstance(decimals, int):
|
||
decimals = None
|
||
synonyms = item.get("synonyms")
|
||
questions = item.get("open_questions")
|
||
grain, time_field = _text(item.get("grain")), _text(item.get("time_field"))
|
||
numerator, denominator = _text(item.get("numerator")), _text(item.get("denominator"))
|
||
dimensions, filters = _texts(item.get("dimensions")), _texts(item.get("filters"))
|
||
return {
|
||
"id": str(item.get("id") or "").strip(),
|
||
"name": name,
|
||
"synonyms": [s.strip() for s in synonyms if isinstance(s, str) and s.strip()]
|
||
if isinstance(synonyms, list) else [],
|
||
"unit": item.get("unit") if isinstance(item.get("unit"), str) else "",
|
||
"decimals": decimals,
|
||
"expression": expression.strip(),
|
||
"contract": {"shape": "scalar"},
|
||
**({"grain": grain} if grain else {}),
|
||
**({"time_field": time_field} if time_field else {}),
|
||
**({"numerator": numerator} if numerator else {}),
|
||
**({"denominator": denominator} if denominator else {}),
|
||
**({"dimensions": dimensions} if dimensions else {}),
|
||
**({"filters": filters} if filters else {}),
|
||
"definition": {
|
||
"text": definition.get("text") if isinstance(definition.get("text"), str) else "",
|
||
"scope": definition.get("scope") if isinstance(definition.get("scope"), str) else "",
|
||
"exclusions": [str(v) for v in definition.get("exclusions") or [] if isinstance(v, str)]
|
||
if isinstance(definition.get("exclusions"), list) else [],
|
||
"owner": definition.get("owner") if isinstance(definition.get("owner"), str) else "",
|
||
"effective_from": definition.get("effective_from")
|
||
if isinstance(definition.get("effective_from"), str) else "",
|
||
"citations": citations,
|
||
},
|
||
"open_questions": [q for q in questions if isinstance(q, str) and q.strip()]
|
||
if isinstance(questions, list) else [],
|
||
}
|
||
|
||
|
||
def normalize_plan(raw: Any, chunks: Mapping[str, dict], *, allowed_tables: set[str]) -> tuple[dict, list[dict]]:
|
||
"""校验模型交回的草案。返回 (可落库的 plan, problems)。
|
||
|
||
problems 的形状是 {"name": …, "reason": …}，name 能认出是哪一条就用名字，认不出按序号说。硬规则：SQL 单条只读、表都在
|
||
目录里；每条指标的 expression 是 cell(nodes.fetch, 0, '列名')；至少一条引用能在原文里逐字核对上（复用
|
||
docdraft.normalize_metrics 的核对）。个别指标不合格只丢那一条，problems 里写清为什么；SQL 不合格时 plan["sql"] 为空，
|
||
由调用方决定整份草稿算失败。
|
||
"""
|
||
body = raw if isinstance(raw, Mapping) else {}
|
||
problems: list[dict] = []
|
||
entry_fields = _entry_fields(body.get("entry_fields"), problems)
|
||
if len(entry_fields) > MAX_ENTRY_FIELDS:
|
||
# 不自动删：入口删了 SQL 还引用着，跑不通。照实说清楚，让人把不常变的条件写成固定条件重起一份
|
||
problems.append({"name": "入口字段",
|
||
"reason": f"入口有 {len(entry_fields)} 个（上限 {MAX_ENTRY_FIELDS}）：同一维度只该有一个，"
|
||
f"不常变的条件应当写死在 sql 里（用「固定条件」重起一份）"})
|
||
|
||
sql = body.get("sql")
|
||
sql_text = sql.strip() if isinstance(sql, str) else ""
|
||
if not sql_text:
|
||
problems.append({"name": "sql", "reason": "没有给出取数 SQL"})
|
||
else:
|
||
reason = _sql_problem(sql_text, allowed_tables)
|
||
if reason:
|
||
problems.append({"name": "sql", "reason": reason})
|
||
sql_text = ""
|
||
else:
|
||
declared = {field["name"] for field in entry_fields}
|
||
missing = sorted(set(_INPUT.findall(sql_text)) - declared)
|
||
if missing:
|
||
problems.append({"name": "sql",
|
||
"reason": f"SQL 用了没有声明的入口字段：{'、'.join(missing)}"})
|
||
sql_text = ""
|
||
|
||
checked, dropped = docdraft.normalize_metrics(body.get("metrics"), dict(chunks))
|
||
problems.extend(dropped)
|
||
metrics: list[dict] = []
|
||
seen_ids: set[str] = set()
|
||
for position, item in enumerate(checked, start=1):
|
||
metric = _metric(item, position, problems)
|
||
if metric is None:
|
||
continue
|
||
if not metric["id"] or metric["id"] in seen_ids:
|
||
metric["id"] = f"m{len(metrics) + 1}"
|
||
while metric["id"] in seen_ids:
|
||
metric["id"] += "_"
|
||
seen_ids.add(metric["id"])
|
||
metrics.append(metric)
|
||
if not metrics:
|
||
problems.append({"name": "metrics", "reason": "没有一条指标通过校验"})
|
||
|
||
questions = body.get("open_questions")
|
||
plan_out = {
|
||
"entry_fields": entry_fields,
|
||
"sql": sql_text,
|
||
"metrics": metrics,
|
||
"open_questions": [q.strip() for q in questions if isinstance(q, str) and q.strip()]
|
||
if isinstance(questions, list) else [],
|
||
}
|
||
return plan_out, problems
|
||
|
||
|
||
# ==========================================================================
|
||
# 建图（确定性，和 certify.build_graph 同一个形状）
|
||
# ==========================================================================
|
||
|
||
|
||
def _blocks(value: Any) -> list[dict]:
|
||
"""blocks 可以是 [{"sql", "entry_fields", "metrics"}]，也可以是老调用的单个 plan 字典（当成只有一组）。"""
|
||
if isinstance(value, Mapping):
|
||
return [dict(value)]
|
||
return [dict(block) for block in value or [] if isinstance(block, Mapping)]
|
||
|
||
|
||
def _merged_type(left: str, right: str) -> str:
|
||
"""两组把同一个入口字段声明成不同（已规范的）类型时，合并成什么：两边对调结果一样，与哪一组先出现无关（R02）。
|
||
|
||
- 整数和数都是数（records.store.union_compatible 也把这两类算一列）：并成数——数能填小数，能把整数那一组的值收下，
|
||
反过来 3.5 会被「整数」拒收；
|
||
- 别的组合一律并成文字：文字跟谁都能并；是或否和数/整数不能互相表示（true 不是 1），退回文字让人填，不拿某一组的
|
||
说法去判另一组的值。合并结果永远落在 ENTRY_TYPES 里。"""
|
||
if left == right:
|
||
return left
|
||
return "number" if {left, right} <= {"number", "integer"} else "string"
|
||
|
||
|
||
def _merge_fields(target: dict[str, dict], value: Any, notes: list[str], conflicts: set[str]) -> None:
|
||
"""入口字段按 name 合并，冲突取更保守的一边，**不抛错**。
|
||
|
||
同一份文档分几轮起草，各组对同一个字段的声明常常对不上：一组说必填、另一组给了缺省；一组按数、另一组按文本。
|
||
这不是错误——它们本来就是从不同批指标里各看了一次。合并时必填取并集（有一组要就必须给），类型按 _merged_type 的
|
||
规则定（数+整数并成数，别的并成文字，结果与各组先后无关）。默认值冲突时清掉默认值、要求人填，不能自动把某一组
|
||
的范围用于其他组；0 和 False 都是有效值。
|
||
|
||
早先这里抛 ValueError，结果一份 34 条指标的草稿因为一个字段的 required 不同就整份作废——代价完全不成比例。
|
||
"""
|
||
for field in value if isinstance(value, list) else []:
|
||
if not isinstance(field, Mapping):
|
||
continue
|
||
name = str(field.get("name") or "").strip()
|
||
if not name:
|
||
continue
|
||
item = dict(field)
|
||
item["name"] = name
|
||
item["type"] = _entry_type(item.get("type"))
|
||
old = target.get(name)
|
||
if old is None:
|
||
target[name] = item
|
||
continue
|
||
merged = dict(old)
|
||
old_type, new_type = _entry_type(old.get("type")), item["type"]
|
||
merged["type"] = _merged_type(old_type, new_type)
|
||
if old_type != new_type:
|
||
notes.append(f"入口字段「{name}」在各组里的类型不一致（{old_type} / {new_type}），按「{merged['type']}」合并")
|
||
if bool(old.get("required")) != bool(item.get("required")):
|
||
merged["required"] = bool(old.get("required")) or bool(item.get("required"))
|
||
notes.append(f"入口字段「{name}」在各组里的必填不一致，合并后按"
|
||
f"{'必填' if merged['required'] else '可选'}")
|
||
old_default, new_default = merged.get("default"), item.get("default")
|
||
if old_default not in (None, "") and new_default not in (None, "") \
|
||
and (old_default != new_default or isinstance(old_default, bool) != isinstance(new_default, bool)):
|
||
conflicts.add(name)
|
||
notes.append(f"入口字段「{name}」在各组里的默认值不一致，已取消默认值，请确认后填写")
|
||
if name in conflicts:
|
||
merged.pop("default", None)
|
||
merged["required"] = True
|
||
elif old_default in (None, "") and new_default not in (None, ""):
|
||
merged["default"] = item["default"]
|
||
if not merged.get("description") and item.get("description"):
|
||
merged["description"] = item["description"]
|
||
target[name] = merged
|
||
|
||
|
||
def merge_entry_fields(blocks: Any) -> tuple[list[dict], list[str]]:
|
||
"""各组草案的入口字段合成一份：(字段, 不一致的说明)。调用方把说明摆给起草的人看。"""
|
||
target: dict[str, dict] = {}
|
||
notes: list[str] = []
|
||
conflicts: set[str] = set()
|
||
for block in _blocks(blocks):
|
||
_merge_fields(target, block.get("entry_fields"), notes, conflicts)
|
||
return list(target.values()), notes
|
||
|
||
|
||
def unique_metric_ids(blocks: Any) -> None:
|
||
"""把各组指标的 id 在全图里改成唯一，就地改调用方的 blocks。调用方在建图前调一次。
|
||
|
||
模型按组起草，每组都从 m1 开始；多组拼一张图时 id 会撞。build_graph 只改图里的副本，草稿 payload 和 groups
|
||
仍留着撞车的 id，页面的「指标 → 试跑值」就会张冠李戴（第二组的 m1 显示成第一组的数）。这里先改同一份 blocks，
|
||
图、payload、groups 和试跑值用同一组 id；本来就唯一的保持原 id 不动。
|
||
"""
|
||
metrics = [metric for block in _blocks(blocks) for metric in block.get("metrics") or []
|
||
if isinstance(metric, dict)]
|
||
ids = [str(metric.get("id") or "").strip() for metric in metrics]
|
||
if "" not in ids and len(set(ids)) == len(ids):
|
||
return
|
||
for position, metric in enumerate(metrics, start=1):
|
||
metric["id"] = f"m{position}"
|
||
|
||
|
||
def build_graph(blocks: Any, *, name: str, question: str = "", datasource: str = "") -> dict:
|
||
"""各组草案 → 一张草稿图：start(input) → fetch…(tool，每组一个) → card(metrics) → write(report) → out(output)。
|
||
|
||
blocks 是 [{"sql", "entry_fields", "metrics"}]，也兼容老调用的单个 plan 字典（当成只有一组）。一组时取数节点仍叫
|
||
fetch，多组叫 fetch_1、fetch_2…（和 certify.build_graph 一致）；第 i 组指标表达式里的 nodes.fetch 改写成对应的
|
||
节点 id，已经带下标的（nodes.fetch_2）不动。指标 id 跨组重名时按 m1…mN 全局重排，口径卡和出口写重排后的 id——
|
||
重名会让口径卡读错值，宁可换号也不能让两个指标共用一个 id。入口字段按 name 合并，冲突取更保守的一边并记一条
|
||
note（见 merge_entry_fields），**不抛错**：一个字段的声明差异不该把整份草稿判死。datasource 是取数节点指向的
|
||
数据源名，调用方按这次的数据源补，也可以写在某一组里；各组指向不同数据源时用第一组的（草稿图只连一个数据源）。
|
||
|
||
调用两次结果逐字一致：节点 id、坐标都由输入决定，没有时间戳。报告配置、出口契约照 certify.build_graph 的写法，
|
||
发布门禁认的就是这一套（REPORT_POLICY、metrics_from、required）。
|
||
"""
|
||
from app.engine.governance import REPORT_POLICY
|
||
from app.engine.layout import auto_layout
|
||
from app.engine.schema import GraphSpec
|
||
from app.ledger.certify import REPORT_INSTRUCTIONS, REPORT_INSTRUCTIONS_PARAMS
|
||
from app.tools.datasource import QUERY_PREFIX
|
||
|
||
groups = _blocks(blocks) or [{}]
|
||
multi = len(groups) > 1
|
||
|
||
def nid(index: int) -> str:
|
||
return f"fetch_{index + 1}" if multi else "fetch"
|
||
|
||
merged: dict[str, dict] = {}
|
||
conflicts: set[str] = set()
|
||
metrics: list[dict] = []
|
||
fetch_nodes: list[dict[str, Any]] = []
|
||
source = str(datasource or "")
|
||
for index, group in enumerate(groups):
|
||
_merge_fields(merged, group.get("entry_fields"), [], conflicts)
|
||
node_id = nid(index)
|
||
for metric in group.get("metrics") or []:
|
||
if not isinstance(metric, Mapping):
|
||
continue
|
||
item = dict(metric)
|
||
item["expression"] = _FETCH_REF.sub(f"nodes.{node_id}", str(item.get("expression") or ""))
|
||
metrics.append(item)
|
||
source = source or str(group.get("datasource") or "")
|
||
fetch_nodes.append({"id": node_id, "type": "tool", "position": {"x": 0, "y": 0},
|
||
"data": {"label": f"取数 {index + 1}" if multi else "取数",
|
||
"config": {"tool": f"{QUERY_PREFIX}{source}",
|
||
"args": {"sql": str(group.get("sql") or "")}}}})
|
||
|
||
ids = [str(metric.get("id") or "").strip() for metric in metrics]
|
||
if "" in ids or len(set(ids)) != len(ids):
|
||
for position, metric in enumerate(metrics, start=1):
|
||
metric["id"] = f"m{position}"
|
||
ids = [metric["id"] for metric in metrics]
|
||
|
||
fields = list(merged.values())
|
||
report_cfg: dict[str, Any] = {}
|
||
if fields:
|
||
report_cfg["instructions"] = REPORT_INSTRUCTIONS_PARAMS.format(
|
||
params="，".join(f"{f['name']} = {{{{ input.{f['name']} }}}}" for f in fields))
|
||
else:
|
||
report_cfg["instructions"] = REPORT_INSTRUCTIONS.format(question=question or name or "（问题）")
|
||
report_cfg.update({key: value for key, value, _ in REPORT_POLICY})
|
||
report_cfg["metrics_from"] = ["card"]
|
||
contract: dict[str, Any] = {"report_from": "write", "metrics_from": "card", "required": ids}
|
||
chain = [
|
||
{"id": "start", "type": "input", "position": {"x": 0, "y": 0},
|
||
"data": {"label": "入口", "config": {"fields": fields}}},
|
||
*fetch_nodes,
|
||
{"id": "card", "type": "metrics", "position": {"x": 0, "y": 0},
|
||
"data": {"label": "口径卡", "config": {"caliber": (name.strip() or question or "口径")[:40],
|
||
"caliber_version": "v1", "metrics": metrics}}},
|
||
{"id": "write", "type": "report", "position": {"x": 0, "y": 0},
|
||
"data": {"label": "报告", "config": report_cfg}},
|
||
{"id": "out", "type": "output", "position": {"x": 0, "y": 0},
|
||
"data": {"label": "出口", "config": {"fields": [{"name": "报告", "value": "{{ nodes.write.text }}"}],
|
||
"contract": contract}}},
|
||
]
|
||
edges = [{"source": a["id"], "target": b["id"]} for a, b in zip(chain, chain[1:])]
|
||
return auto_layout(GraphSpec.model_validate({"nodes": chain, "edges": edges})).model_dump(mode="json")
|
||
|
||
|
||
__all__ = ["ENTRY_TYPES", "MAX_INPUT_CHARS", "MAX_SQL_CHARS", "MAX_TABLES_CHARS", "METRIC_FIELDS",
|
||
"PROMPT_VERSION", "build_graph", "fit_material", "normalize_plan", "plan", "plan_groups",
|
||
"prompt", "repair_group", "unique_metric_ids"]
|