mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 03:30:48 +08:00
oliver
605751b6a6
Prefer report tools without hard-hiding CLI, keep soft CLI budgets, store compact+result_ref for the model, and clear frozen wire after MCP sync. Co-authored-by: Cursor <cursoragent@cursor.com>
94 lines
3.6 KiB
Python
94 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
from runtime import direct_loop as dl
|
|
|
|
|
|
class _DummyStore:
|
|
def __init__(self, settings: dict[str, str] | None = None):
|
|
self.settings = dict(settings or {})
|
|
|
|
def get_setting(self, key: str) -> str:
|
|
return str(self.settings.get(key, ""))
|
|
|
|
def list_mcp_servers(self, *, enabled_only: bool = False):
|
|
del enabled_only
|
|
return []
|
|
|
|
def list_mcp_server_tools(self, *, server_id: str):
|
|
del server_id
|
|
return []
|
|
|
|
|
|
def test_tool_wire_freeze_default_on(monkeypatch) -> None:
|
|
monkeypatch.setattr(dl, "_prepare_llm_tools", lambda **kwargs: [])
|
|
monkeypatch.setenv("AIA_TOOL_WIRE_FROZEN_ON_STARTUP", "")
|
|
store = _DummyStore()
|
|
_ = dl.warm_tool_wire_cache(store=store, tools=object(), base_url="", roles=["generalist"])
|
|
st = dl.tool_wire_freeze_status(store=store)
|
|
assert st["enabled"] is True
|
|
assert st["frozen"] is True
|
|
|
|
|
|
def test_tool_wire_freeze_disabled_by_setting(monkeypatch) -> None:
|
|
monkeypatch.setattr(dl, "_prepare_llm_tools", lambda **kwargs: [])
|
|
store = _DummyStore({"AIA_TOOL_WIRE_FROZEN_ON_STARTUP": "0"})
|
|
_ = dl.warm_tool_wire_cache(store=store, tools=object(), base_url="", roles=["generalist"])
|
|
st = dl.tool_wire_freeze_status(store=store)
|
|
assert st["enabled"] is False
|
|
assert st["frozen"] is False
|
|
|
|
|
|
def test_warm_tool_wire_cache_clears_frozen_stale_entries(monkeypatch) -> None:
|
|
"""Regression: prewarm must rebuild after MCP sync; frozen mode used to return stale wire."""
|
|
calls = {"n": 0}
|
|
|
|
def _prep(**_kwargs):
|
|
calls["n"] += 1
|
|
return [{"type": "function", "function": {"name": f"tool_{calls['n']}"}}]
|
|
|
|
monkeypatch.setattr(dl, "_prepare_llm_tools", _prep)
|
|
monkeypatch.setenv("AIA_TOOL_WIRE_FROZEN_ON_STARTUP", "1")
|
|
store = _DummyStore({"AIA_TOOL_WIRE_FROZEN_ON_STARTUP": "1"})
|
|
with dl._TOOL_WIRE_CACHE_LOCK:
|
|
dl._TOOL_WIRE_FROZEN_SIGNATURE = "rt=1|stale"
|
|
dl._TOOL_WIRE_CACHE.clear()
|
|
dl._TOOL_WIRE_CACHE["poison"] = (0.0, [{"type": "function", "function": {"name": "old"}}])
|
|
out = dl.warm_tool_wire_cache(store=store, tools=object(), base_url="", roles=["generalist", "ops"])
|
|
assert calls["n"] == 2
|
|
assert int(out.get("cache_cleared") or 0) == 1
|
|
assert "poison" not in dl._TOOL_WIRE_CACHE
|
|
st = dl.tool_wire_freeze_status(store=store)
|
|
assert st["frozen"] is True
|
|
assert st["last_warm_count"] == 2
|
|
|
|
|
|
def test_invalidate_tool_wire_cache_clears_freeze(monkeypatch) -> None:
|
|
monkeypatch.setattr(dl, "_prepare_llm_tools", lambda **kwargs: [])
|
|
store = _DummyStore({"AIA_TOOL_WIRE_FROZEN_ON_STARTUP": "1"})
|
|
_ = dl.warm_tool_wire_cache(store=store, tools=object(), base_url="", roles=["generalist"])
|
|
assert dl.tool_wire_freeze_status(store=store)["frozen"] is True
|
|
out = dl.invalidate_tool_wire_cache(reason="test")
|
|
assert out["ok"] is True
|
|
assert dl.tool_wire_freeze_status(store=store)["frozen"] is False
|
|
|
|
|
|
def test_mcp_tools_fingerprint_changes_with_catalog() -> None:
|
|
class _Store(_DummyStore):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.tools = [{"tool_name": "a"}]
|
|
|
|
def list_mcp_servers(self, *, enabled_only: bool = False):
|
|
del enabled_only
|
|
return [{"server_id": "netx"}]
|
|
|
|
def list_mcp_server_tools(self, *, server_id: str):
|
|
del server_id
|
|
return list(self.tools)
|
|
|
|
store = _Store()
|
|
fp1 = dl._mcp_tools_fingerprint(store)
|
|
store.tools = [{"tool_name": "a"}, {"tool_name": "b"}]
|
|
fp2 = dl._mcp_tools_fingerprint(store)
|
|
assert fp1 != fp2
|
|
assert fp1 != "x"
|