mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 03:30:48 +08:00
fix(chat): compress-history on PostgreSQL dict rows
- 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>
This commit is contained in:
parent
0dc0c4663f
commit
acfb2cf73b
4 changed files with 56 additions and 4 deletions
|
|
@ -3,7 +3,18 @@ from __future__ import annotations
|
|||
import json
|
||||
|
||||
from svc.persistence.sqlite_store import SqliteStore
|
||||
from runtime.chat.history_tool_result_compact import compact_tool_results_in_session_history
|
||||
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
|
||||
|
|
|
|||
|
|
@ -12,6 +12,22 @@ from runtime.tools.public.memory_wiki_tools import (
|
|||
)
|
||||
|
||||
|
||||
def test_memory_wiki_search_tool_schema_exposes_wiki_search_args() -> None:
|
||||
props = memory_wiki_search_tool().parameters.get("properties") or {}
|
||||
for key in (
|
||||
"query",
|
||||
"limit",
|
||||
"offset",
|
||||
"context_lines",
|
||||
"is_regex",
|
||||
"case_sensitive",
|
||||
"path_prefix",
|
||||
"expand_query",
|
||||
"max_rounds",
|
||||
):
|
||||
assert key in props, f"missing schema property: {key}"
|
||||
|
||||
|
||||
def test_memory_wiki_tools_handlers_load_without_dataclass_module_error() -> None:
|
||||
status = memory_wiki_status_tool().handler({})
|
||||
assert isinstance(status, dict)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue