mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 03:30:48 +08:00
oliver
366bc2a320
Keep field essentials (xlsx/deliverable, wiki, FS read, schedule) plus full MCP; drop unused public noise and ensure binding edits clear frozen tool wire. Co-authored-by: Cursor <cursoragent@cursor.com>
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from runtime.tools.base import ToolSpec
|
|
from runtime.tools.ops_system_tool_allowlist import (
|
|
filter_collected_tool_sources,
|
|
filter_system_tool_specs,
|
|
ops_system_tool_allowlist,
|
|
should_slim_system_tools_for_specialist,
|
|
)
|
|
|
|
|
|
def _spec(name: str) -> ToolSpec:
|
|
return ToolSpec(
|
|
name=name,
|
|
description="t",
|
|
parameters={"type": "object", "properties": {}},
|
|
handler=lambda _a: {"ok": True},
|
|
)
|
|
|
|
|
|
def test_ops_slim_applies_only_to_ops(monkeypatch) -> None:
|
|
monkeypatch.delenv("AIA_OPS_SYSTEM_TOOL_SLIM", raising=False)
|
|
assert should_slim_system_tools_for_specialist("ops")
|
|
assert not should_slim_system_tools_for_specialist("generalist")
|
|
assert not should_slim_system_tools_for_specialist("memory")
|
|
|
|
|
|
def test_ops_slim_can_disable(monkeypatch) -> None:
|
|
monkeypatch.setenv("AIA_OPS_SYSTEM_TOOL_SLIM", "0")
|
|
assert not should_slim_system_tools_for_specialist("ops")
|
|
|
|
|
|
def test_filter_system_keeps_allowlisted_only() -> None:
|
|
tools = [
|
|
_spec("write_xlsx"),
|
|
_spec("git_status"),
|
|
_spec("bailian_webparser"),
|
|
_spec("fetch_tool_result"),
|
|
_spec("system_time"),
|
|
]
|
|
kept = filter_system_tool_specs(tools, specialist="ops")
|
|
names = {t.name for t in kept}
|
|
assert names == {"write_xlsx", "fetch_tool_result", "system_time"}
|
|
assert len(filter_system_tool_specs(tools, specialist="generalist")) == 5
|
|
|
|
|
|
def test_filter_collected_keeps_mcp_and_expert() -> None:
|
|
collected = [
|
|
("public", _spec("git_status")),
|
|
("public", _spec("write_xlsx")),
|
|
("expert", _spec("ume_alarm_xlsx_report")),
|
|
("mcp", _spec("mcp__netx__execManagedNe")),
|
|
("plugin", _spec("cloudflare_image_generate")),
|
|
]
|
|
out = filter_collected_tool_sources(collected, specialist="ops")
|
|
names = [(s, t.name) for s, t in out]
|
|
assert ("mcp", "mcp__netx__execManagedNe") in names
|
|
assert ("expert", "ume_alarm_xlsx_report") in names
|
|
assert ("public", "write_xlsx") in names
|
|
assert ("public", "git_status") not in names
|
|
assert ("plugin", "cloudflare_image_generate") not in names
|
|
|
|
|
|
def test_allowlist_env_override(monkeypatch) -> None:
|
|
monkeypatch.setenv("AIA_OPS_SYSTEM_TOOL_ALLOWLIST", "system_time,write_xlsx")
|
|
assert ops_system_tool_allowlist() == frozenset({"system_time", "write_xlsx"})
|