mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 00:40:45 +08:00
oliver
acfb2cf73b
- Normalize select id,content rows from dict_row vs sqlite Row in history_tool_result_compact. - Expand memory_wiki_search ToolSpec to match wiki_search parameters; add schema regression test. Co-authored-by: Cursor <cursoragent@cursor.com>
56 lines
2.3 KiB
Python
56 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
from svc.persistence.sqlite_store import SqliteStore
|
|
from runtime.chat.history_tool_result_compact import (
|
|
_chat_message_id_content,
|
|
compact_tool_results_in_session_history,
|
|
)
|
|
|
|
|
|
def test_chat_message_id_content_dict_row() -> None:
|
|
assert _chat_message_id_content({"id": 42, "content": "hello"}) == (42, "hello")
|
|
|
|
|
|
def test_chat_message_id_content_tuple_like() -> None:
|
|
assert _chat_message_id_content((9, "z")) == (9, "z")
|
|
|
|
|
|
def test_compact_tool_results_in_session_history_rewrites_large_tool_message(tmp_path) -> None: # noqa: ANN001
|
|
db = tmp_path / "ops.sqlite"
|
|
store = SqliteStore(str(db))
|
|
sess = store.create_session("t")
|
|
store.add_message(session_id=sess.id, role="user", content="hi", event_type="user_text")
|
|
big = {"ok": True, "data": "x" * 10000}
|
|
store.add_message(session_id=sess.id, role="tool", content=json.dumps(big), event_type="tool_result")
|
|
|
|
out = compact_tool_results_in_session_history(store=store, session_id=sess.id, cap_chars=8000, limit_messages=200)
|
|
assert out.ok is True
|
|
assert out.compacted_tool_messages == 1
|
|
assert out.rewritten_all_tool_messages >= 0
|
|
|
|
msgs = store.get_messages(session_id=sess.id, limit=50)
|
|
tool = next((m for m in msgs if str(getattr(m, "role", "")) == "tool"), None)
|
|
assert tool is not None
|
|
txt = str(getattr(tool, "content", "") or "")
|
|
obj = json.loads(txt)
|
|
assert obj.get("_tool_result_guarded") is True
|
|
assert int(obj.get("guard_cap_chars") or 0) == 8000
|
|
|
|
|
|
def test_compact_tool_results_rewrite_all_small_rows(tmp_path) -> None: # noqa: ANN001
|
|
db = tmp_path / "ops.sqlite"
|
|
store = SqliteStore(str(db))
|
|
sess = store.create_session("t2")
|
|
store.add_message(session_id=sess.id, role="tool", content=json.dumps({"ok": True, "value": "small"}), event_type="tool_result")
|
|
|
|
out = compact_tool_results_in_session_history(store=store, session_id=sess.id, cap_chars=8000, limit_messages=200, rewrite_all=True)
|
|
assert out.ok is True
|
|
assert out.rewritten_all_tool_messages >= 1
|
|
msgs = store.get_messages(session_id=sess.id, limit=10)
|
|
tool = next((m for m in msgs if str(getattr(m, "role", "")) == "tool"), None)
|
|
assert tool is not None
|
|
obj = json.loads(str(getattr(tool, "content", "") or "{}"))
|
|
assert obj.get("_history_full_rewrite") is True
|
|
|