feat(webui): add temporary chat mode

This commit is contained in:
Xubin Ren
2026-08-07 17:07:15 +08:00
committed by chengyongru
parent 332c159b93
commit 324a61dff1
41 changed files with 1500 additions and 119 deletions
+12
View File
@@ -346,6 +346,18 @@ class TestBuildSystemPrompt:
assert "## AGENTS.md" not in result
assert "[Archived Context Summary]" not in result
def test_can_exclude_long_term_memory_without_changing_agent_identity(self, tmp_path):
builder = _builder(tmp_path)
builder.memory.write_memory("# Memory\n- private detail")
result = builder.build_system_prompt(
include_long_term_memory=False,
include_memory_recent_history=False,
)
assert "private detail" not in result
assert "workspace" in result.lower()
# ---------------------------------------------------------------------------
# build_messages
+150
View File
@@ -0,0 +1,150 @@
import asyncio
from unittest.mock import AsyncMock, MagicMock
import pytest
from nanobot.agent.loop import AgentLoop
from nanobot.bus.events import InboundMessage
from nanobot.bus.queue import MessageBus
from nanobot.providers.base import GenerationSettings, LLMResponse
def _message(key: str, content: str) -> InboundMessage:
return InboundMessage(
channel="websocket",
sender_id="user",
chat_id=key.removeprefix("websocket:"),
content=content,
session_key_override=key,
transient_session=True,
)
@pytest.mark.asyncio
async def test_temporary_chat_keeps_agent_capabilities_and_only_live_history(tmp_path) -> None:
(tmp_path / "AGENTS.md").write_text("project instruction", encoding="utf-8")
provider = MagicMock()
provider.get_default_model.return_value = "test-model"
provider.generation = GenerationSettings()
provider.chat_with_retry = AsyncMock(side_effect=[
LLMResponse(content="first answer", usage={}),
LLMResponse(content="second answer", usage={}),
])
loop = AgentLoop(
bus=MessageBus(),
provider=provider,
workspace=tmp_path,
model="test-model",
)
loop.context.memory.write_memory("# Memory\n- private remembered detail")
key = "websocket:temporary-test"
loop.sessions.get_or_create_transient(key)
await loop._process_message(_message(key, "first question"))
await loop._process_message(_message(key, "second question"))
first_call, second_call = provider.chat_with_retry.await_args_list
assert first_call.kwargs["tools"]
assert "project instruction" in str(first_call.kwargs["messages"])
assert "private remembered detail" not in str(first_call.kwargs["messages"])
assert "first answer" in str(second_call.kwargs["messages"])
session = loop.sessions.get_cached(key)
assert session is not None
assert [message["role"] for message in session.messages] == [
"user",
"assistant",
"user",
"assistant",
]
assert loop.sessions.read_session_file(key) is None
@pytest.mark.asyncio
async def test_temporary_chat_compacts_only_in_memory(tmp_path) -> None:
provider = MagicMock()
provider.get_default_model.return_value = "test-model"
provider.generation = GenerationSettings(max_tokens=256)
provider.estimate_prompt_tokens.return_value = (100, "test")
provider.chat_with_retry = AsyncMock(
return_value=LLMResponse(content="Earlier temporary decisions.", usage={})
)
loop = AgentLoop(
bus=MessageBus(),
provider=provider,
workspace=tmp_path,
model="test-model",
context_window_tokens=4096,
)
key = "websocket:temporary-compact"
session = loop.sessions.get_or_create_transient(key)
for index in range(6):
session.add_message("user", f"question {index}")
session.add_message("assistant", f"answer {index}")
await loop.consolidator.maybe_consolidate_by_tokens(
session,
runtime=loop.runtime_for_session(session),
replay_max_messages=4,
)
assert session.last_consolidated > 0
assert len(session.messages) == 12
assert session.metadata["_last_summary"]["text"] == "Earlier temporary decisions."
assert loop.sessions.read_session_file(key) is None
assert loop.context.memory.read_unprocessed_history(since_cursor=0) == []
_, summary = loop.auto_compact.prepare_session(session, key)
assert summary is not None
@pytest.mark.asyncio
async def test_temporary_chat_uses_the_regular_compaction_pipeline(tmp_path) -> None:
provider = MagicMock()
provider.get_default_model.return_value = "test-model"
provider.generation = GenerationSettings()
provider.chat_with_retry = AsyncMock(
return_value=LLMResponse(content="answer", usage={})
)
loop = AgentLoop(
bus=MessageBus(),
provider=provider,
workspace=tmp_path,
model="test-model",
)
loop.consolidator.maybe_consolidate_by_tokens = AsyncMock()
key = "websocket:temporary-pipeline"
loop.sessions.get_or_create_transient(key)
await loop._process_message(_message(key, "question"))
loop.consolidator.maybe_consolidate_by_tokens.assert_awaited()
@pytest.mark.asyncio
async def test_discarded_temporary_turn_cannot_create_a_session_file(tmp_path) -> None:
provider_started = asyncio.Event()
provider = MagicMock()
provider.get_default_model.return_value = "test-model"
provider.generation = GenerationSettings()
async def block_provider(**_kwargs):
provider_started.set()
await asyncio.Event().wait()
provider.chat_with_retry = AsyncMock(side_effect=block_provider)
loop = AgentLoop(
bus=MessageBus(),
provider=provider,
workspace=tmp_path,
model="test-model",
)
key = "websocket:temporary-cancelled"
loop.sessions.get_or_create_transient(key)
task = asyncio.create_task(loop._dispatch(_message(key, "private")))
loop._active_tasks.setdefault(key, set()).add(task)
await provider_started.wait()
assert loop.sessions.discard_transient(key) is True
assert await loop._cancel_active_tasks(key) == 1
assert task.cancelled()
assert loop.sessions.read_session_file(key) is None
+26
View File
@@ -231,6 +231,32 @@ async def test_read_session_filters_by_query_and_returns_recent_matches(tmp_path
]
@pytest.mark.asyncio
async def test_read_session_searches_current_temporary_chat_in_memory(tmp_path):
manager = SessionManager(tmp_path)
session = manager.get_or_create_transient("websocket:temporary-current")
session.messages = [
{"role": "user", "content": "the launch codename is firefly"},
{"role": "assistant", "content": "I will remember that during this chat"},
{"role": "user", "content": "unrelated recent message"},
]
session.last_consolidated = 2
with _webui_request("websocket:temporary-current"):
result = _decode(await ReadSessionTool(manager).execute(
session_key="current",
query="firefly",
))
assert result["session_key"] == "websocket:temporary-current"
assert result["session_ref"] is None
assert result["title"] == "Current conversation"
assert [message["content"] for message in result["messages"]] == [
"the launch codename is firefly",
]
assert manager.read_session_file("websocket:temporary-current") is None
@pytest.mark.asyncio
async def test_read_session_reports_invalid_requests(tmp_path):
with _webui_request():