mirror of
https://github.com/ThearchyHelios/AgentLab.git
synced 2026-11-15 19:51:59 +00:00
ThearchyHelios
198 lines
10 KiB
Python
198 lines
10 KiB
Python
"""驳回出口之后，待办、通知、定时任务的 Webhook 照实说「被驳回，转到 X」（docs/design/orchestration-round/00-总方案.md §3）。
|
||
|
||
实例页已经这么说（test_reject_exit.test_the_instance_says_rejected_and_where_it_went）。这里对的是另外几处：
|
||
- 待办：中断内容带着驳回之后去哪（reject_route）：接了驳回出口的写转到哪几步，没接的写运行就此结束，审批模式没接、没设
|
||
驳回即终止的写这一路到此为止。待办卡的提示、驳回后的提示、处理过的结果照它说，不再一律写「转入驳回分支」「本次运行终止」；
|
||
- 子工作流里的关卡走了驳回出口：父运行的终态事件照实带上（「子工作流 › 把关」被驳回，转到「子工作流 › 退回」），实例页、
|
||
通知说这一句，不写内部 id、不写成「已完成」；
|
||
- 定时任务的 Webhook：途经驳回、照驳回那条路成功结束的，标题和状态写途经驳回，另起一行写谁被驳回、转到哪一步。
|
||
图和数据都是虚构的。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import asyncio
|
||
|
||
import pytest
|
||
from sqlalchemy import select
|
||
|
||
from app.db.base import SessionLocal
|
||
from app.db.models import Run, RunEvent, Task
|
||
from app.engine.runner import run_manager
|
||
from tests import ledger_samples as S
|
||
|
||
|
||
@pytest.fixture(autouse=True)
|
||
async def engine_up():
|
||
await run_manager.setup()
|
||
yield
|
||
await run_manager.shutdown()
|
||
|
||
|
||
def node(nid, ntype, label=None, **config):
|
||
return {"id": nid, "type": ntype, "position": {"x": 0, "y": 0}, "data": {"label": label or nid, "config": config}}
|
||
|
||
|
||
START = node("start", "input", fields=[{"name": "q"}])
|
||
DONE = node("done", "output", "办结", fields=[{"name": "结果", "value": "通过了"}])
|
||
BACK = node("back", "output", "退回申请人", fields=[{"name": "结果", "value": "退回：{{ nodes.gate.note }}"}])
|
||
|
||
|
||
def gated(gate: dict, *, exit_to: dict | None = BACK) -> dict:
|
||
normal = "approved" if gate["data"]["config"].get("mode") == "approve" else "out"
|
||
nodes = [START, gate, DONE] + ([exit_to] if exit_to else [])
|
||
edges = [{"source": "start", "target": "gate"}, {"source": "gate", "target": "done", "sourceHandle": normal}]
|
||
if exit_to:
|
||
edges.append({"source": "gate", "target": exit_to["id"], "sourceHandle": "rejected"})
|
||
return {"nodes": nodes, "edges": edges}
|
||
|
||
|
||
async def _wait(run_id: str, statuses: tuple[str, ...], timeout: float = 20.0) -> Run:
|
||
deadline = asyncio.get_event_loop().time() + timeout
|
||
while asyncio.get_event_loop().time() < deadline:
|
||
async with SessionLocal() as session:
|
||
row = await session.get(Run, run_id)
|
||
if row is not None and row.status in statuses and not run_manager.is_active(run_id):
|
||
return row
|
||
await asyncio.sleep(0.05)
|
||
raise AssertionError(f"运行 {run_id} 没有进入 {statuses}")
|
||
|
||
|
||
async def _parked(graph: dict) -> tuple[Run, dict]:
|
||
"""发起、停在关卡上，交回运行和那条待办的中断内容。"""
|
||
run = await run_manager.start(graph=graph, input_payload={"q": "x"})
|
||
await _wait(run.id, ("interrupted",))
|
||
await run_manager.wait_idle(run.id)
|
||
async with SessionLocal() as session:
|
||
[task] = list((await session.execute(select(Task).where(Task.run_id == run.id))).scalars())
|
||
from app.db.models import Approval
|
||
|
||
approval = await session.get(Approval, task.approval_id)
|
||
return run, dict(approval.payload or {})
|
||
|
||
|
||
APPROVE = node("gate", "human", "把关", mode="approve", title="可以吗")
|
||
INPUT = node("gate", "human", "补充", mode="input", title="补一下")
|
||
TOOL = node("gate", "tool", "算一下", tool="calculator", args={"expression": "1+1"}, approval="always")
|
||
|
||
|
||
@pytest.mark.parametrize("gate", [APPROVE, INPUT, TOOL], ids=["approve", "input", "tool"])
|
||
async def test_the_task_says_where_a_rejection_goes(gate):
|
||
_, payload = await _parked(gated(gate))
|
||
assert payload["reject_route"] == {"to": [{"node_id": "back", "label": "退回申请人"}]}
|
||
|
||
|
||
@pytest.mark.parametrize("gate", [INPUT, TOOL], ids=["input", "tool"])
|
||
async def test_without_the_exit_the_task_says_the_run_ends(gate):
|
||
_, payload = await _parked(gated(gate, exit_to=None))
|
||
assert payload["reject_route"] == {"ends": "run"}
|
||
|
||
|
||
async def test_an_approval_without_the_exit_says_this_path_stops_here():
|
||
_, payload = await _parked(gated(APPROVE, exit_to=None))
|
||
assert payload["reject_route"] == {"ends": "path"}
|
||
stop = node("gate", "human", "把关", mode="approve", title="可以吗", stop_on_reject=True)
|
||
_, payload = await _parked(gated(stop))
|
||
assert payload["reject_route"] == {"ends": "run"}, "驳回即终止优先：驳回出口那条线走不到"
|
||
|
||
|
||
async def test_an_agent_tool_approval_has_no_route():
|
||
"""agent 自己调工具的审批不是关卡：驳回了 agent 照常往下想，中断内容不带 reject_route。"""
|
||
from app.engine.replay import reject_route
|
||
from app.engine.schema import GraphSpec
|
||
|
||
spec = GraphSpec.model_validate(gated(node("gate", "agent", "助手", prompt="x")))
|
||
|
||
class Ctx:
|
||
class run: # noqa: N801 - 只要 spec 和 global_id
|
||
pass
|
||
|
||
ctx = Ctx()
|
||
ctx.node = spec.node_map()["gate"]
|
||
ctx.run.spec = spec
|
||
assert reject_route(ctx) is None
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# 子工作流里的关卡走了驳回出口
|
||
# --------------------------------------------------------------------------
|
||
|
||
|
||
async def _sub_rejected() -> Run:
|
||
inner = {"nodes": [node("s", "input"), node("gate", "human", "把关", mode="approve", title="可以吗"),
|
||
node("ok", "output", "通过", fields=[{"name": "r", "value": "ok"}]),
|
||
node("back", "output", "退回", fields=[{"name": "r", "value": "退回：{{ nodes.gate.note }}"}])],
|
||
"edges": [{"source": "s", "target": "gate"},
|
||
{"source": "gate", "target": "ok", "sourceHandle": "approved"},
|
||
{"source": "gate", "target": "back", "sourceHandle": "rejected"}]}
|
||
sub_id = await S.publish("复核子流程", graph=inner, level="draft")
|
||
graph = S.chain(node("start", "input"), node("sub", "subgraph", "复核子流程", workflow_id=sub_id, workflow_version=1),
|
||
node("out", "output", "成果", fields=[{"name": "r", "value": "{{ nodes.sub.output.r }}"}]))
|
||
run = await run_manager.start(graph=graph, input_payload={})
|
||
await _wait(run.id, ("interrupted",))
|
||
await run_manager.wait_idle(run.id)
|
||
await run_manager.resume(run.id, {"approved": False, "note": "不对"}, actor="张工")
|
||
return await _wait(run.id, ("cancelled", "failed", "succeeded"))
|
||
|
||
|
||
async def test_a_reject_exit_inside_a_subworkflow_is_named_in_the_parent():
|
||
from app.apps.resolve import _trails, instance_state
|
||
|
||
final = await _sub_rejected()
|
||
assert final.status == "succeeded", final.error
|
||
assert final.output == {"r": "退回：不对"}
|
||
async with SessionLocal() as session:
|
||
[ended] = list((await session.execute(select(RunEvent).where(
|
||
RunEvent.run_id == final.id, RunEvent.type == "run.finished"))).scalars())
|
||
run = await session.get(Run, final.id)
|
||
state = instance_state(run, (await _trails(session, [run]))[run.id])
|
||
[via] = ended.data["rejected_via"]
|
||
assert (via["node_id"], via["label"]) == ("sub/gate", "复核子流程 › 把关")
|
||
assert via["to"] == [{"node_id": "sub/back", "label": "复核子流程 › 退回"}]
|
||
# 实例页、通知（notify_run 取的就是这一句）：照实说被驳回、转到哪，不写「sub/gate」，不画成已完成
|
||
assert state == {"code": "cancelled", "text": "「复核子流程 › 把关」被驳回，转到「复核子流程 › 退回」"}
|
||
|
||
|
||
async def test_the_inner_task_names_the_route_with_the_subworkflow():
|
||
inner = {"nodes": [node("s", "input"), node("gate", "human", "补充", mode="input", title="补一下"),
|
||
node("back", "output", "退回", fields=[{"name": "r", "value": "x"}]),
|
||
node("ok", "output", "通过", fields=[{"name": "r", "value": "y"}])],
|
||
"edges": [{"source": "s", "target": "gate"}, {"source": "gate", "target": "ok"},
|
||
{"source": "gate", "target": "back", "sourceHandle": "rejected"}]}
|
||
sub_id = await S.publish("补充子流程", graph=inner, level="draft")
|
||
graph = S.chain(node("start", "input"), node("sub", "subgraph", "补充子流程", workflow_id=sub_id, workflow_version=1),
|
||
node("out", "output", fields=[{"name": "r", "value": "{{ nodes.sub.output.r }}"}]))
|
||
_, payload = await _parked(graph)
|
||
assert payload["reject_route"] == {"to": [{"node_id": "sub/back", "label": "补充子流程 › 退回"}]}
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# 定时任务的 Webhook
|
||
# --------------------------------------------------------------------------
|
||
|
||
|
||
def test_the_webhook_says_a_run_went_through_a_rejection():
|
||
from app.scheduler.webhook import Notice, message
|
||
|
||
via = [{"node_id": "gate", "label": "把关", "kind": "approve", "decision": "reject", "actor": "张工", "note": "不对",
|
||
"to": [{"node_id": "back", "label": "退回申请人"}]}]
|
||
title, lines, _ = message(Notice(event="finished", schedule_id="s1", schedule_name="周报", timezone="UTC",
|
||
status="succeeded", rejected_via=via))
|
||
assert "途经驳回" in title
|
||
said = dict(lines)
|
||
assert said["状态"] == "已完成（途经驳回）"
|
||
assert said["驳回"] == "「把关」被驳回，转到「退回申请人」"
|
||
# 没途经驳回的照旧
|
||
title, lines, _ = message(Notice(event="finished", schedule_id="s1", schedule_name="周报", timezone="UTC",
|
||
status="succeeded"))
|
||
assert title == "定时任务「周报」运行完成" and dict(lines)["状态"] == "已完成" and "驳回" not in dict(lines)
|
||
|
||
|
||
async def test_the_webhook_notice_reads_the_route_from_the_end_event():
|
||
from app.scheduler.webhook import _rejected_via_of
|
||
|
||
final = await _sub_rejected()
|
||
async with SessionLocal() as session:
|
||
run = await session.get(Run, final.id)
|
||
via = await _rejected_via_of(session, run)
|
||
assert [v["label"] for v in via] == ["复核子流程 › 把关"]
|