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

renamed all (#3207)

Browse Source
This commit is contained in:
Eduard van Valkenburg
2026-01-14 06:54:07 +01:00
committed by GitHub
Unverified
parent 1ae0b09e42
commit d8cf8361bd
125 changed files with 1024 additions and 1027 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/purview/agent_framework_purview/_middleware.py
+5 -5
View File
@@ -57,9 +57,9 @@ class PurviewPolicyMiddleware(AgentMiddleware):
context.messages, Activity.UPLOAD_TEXT
)
if should_block_prompt:
from agent_framework import AgentRunResponse, ChatMessage, Role
from agent_framework import AgentResponse, ChatMessage, Role
context.result = AgentRunResponse(
context.result = AgentResponse(
messages=[ChatMessage(role=Role.SYSTEM, text=self._settings.blocked_prompt_message)]
)
context.terminate = True
@@ -76,7 +76,7 @@ class PurviewPolicyMiddleware(AgentMiddleware):
await next(context)
try:
# Post (response) check only if we have a normal AgentRunResponse
# Post (response) check only if we have a normal AgentResponse
# Use the same user_id from the request for the response evaluation
if context.result and not context.is_streaming:
should_block_response, _ = await self._processor.process_messages(
@@ -85,9 +85,9 @@ class PurviewPolicyMiddleware(AgentMiddleware):
user_id=resolved_user_id,
)
if should_block_response:
from agent_framework import AgentRunResponse, ChatMessage, Role
from agent_framework import AgentResponse, ChatMessage, Role
context.result = AgentRunResponse(
context.result = AgentResponse(
messages=[ChatMessage(role=Role.SYSTEM, text=self._settings.blocked_response_message)]
)
else:
python/packages/purview/tests/test_middleware.py
+7 -7
View File
@@ -5,7 +5,7 @@
from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from agent_framework import AgentRunContext, AgentRunResponse, ChatMessage, Role
from agent_framework import AgentResponse, AgentRunContext, ChatMessage, Role
from azure.core.credentials import AccessToken
from agent_framework_purview import PurviewPolicyMiddleware, PurviewSettings
@@ -57,7 +57,7 @@ class TestPurviewPolicyMiddleware:
async def mock_next(ctx: AgentRunContext) -> None:
nonlocal next_called
next_called = True
ctx.result = AgentRunResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="I'm good, thanks!")])
ctx.result = AgentResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="I'm good, thanks!")])
await middleware.process(context, mock_next)
@@ -104,7 +104,7 @@ class TestPurviewPolicyMiddleware:
with patch.object(middleware._processor, "process_messages", side_effect=mock_process_messages):
async def mock_next(ctx: AgentRunContext) -> None:
ctx.result = AgentRunResponse(
ctx.result = AgentResponse(
messages=[ChatMessage(role=Role.ASSISTANT, text="Here's some sensitive information")]
)
@@ -145,7 +145,7 @@ class TestPurviewPolicyMiddleware:
with patch.object(middleware._processor, "process_messages", return_value=(False, "user-123")) as mock_process:
async def mock_next(ctx: AgentRunContext) -> None:
ctx.result = AgentRunResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="Response")])
ctx.result = AgentResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="Response")])
await middleware.process(context, mock_next)
@@ -167,7 +167,7 @@ class TestPurviewPolicyMiddleware:
) as mock_process:
async def mock_next(ctx: AgentRunContext) -> None:
ctx.result = AgentRunResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="Response")])
ctx.result = AgentResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="Response")])
await middleware.process(context, mock_next)
@@ -199,7 +199,7 @@ class TestPurviewPolicyMiddleware:
with patch.object(middleware._processor, "process_messages", side_effect=mock_process_messages):
async def mock_next(ctx: AgentRunContext) -> None:
ctx.result = AgentRunResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="Response")])
ctx.result = AgentResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="Response")])
await middleware.process(context, mock_next)
@@ -225,7 +225,7 @@ class TestPurviewPolicyMiddleware:
with patch.object(middleware._processor, "process_messages", side_effect=mock_process_messages):
async def mock_next(ctx):
ctx.result = AgentRunResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="Response")])
ctx.result = AgentResponse(messages=[ChatMessage(role=Role.ASSISTANT, text="Response")])
# Should not raise, just log
await middleware.process(context, mock_next)