mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-04-30 14:56:01 +00:00
cc5a666d5d
Follow-up to #3212, fully backward compatible: - Extract the 14-day staleness threshold as `_STALE_THRESHOLD_DAYS` module constant and pass it into the Phase 1 prompt template as `{{ stale_threshold_days }}`. The number lived in three places before (code threshold, prompt instruction, docstring); now there is one. - Add `DreamConfig.annotate_line_ages` (default True = current behavior) and propagate it through `Dream.__init__` and the gateway wiring in cli/commands.py. Gives users a knob to disable the feature without a code patch if an LLM reacts poorly to the `← Nd` suffix. - Harden `_annotate_with_ages` against dirty working trees: when HEAD blob line count disagrees with the working-tree content length, skip annotation entirely instead of assigning ages to the wrong lines. The previous `i >= len(ages)` guard only handled one direction of the mismatch. - Inline-comment the `max_iterations` 10→15 bump with a pointer to exp002 so future blame has context. - Add 4 regression tests: end-to-end `← 30d` reaches prompt, 14/15 threshold boundary, `annotate_line_ages=False` bypasses git entirely (verified via `assert_not_called`), length-mismatch defense, and template-var rendering. Made-with: Cursor
259 lines
12 KiB
Python
259 lines
12 KiB
Python
"""Tests for the Dream class — two-phase memory consolidation via AgentRunner."""
|
|
|
|
import pytest
|
|
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
from nanobot.agent.memory import Dream, MemoryStore
|
|
from nanobot.agent.runner import AgentRunResult
|
|
from nanobot.agent.skills import BUILTIN_SKILLS_DIR
|
|
from nanobot.utils.gitstore import LineAge
|
|
|
|
|
|
@pytest.fixture
|
|
def store(tmp_path):
|
|
s = MemoryStore(tmp_path)
|
|
s.write_soul("# Soul\n- Helpful")
|
|
s.write_user("# User\n- Developer")
|
|
s.write_memory("# Memory\n- Project X active")
|
|
return s
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_provider():
|
|
p = MagicMock()
|
|
p.chat_with_retry = AsyncMock()
|
|
return p
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_runner():
|
|
return MagicMock()
|
|
|
|
|
|
@pytest.fixture
|
|
def dream(store, mock_provider, mock_runner):
|
|
d = Dream(store=store, provider=mock_provider, model="test-model", max_batch_size=5)
|
|
d._runner = mock_runner
|
|
return d
|
|
|
|
|
|
def _make_run_result(
|
|
stop_reason="completed",
|
|
final_content=None,
|
|
tool_events=None,
|
|
usage=None,
|
|
):
|
|
return AgentRunResult(
|
|
final_content=final_content or stop_reason,
|
|
stop_reason=stop_reason,
|
|
messages=[],
|
|
tools_used=[],
|
|
usage={},
|
|
tool_events=tool_events or [],
|
|
)
|
|
|
|
|
|
class TestDreamRun:
|
|
async def test_noop_when_no_unprocessed_history(self, dream, mock_provider, mock_runner, store):
|
|
"""Dream should not call LLM when there's nothing to process."""
|
|
result = await dream.run()
|
|
assert result is False
|
|
mock_provider.chat_with_retry.assert_not_called()
|
|
mock_runner.run.assert_not_called()
|
|
|
|
async def test_calls_runner_for_unprocessed_entries(self, dream, mock_provider, mock_runner, store):
|
|
"""Dream should call AgentRunner when there are unprocessed history entries."""
|
|
store.append_history("User prefers dark mode")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="New fact")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result(
|
|
tool_events=[{"name": "edit_file", "status": "ok", "detail": "memory/MEMORY.md"}],
|
|
))
|
|
result = await dream.run()
|
|
assert result is True
|
|
mock_runner.run.assert_called_once()
|
|
spec = mock_runner.run.call_args[0][0]
|
|
assert spec.max_iterations == 10
|
|
assert spec.fail_on_tool_error is False
|
|
|
|
async def test_advances_dream_cursor(self, dream, mock_provider, mock_runner, store):
|
|
"""Dream should advance the cursor after processing."""
|
|
store.append_history("event 1")
|
|
store.append_history("event 2")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="Nothing new")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
await dream.run()
|
|
assert store.get_last_dream_cursor() == 2
|
|
|
|
async def test_compacts_processed_history(self, dream, mock_provider, mock_runner, store):
|
|
"""Dream should compact history after processing."""
|
|
store.append_history("event 1")
|
|
store.append_history("event 2")
|
|
store.append_history("event 3")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="Nothing new")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
await dream.run()
|
|
# After Dream, cursor is advanced and 3, compact keeps last max_history_entries
|
|
entries = store.read_unprocessed_history(since_cursor=0)
|
|
assert all(e["cursor"] > 0 for e in entries)
|
|
|
|
async def test_skill_phase_uses_builtin_skill_creator_path(self, dream, mock_provider, mock_runner, store):
|
|
"""Dream should point skill creation guidance at the builtin skill-creator template."""
|
|
store.append_history("Repeated workflow one")
|
|
store.append_history("Repeated workflow two")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="[SKILL] test-skill: test description")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
|
|
await dream.run()
|
|
|
|
spec = mock_runner.run.call_args[0][0]
|
|
system_prompt = spec.initial_messages[0]["content"]
|
|
expected = str(BUILTIN_SKILLS_DIR / "skill-creator" / "SKILL.md")
|
|
assert expected in system_prompt
|
|
|
|
async def test_skill_write_tool_accepts_workspace_relative_skill_path(self, dream, store):
|
|
"""Dream skill creation should allow skills/<name>/SKILL.md relative to workspace root."""
|
|
write_tool = dream._tools.get("write_file")
|
|
assert write_tool is not None
|
|
|
|
result = await write_tool.execute(
|
|
path="skills/test-skill/SKILL.md",
|
|
content="---\nname: test-skill\ndescription: Test\n---\n",
|
|
)
|
|
|
|
assert "Successfully wrote" in result
|
|
assert (store.workspace / "skills" / "test-skill" / "SKILL.md").exists()
|
|
|
|
async def test_phase1_prompt_includes_line_age_annotations(self, dream, mock_provider, mock_runner, store):
|
|
"""Phase 1 prompt should have per-line age suffixes in MEMORY.md when git is available."""
|
|
store.append_history("some event")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="[SKIP]")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
|
|
# Init git so line_ages works
|
|
store.git.init()
|
|
store.git.auto_commit("initial memory state")
|
|
|
|
await dream.run()
|
|
|
|
# The MEMORY.md section should not crash and should contain the memory content
|
|
call_args = mock_provider.chat_with_retry.call_args
|
|
user_msg = call_args.kwargs.get("messages", call_args[1].get("messages"))[1]["content"]
|
|
assert "## Current MEMORY.md" in user_msg
|
|
|
|
async def test_phase1_annotates_only_memory_not_soul_or_user(self, dream, mock_provider, mock_runner, store):
|
|
"""SOUL.md and USER.md should never have age annotations — they are permanent."""
|
|
store.append_history("some event")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="[SKIP]")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
|
|
store.git.init()
|
|
store.git.auto_commit("initial state")
|
|
|
|
await dream.run()
|
|
|
|
call_args = mock_provider.chat_with_retry.call_args
|
|
user_msg = call_args.kwargs.get("messages", call_args[1].get("messages"))[1]["content"]
|
|
# The ← suffix should only appear in MEMORY.md section
|
|
memory_section = user_msg.split("## Current MEMORY.md")[1].split("## Current SOUL.md")[0]
|
|
soul_section = user_msg.split("## Current SOUL.md")[1].split("## Current USER.md")[0]
|
|
user_section = user_msg.split("## Current USER.md")[1]
|
|
# SOUL and USER should not contain age arrows
|
|
assert "\u2190" not in soul_section
|
|
assert "\u2190" not in user_section
|
|
|
|
async def test_phase1_prompt_works_without_git(self, dream, mock_provider, mock_runner, store):
|
|
"""Phase 1 should work fine even if git is not initialized (no age annotations)."""
|
|
store.append_history("some event")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="[SKIP]")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
|
|
await dream.run()
|
|
|
|
# Should still succeed — just without age annotations
|
|
mock_provider.chat_with_retry.assert_called_once()
|
|
call_args = mock_provider.chat_with_retry.call_args
|
|
user_msg = call_args.kwargs.get("messages", call_args[1].get("messages"))[1]["content"]
|
|
assert "## Current MEMORY.md" in user_msg
|
|
|
|
async def test_phase1_prompt_carries_age_suffix_for_stale_lines(
|
|
self, dream, mock_provider, mock_runner, store,
|
|
):
|
|
"""End-to-end: ages >14d must appear verbatim in the LLM prompt, ages ≤14d must not."""
|
|
# MEMORY.md fixture has 2 non-blank lines ("# Memory" and "- Project X active").
|
|
# Inject four ages to cover threshold boundaries: >14 suffix, ==14 no suffix, <14 no suffix.
|
|
store.write_memory("# Memory\n- Project X active\n- fresh item\n- edge case line")
|
|
store.append_history("some event")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="[SKIP]")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
|
|
fake_ages = [
|
|
LineAge(age_days=30), # "# Memory" → should get ← 30d
|
|
LineAge(age_days=20), # "- Project X..." → should get ← 20d
|
|
LineAge(age_days=14), # "- fresh item" → ==14, threshold is strictly >14, no suffix
|
|
LineAge(age_days=5), # "- edge case..." → no suffix
|
|
]
|
|
with patch.object(store.git, "line_ages", return_value=fake_ages):
|
|
await dream.run()
|
|
|
|
call_args = mock_provider.chat_with_retry.call_args
|
|
user_msg = call_args.kwargs.get("messages", call_args[1].get("messages"))[1]["content"]
|
|
memory_section = user_msg.split("## Current MEMORY.md")[1].split("## Current SOUL.md")[0]
|
|
assert "\u2190 30d" in memory_section
|
|
assert "\u2190 20d" in memory_section
|
|
assert "\u2190 14d" not in memory_section
|
|
assert "\u2190 5d" not in memory_section
|
|
|
|
async def test_phase1_skips_annotation_when_disabled(
|
|
self, dream, mock_provider, mock_runner, store,
|
|
):
|
|
"""`annotate_line_ages=False` must bypass the git lookup entirely and keep MEMORY.md raw."""
|
|
store.append_history("some event")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="[SKIP]")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
|
|
dream.annotate_line_ages = False
|
|
# line_ages must be bypassed entirely — verify with a spy rather than a
|
|
# raising side_effect, because _annotate_with_ages catches Exception
|
|
# (which swallows AssertionError) and would hide an accidental call.
|
|
with patch.object(store.git, "line_ages") as mock_line_ages:
|
|
await dream.run()
|
|
mock_line_ages.assert_not_called()
|
|
|
|
call_args = mock_provider.chat_with_retry.call_args
|
|
user_msg = call_args.kwargs.get("messages", call_args[1].get("messages"))[1]["content"]
|
|
assert "\u2190" not in user_msg
|
|
|
|
async def test_phase1_skips_annotation_on_line_ages_length_mismatch(
|
|
self, dream, mock_provider, mock_runner, store,
|
|
):
|
|
"""If ages length != lines length (dirty working tree), skip annotation instead of mis-tagging."""
|
|
# MEMORY.md has 2 non-blank lines but we hand back only 1 age → mismatch.
|
|
store.append_history("some event")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="[SKIP]")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
|
|
with patch.object(store.git, "line_ages", return_value=[LineAge(age_days=999)]):
|
|
await dream.run()
|
|
|
|
call_args = mock_provider.chat_with_retry.call_args
|
|
user_msg = call_args.kwargs.get("messages", call_args[1].get("messages"))[1]["content"]
|
|
memory_section = user_msg.split("## Current MEMORY.md")[1].split("## Current SOUL.md")[0]
|
|
# No age arrow at all — we refused to annotate rather than tag the wrong line.
|
|
assert "\u2190" not in memory_section
|
|
|
|
async def test_phase1_prompt_uses_threshold_from_template_var(
|
|
self, dream, mock_provider, mock_runner, store,
|
|
):
|
|
"""System prompt should reference the stale-threshold constant, not a hardcoded 14."""
|
|
store.append_history("some event")
|
|
mock_provider.chat_with_retry.return_value = MagicMock(content="[SKIP]")
|
|
mock_runner.run = AsyncMock(return_value=_make_run_result())
|
|
|
|
await dream.run()
|
|
|
|
system_msg = mock_provider.chat_with_retry.call_args.kwargs["messages"][0]["content"]
|
|
# The template renders with stale_threshold_days=14 → LLM must see "N>14"
|
|
assert "N>14" in system_msg
|
|
|