mirror of
https://github.com/ThearchyHelios/AgentLab.git
synced 2026-11-15 04:51:57 +00:00
ThearchyHelios
60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
"""Copilot 的数据源上下文注入。
|
||
|
||
这组守的是"Copilot 知道有哪些库、写得出能跑的 SQL"。断了的表现很隐蔽：
|
||
它不会报错，只会退回让用户自己写 SQL 的代码节点——接了数据库等于白接。
|
||
|
||
按库的大小定「给详细摘要还是只列表名」的那一套（预算、40 张表的阈值）已经拿掉：system 里每个库一行概览，
|
||
要看表时模型自己 look（look 文档 D3，见 test_copilot_table_pick.py）。这里留着结构化输出和摘要本身的几条。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
from types import SimpleNamespace
|
||
|
||
from app.api.copilot import GRAPH_SCHEMA
|
||
from app.data.introspect import summary
|
||
|
||
|
||
def _source(n_tables: int, *, name: str = "warehouse", schema: str | None = "ANALYTICS"):
|
||
tables = {
|
||
f"t{i}": {
|
||
"qualified": f"{schema}.t{i}" if schema else f"t{i}",
|
||
"schema": schema,
|
||
"columns": [
|
||
{"name": f"c{j}", "type": "VARCHAR", "nullable": True, "comment": None}
|
||
for j in range(20)
|
||
],
|
||
"primary_key": [], "comment": None, "is_view": True,
|
||
}
|
||
for i in range(n_tables)
|
||
}
|
||
return SimpleNamespace(
|
||
id="x", name=name, kind="oracle", readonly=True,
|
||
description="测试库", options={}, schema_cache={"tables": tables, "schema": schema},
|
||
)
|
||
|
||
|
||
def test_graph_schema_allows_arbitrary_config_keys():
|
||
"""config 的键随节点类型千变万化，没法枚举成 properties。
|
||
|
||
不显式写 additionalProperties=True，某些 provider 的结构化输出会按
|
||
"不允许额外字段"处理，把 config 整个过滤成 {}——生成的图每个节点都是
|
||
空壳，校验挂在"工具节点还没选工具"。这条回归过一次，代价是整条非流式
|
||
生成路径不可用。
|
||
"""
|
||
node = GRAPH_SCHEMA["properties"]["nodes"]["items"]
|
||
assert node["properties"]["config"]["additionalProperties"] is True
|
||
assert "config" in node["required"]
|
||
|
||
|
||
def test_compact_mode_is_substantially_smaller():
|
||
"""压缩比要真的有效，否则降级没意义。实测约 35%。"""
|
||
full = summary(_source(53))
|
||
compact = summary(_source(53), detail=False)
|
||
assert len(compact) < len(full) * 0.5
|
||
|
||
|
||
def test_summary_without_schema_uses_bare_names():
|
||
"""没有 schema 概念的库不要硬拼前缀，否则生成的 SQL 反而错。"""
|
||
text = summary(_source(2, schema=None))
|
||
assert "t0" in text and "None.t0" not in text
|