mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 03:13:19 +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>
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from runtime.chat.tool_result_store import (
|
|
attach_result_ref,
|
|
load_tool_result_blob,
|
|
save_tool_result_blob,
|
|
)
|
|
from runtime.tools.public.fetch_tool_result_tool import fetch_tool_result_tool
|
|
|
|
|
|
def test_save_and_fetch_tool_result_blob(monkeypatch, tmp_path: Path) -> None:
|
|
monkeypatch.setattr("runtime.chat.tool_result_store.attachments_dir", lambda: tmp_path)
|
|
big = {"ok": True, "rows": ["x" * 100 for _ in range(80)]}
|
|
ref = save_tool_result_blob(session_id="sess-1", tool_call_id="tc1", result=big)
|
|
assert ref and ref.startswith("tr:")
|
|
loaded = load_tool_result_blob(ref, session_id="sess-1")
|
|
assert loaded["ok"] is True
|
|
assert loaded["result"]["ok"] is True
|
|
assert loaded["truncated"] is False
|
|
|
|
|
|
def test_attach_result_ref_adds_fetch_hint() -> None:
|
|
out = attach_result_ref({"ok": True, "_truncated_for_llm": True, "preview": "p"}, result_ref="tr:abcd")
|
|
assert out["result_ref"] == "tr:abcd"
|
|
assert out["fetch_tool"] == "fetch_tool_result"
|
|
assert "fetch_tool_result" in str(out.get("hint") or "")
|
|
|
|
|
|
def test_fetch_tool_result_public_tool(monkeypatch, tmp_path: Path) -> None:
|
|
monkeypatch.setattr("runtime.chat.tool_result_store.attachments_dir", lambda: tmp_path)
|
|
ref = save_tool_result_blob(
|
|
session_id="sess-2",
|
|
tool_call_id="tc2",
|
|
result={"ok": True, "data": "hello" * 2000},
|
|
force=True,
|
|
)
|
|
assert ref
|
|
monkeypatch.setattr(
|
|
"runtime.tools.public.fetch_tool_result_tool.current_tool_lane_sessions",
|
|
lambda: (None, "sess-2"),
|
|
)
|
|
tool = fetch_tool_result_tool()
|
|
out = tool.handler({"result_ref": ref})
|
|
assert out["ok"] is True
|
|
assert out["result_ref"] == ref
|