Grayscale/nanobot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(agent): capture runtime at turn admission

Browse Source
This commit is contained in:
chengyongru
2026-07-10 17:54:34 +08:00
committed by Xubin Ren
parent 45ace23580
commit 198fd9f869
13 changed files with 209 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/agent/test_loop_progress.py
+13 -4
View File
@@ -76,7 +76,9 @@ class TestToolEventProgress:
) -> None:
progress.append((content, tool_hint, tool_events))
final_content, _, _, _, _ = await loop._run_agent_loop([], on_progress=on_progress)
final_content, _, _, _, _ = await loop._run_agent_loop(
[], runtime=loop.llm_runtime(), on_progress=on_progress
)
assert final_content == "Done"
assert progress == [
@@ -145,7 +147,9 @@ class TestToolEventProgress:
if file_edit_events:
file_events.extend(file_edit_events)
final_content, _, _, _, _ = await loop._run_agent_loop([], on_progress=on_progress)
final_content, _, _, _, _ = await loop._run_agent_loop(
[], runtime=loop.llm_runtime(), on_progress=on_progress
)
assert final_content == "Done"
assert [event["phase"] for event in file_events] == ["start", "end"]
@@ -213,7 +217,9 @@ class TestToolEventProgress:
prepare_file_edit_trackers,
)
final_content, _, _, _, _ = await loop._run_agent_loop([], on_progress=on_progress)
final_content, _, _, _, _ = await loop._run_agent_loop(
[], runtime=loop.llm_runtime(), on_progress=on_progress
)
assert final_content == "Done"
assert target.read_text(encoding="utf-8") == "new\n"
@@ -249,7 +255,9 @@ class TestToolEventProgress:
if file_edit_events:
file_events.extend(file_edit_events)
await loop._run_agent_loop([], on_progress=on_progress)
await loop._run_agent_loop(
[], runtime=loop.llm_runtime(), on_progress=on_progress
)
assert file_events == []
@@ -623,6 +631,7 @@ class TestToolEventProgress:
final_content, _, _, _, _ = await loop._run_agent_loop(
[],
runtime=loop.llm_runtime(),
on_progress=on_progress,
on_stream=on_stream,
)