agent/agent-framework
2
0
Fork 0
mirror of https://github.com/microsoft/agent-framework.git synced 2026-06-16 21:04:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Python: Fix streamed workflow agent continuation context by finalizing AgentExecutor streams (#3882)

Browse Source 
* Fix streamed workflow agent continuation context by finalizing AgentExecutor streams

* Fix stream handling

* Fixes

* Fix DevUI and tests
This commit is contained in:
Evan Mattson
2026-02-13 07:45:46 +09:00
committed by GitHub
Unverified
parent 2203fa0f8b
commit a276c1295a
17 changed files with 359 additions and 267 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/devui/tests/devui/test_server.py
+8 -7
View File
@@ -379,26 +379,27 @@ async def test_checkpoint_api_endpoints(test_entities_dir):
storage = executor.checkpoint_manager.get_checkpoint_storage(conv_id)
checkpoint = WorkflowCheckpoint(
checkpoint_id="test_checkpoint_1",
workflow_id="test_workflow",
workflow_name="test_workflow",
graph_signature_hash="test_graph_hash",
state={"key": "value"},
iteration_count=1,
)
await storage.save_checkpoint(checkpoint)
await storage.save(checkpoint)
# Test list checkpoints endpoint
checkpoints = await storage.list_checkpoints()
checkpoints = await storage.list_checkpoints(workflow_name="test_workflow")
assert len(checkpoints) == 1
assert checkpoints[0].checkpoint_id == "test_checkpoint_1"
assert checkpoints[0].workflow_id == "test_workflow"
assert checkpoints[0].workflow_name == "test_workflow"
# Test delete checkpoint endpoint
deleted = await storage.delete_checkpoint("test_checkpoint_1")
deleted = await storage.delete("test_checkpoint_1")
assert deleted is True
# Verify checkpoint was deleted
remaining = await storage.list_checkpoints()
remaining = await storage.list_checkpoints(workflow_name="test_workflow")
assert len(remaining) == 0
# Test delete non-existent checkpoint
deleted = await storage.delete_checkpoint("nonexistent")
deleted = await storage.delete("nonexistent")
assert deleted is False