fix(agent): refresh goal continuation context

This commit is contained in:
chengyongru
2026-06-16 11:19:24 +08:00
committed by Xubin Ren
parent 3ce0cd972e
commit 27d869d3cc
4 changed files with 95 additions and 12 deletions
+35
View File
@@ -953,6 +953,41 @@ async def test_process_message_uses_explicit_session_metadata_for_goal_context(
assert GOAL_STATE_KEY not in kwargs["session_metadata"]
@pytest.mark.asyncio
async def test_run_agent_loop_goal_continue_message_reads_latest_metadata(
tmp_path: Path,
) -> None:
from nanobot.agent.runner import AgentRunResult
loop = _make_full_loop(tmp_path)
session = loop.sessions.get_or_create("websocket:late-goal")
seen: dict[str, str | None] = {}
async def fake_run(spec):
assert callable(spec.goal_continue_message)
session.metadata[GOAL_STATE_KEY] = {
"status": "active",
"objective": "Goal created during this runner call.",
}
seen["goal_continue"] = spec.goal_continue_message()
return AgentRunResult(
final_content="ok",
messages=[{"role": "assistant", "content": "ok"}],
)
loop.runner.run = fake_run # type: ignore[method-assign]
await loop._run_agent_loop(
[],
session=session,
channel="websocket",
chat_id="late-goal",
session_key=session.key,
)
assert "Goal created during this runner call." in (seen["goal_continue"] or "")
def test_set_tool_context_uses_effective_key_for_spawn_tool(tmp_path: Path) -> None:
loop = _make_full_loop(tmp_path)
spawn_tool = loop.tools.get("spawn")