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

[Purview] Mark responses as responses and fix epoch bug for python long overflow (#4225)

Browse Source
This commit is contained in:
Rishabh Chawla
2026-02-25 18:40:36 +00:00
committed by GitHub
parent 6138487888
commit 2e26bb9387
5 changed files with 57 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/purview/tests/purview/test_middleware.py
+4 -2
View File
@@ -148,8 +148,10 @@ class TestPurviewPolicyMiddleware:
await middleware.process(context, mock_next)
assert mock_process.call_count == 2
for call in mock_process.call_args_list:
assert call[0][1] == Activity.UPLOAD_TEXT
# First call (pre-check) should be UPLOAD_TEXT for user prompt
assert mock_process.call_args_list[0][0][1] == Activity.UPLOAD_TEXT
# Second call (post-check) should be DOWNLOAD_TEXT for agent response
assert mock_process.call_args_list[1][0][1] == Activity.DOWNLOAD_TEXT
async def test_middleware_streaming_skips_post_check(
self, middleware: PurviewPolicyMiddleware, mock_agent: MagicMock