mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 03:30:48 +08:00
oliver
76f62d8fac
Align tool validation with playbook recipes, add turn checklist/idle guard, and cap interim WA status ticks so multi-hop CLI turns stop flooding the group. Co-authored-by: Cursor <cursoragent@cursor.com>
130 lines
4.4 KiB
Python
130 lines
4.4 KiB
Python
"""Playbook contracts ↔ tool schema alignment + idle guard."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from runtime.chat.turn_idle_guard import TurnIdleTracker
|
|
from runtime.tools.experts.network_ops.ume_alarm_xlsx_report import ume_alarm_xlsx_report_tool
|
|
from runtime.tools.playbook_contracts import (
|
|
build_turn_checklist,
|
|
enrich_invalid_arguments_with_playbook,
|
|
playbook_example_for_tool,
|
|
schema_playbook_mismatches,
|
|
short_intent_first_step,
|
|
)
|
|
from runtime.tools.public.write_xlsx_tool import write_xlsx_tool
|
|
from runtime.tools.tool_validation import format_invalid_arguments_error
|
|
|
|
|
|
def test_ume_alarm_xlsx_playbook_examples_match_schema() -> None:
|
|
tool = ume_alarm_xlsx_report_tool()
|
|
issues = schema_playbook_mismatches(tool.name, tool.parameters)
|
|
assert issues == [], issues
|
|
|
|
|
|
def test_write_xlsx_playbook_examples_match_schema() -> None:
|
|
tool = write_xlsx_tool()
|
|
issues = schema_playbook_mismatches(tool.name, tool.parameters)
|
|
assert issues == [], issues
|
|
|
|
|
|
def test_short_intent_first_step_fiber_cut() -> None:
|
|
step = short_intent_first_step("fiber_cut")
|
|
assert step is not None
|
|
tool, example = step
|
|
assert tool == "ume_alarm_xlsx_report"
|
|
assert example.get("mode") == "fiber_cut"
|
|
assert example.get("deliverable") is True
|
|
|
|
|
|
def test_checklist_contains_step1_tool() -> None:
|
|
text = build_turn_checklist(intent="offline", lang="en", goal="offline NE list")
|
|
assert "ume_alarm_xlsx_report" in text
|
|
assert "mode=\"offline\"" in text or "offline" in text
|
|
assert "Turn checklist" in text
|
|
|
|
|
|
def test_invalid_args_enriched_with_playbook_example() -> None:
|
|
tool = ume_alarm_xlsx_report_tool()
|
|
raw = format_invalid_arguments_error(
|
|
tool.parameters or {},
|
|
"'bogus' is not one of ['list', 'aggregate_by_host', 'fiber_cut', 'offline']",
|
|
lang="en",
|
|
tool_name="ume_alarm_xlsx_report",
|
|
intent="fiber_cut",
|
|
)
|
|
assert raw.get("playbook_example") is True
|
|
assert raw.get("example", {}).get("mode") == "fiber_cut"
|
|
assert "Playbook recipe" in str(raw.get("hint") or "")
|
|
|
|
|
|
def test_enrich_invalid_arguments_mcp_alias() -> None:
|
|
base = {
|
|
"ok": False,
|
|
"error_code": "tool_invalid_arguments",
|
|
"example": {"severity": "x"},
|
|
"hint": "fix it",
|
|
}
|
|
out = enrich_invalid_arguments_with_playbook(
|
|
base,
|
|
tool_name="mcp__netx__aggregateUmeAlarms",
|
|
intent="alarm_tally",
|
|
)
|
|
assert out["example"].get("top_ne") == 20
|
|
assert out.get("playbook_example") is True
|
|
|
|
|
|
def test_idle_guard_nudges_short_intent_narration() -> None:
|
|
tracker = TurnIdleTracker(lang="en", short_intent="fiber_cut")
|
|
d1 = tracker.decide_after_assistant_no_tools()
|
|
assert d1.action == "nudge"
|
|
assert "Idle guard" in d1.nudge_text
|
|
assert "ume_alarm_xlsx_report" in d1.nudge_text
|
|
d2 = tracker.decide_after_assistant_no_tools()
|
|
assert d2.action == "continue"
|
|
|
|
|
|
def test_idle_guard_early_finalize_on_idle_streak() -> None:
|
|
tracker = TurnIdleTracker(lang="en", short_intent="alarm_tally", max_idle_rounds=2)
|
|
# Round 1: all failures
|
|
s1 = tracker.record_from_traces(
|
|
round_idx=1,
|
|
had_tool_calls=True,
|
|
round_traces=[{"name": "ume_alarm_xlsx_report", "ok": False}],
|
|
results_by_id={
|
|
"1": (
|
|
{
|
|
"ok": False,
|
|
"error_code": "tool_invalid_arguments",
|
|
"failure_class": "schema_validation",
|
|
},
|
|
10,
|
|
)
|
|
},
|
|
)
|
|
d1 = tracker.decide_after_tools(s1)
|
|
assert d1.action in {"nudge", "continue", "early_finalize"}
|
|
# Round 2: still no success
|
|
s2 = tracker.record_from_traces(
|
|
round_idx=2,
|
|
had_tool_calls=True,
|
|
round_traces=[{"name": "ume_alarm_xlsx_report", "ok": False}],
|
|
results_by_id={
|
|
"2": (
|
|
{
|
|
"ok": False,
|
|
"error_code": "identical_retry_blocked",
|
|
"failure_class": "retry_guard",
|
|
},
|
|
5,
|
|
)
|
|
},
|
|
)
|
|
d2 = tracker.decide_after_tools(s2)
|
|
assert d2.action == "early_finalize"
|
|
assert d2.reason in {"idle_no_progress", "schema_fail_budget"}
|
|
|
|
|
|
def test_playbook_example_for_mcp_namespaced_tool() -> None:
|
|
ex = playbook_example_for_tool("mcp__netx__execManagedNe")
|
|
assert ex is not None
|
|
assert "commands" in ex
|