mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-09-04 02:01:48 +03:00
refactor(agent): remove loop iteration state (#5549)
* refactor(agent): remove loop iteration state * docs(my): remove stale iteration guidance
This commit is contained in:
@@ -31,33 +31,18 @@ def test_turn_hook_context_preserves_legacy_positional_arguments(tmp_path) -> No
|
||||
assert context.attributes == {}
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_turn_hook_builder_runs_progress_hook_before_extra_hooks() -> None:
|
||||
events: list[str] = []
|
||||
|
||||
hook = build_agent_turn_hook(AgentTurnHookSpec(
|
||||
on_iteration=lambda iteration: events.append(f"progress:{iteration}"),
|
||||
registered_hooks=[RecordingHook(events)],
|
||||
))
|
||||
|
||||
await hook.before_iteration(AgentHookContext(iteration=2, messages=[]))
|
||||
|
||||
assert events == ["progress:2", "hook:2"]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_turn_hook_builder_runs_registered_hooks_before_turn_hooks() -> None:
|
||||
events: list[str] = []
|
||||
|
||||
hook = build_agent_turn_hook(AgentTurnHookSpec(
|
||||
on_iteration=lambda iteration: events.append(f"progress:{iteration}"),
|
||||
registered_hooks=[RecordingHook(events, "registered")],
|
||||
turn_hooks=[RecordingHook(events, "turn")],
|
||||
))
|
||||
|
||||
await hook.before_iteration(AgentHookContext(iteration=2, messages=[]))
|
||||
|
||||
assert events == ["progress:2", "registered:2", "turn:2"]
|
||||
assert events == ["registered:2", "turn:2"]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@@ -75,7 +60,6 @@ async def test_turn_hook_builder_runs_factories_with_matching_registration_order
|
||||
return _create
|
||||
|
||||
hook = build_agent_turn_hook(AgentTurnHookSpec(
|
||||
on_iteration=lambda iteration: events.append(f"progress:{iteration}"),
|
||||
channel="websocket",
|
||||
chat_id="chat-1",
|
||||
message_id="msg-1",
|
||||
@@ -92,7 +76,6 @@ async def test_turn_hook_builder_runs_factories_with_matching_registration_order
|
||||
await hook.before_iteration(AgentHookContext(iteration=2, messages=[]))
|
||||
|
||||
assert events == [
|
||||
"progress:2",
|
||||
"registered_factory:2",
|
||||
"registered:2",
|
||||
"turn_factory:2",
|
||||
|
||||
Reference in New Issue
Block a user