wuyang/zk-data-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add File-history snapshot ids and replay summaries for file edits

Browse Source
This commit is contained in:
Abdelrahman Abdallah
2026-04-02 22:19:32 +02:00
parent 9d90d6a10a
commit 5ad10c660c
5 changed files with 283 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_agent_runtime.py
+74
View File
@@ -749,7 +749,76 @@ class AgentRuntimeTests(unittest.TestCase):
stored = load_agent_session(result.session_id or '', directory=session_dir)
self.assertEqual(len(result.file_history), 1)
self.assertEqual(result.file_history[0]['path'], 'out.txt')
self.assertEqual(result.file_history[0]['history_kind'], 'file_change')
self.assertIn('history_entry_id', result.file_history[0])
self.assertIn('after_snapshot_id', result.file_history[0])
self.assertEqual(stored.file_history[0]['action'], 'write_file')
self.assertEqual(stored.file_history[0]['after_snapshot_id'], result.file_history[0]['after_snapshot_id'])
def test_agent_streams_write_file_tool_output(self) -> None:
responses = [
{
'choices': [
{
'message': {
'role': 'assistant',
'content': 'Creating the file.',
'tool_calls': [
{
'id': 'call_1',
'type': 'function',
'function': {
'name': 'write_file',
'arguments': '{"path": "out.txt", "content": "hello"}',
},
}
],
},
'finish_reason': 'tool_calls',
}
],
'usage': {'prompt_tokens': 4, 'completion_tokens': 3},
},
{
'choices': [
{
'message': {
'role': 'assistant',
'content': 'Write streamed.',
},
'finish_reason': 'stop',
}
],
'usage': {'prompt_tokens': 5, 'completion_tokens': 2},
},
]
with tempfile.TemporaryDirectory() as tmp_dir:
workspace = Path(tmp_dir)
with patch('src.openai_compat.request.urlopen', side_effect=make_urlopen_side_effect(responses)):
agent = LocalCodingAgent(
model_config=ModelConfig(
model='Qwen/Qwen3-Coder-30B-A3B-Instruct',
base_url='http://127.0.0.1:8000/v1',
),
runtime_config=AgentRuntimeConfig(
cwd=workspace,
permissions=AgentPermissions(allow_file_write=True),
),
)
result = agent.run('Create out.txt')
self.assertEqual(result.final_output, 'Write streamed.')
tool_delta_events = [
event for event in result.events
if event.get('type') == 'tool_delta' and event.get('tool_name') == 'write_file'
]
self.assertGreaterEqual(len(tool_delta_events), 1)
tool_message = next(
message for message in result.transcript
if message.get('role') == 'tool'
)
metadata = tool_message.get('metadata', {})
self.assertEqual(metadata.get('streamed'), True)
self.assertEqual(metadata.get('action'), 'write_file')
def test_agent_streams_bash_tool_output_and_mutates_tool_transcript(self) -> None:
responses = [
@@ -1142,8 +1211,13 @@ class AgentRuntimeTests(unittest.TestCase):
]
self.assertEqual(len(replay_messages), 1)
replay_content = replay_messages[0]['content']
self.assertIn('Unique changed paths: 1', replay_content)
self.assertIn('Snapshot ids: 2', replay_content)
self.assertIn('before_snapshot:', replay_content)
self.assertIn('after_snapshot:', replay_content)
self.assertIn('before: hello world', replay_content)
self.assertIn('after: hello mars', replay_content)
self.assertIn('result:', replay_content)
def test_resume_injects_compaction_replay_reminder(self) -> None:
responses = [
tests/test_query_engine_runtime.py
+114
View File
@@ -1034,3 +1034,117 @@ class QueryEngineRuntimeTests(unittest.TestCase):
self.assertIn('## Runtime Orchestration', summary)
self.assertIn('- group_status:completed=1', summary)
self.assertIn('- child_stop:stop=2', summary)
def test_query_engine_runtime_summary_tracks_resumed_delegate_children(self) -> None:
responses = [
{
'choices': [
{
'message': {
'role': 'assistant',
'content': 'Seed child result.',
},
'finish_reason': 'stop',
}
],
'usage': {'prompt_tokens': 5, 'completion_tokens': 2},
},
{
'choices': [
{
'message': {
'role': 'assistant',
'content': 'Delegating into resumed child.',
'tool_calls': [
{
'id': 'call_1',
'type': 'function',
'function': {
'name': 'delegate_agent',
'arguments': '{}',
},
}
],
},
'finish_reason': 'tool_calls',
}
],
'usage': {'prompt_tokens': 8, 'completion_tokens': 3},
},
{
'choices': [
{
'message': {
'role': 'assistant',
'content': 'Resumed child result.',
},
'finish_reason': 'stop',
}
],
'usage': {'prompt_tokens': 6, 'completion_tokens': 2},
},
{
'choices': [
{
'message': {
'role': 'assistant',
'content': 'Parent completed after resumed child.',
},
'finish_reason': 'stop',
}
],
'usage': {'prompt_tokens': 7, 'completion_tokens': 2},
},
]
with tempfile.TemporaryDirectory() as tmp_dir:
workspace = Path(tmp_dir)
session_dir = workspace / '.port_sessions' / 'agent'
with patch(
'src.openai_compat.request.urlopen',
side_effect=make_urlopen_side_effect(responses),
):
seed_agent = LocalCodingAgent(
model_config=ModelConfig(
model='Qwen/Qwen3-Coder-30B-A3B-Instruct',
base_url='http://127.0.0.1:8000/v1',
),
runtime_config=AgentRuntimeConfig(
cwd=workspace,
session_directory=session_dir,
),
)
seeded = seed_agent.run('Seed the delegated child')
resumed_child_id = seeded.session_id or ''
delegate_arguments = json.dumps(
{
'subtasks': [
{
'label': 'resume_child',
'prompt': 'Continue the delegated child.',
'resume_session_id': resumed_child_id,
'max_turns': 2,
}
],
'max_turns': 2,
}
)
responses[1]['choices'][0]['message']['tool_calls'][0]['function']['arguments'] = delegate_arguments
agent = LocalCodingAgent(
model_config=ModelConfig(
model='Qwen/Qwen3-Coder-30B-A3B-Instruct',
base_url='http://127.0.0.1:8000/v1',
),
runtime_config=AgentRuntimeConfig(
cwd=workspace,
session_directory=session_dir,
),
)
engine = QueryEnginePort.from_runtime_agent(agent)
turn = engine.submit_message('Delegate into resumed child')
summary = engine.render_summary()
self.assertEqual(turn.output, 'Parent completed after resumed child.')
self.assertIn('## Runtime Orchestration', summary)
self.assertIn('- resumed_children=1', summary)