mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Removed context parameter from call_next (#3829)
This commit is contained in:
@@ -65,7 +65,7 @@ class PurviewPolicyMiddleware(AgentMiddleware):
|
||||
async def process(
|
||||
self,
|
||||
context: AgentContext,
|
||||
call_next: Callable[[AgentContext], Awaitable[None]],
|
||||
call_next: Callable[[], Awaitable[None]],
|
||||
) -> None: # type: ignore[override]
|
||||
resolved_user_id: str | None = None
|
||||
try:
|
||||
@@ -92,7 +92,7 @@ class PurviewPolicyMiddleware(AgentMiddleware):
|
||||
if not self._settings.ignore_exceptions:
|
||||
raise
|
||||
|
||||
await call_next(context)
|
||||
await call_next()
|
||||
|
||||
try:
|
||||
# Post (response) check only if we have a normal AgentResponse
|
||||
@@ -162,7 +162,7 @@ class PurviewChatPolicyMiddleware(ChatMiddleware):
|
||||
async def process(
|
||||
self,
|
||||
context: ChatContext,
|
||||
call_next: Callable[[ChatContext], Awaitable[None]],
|
||||
call_next: Callable[[], Awaitable[None]],
|
||||
) -> None: # type: ignore[override]
|
||||
resolved_user_id: str | None = None
|
||||
try:
|
||||
@@ -187,7 +187,7 @@ class PurviewChatPolicyMiddleware(ChatMiddleware):
|
||||
if not self._settings.ignore_exceptions:
|
||||
raise
|
||||
|
||||
await call_next(context)
|
||||
await call_next()
|
||||
|
||||
try:
|
||||
# Post (response) evaluation only if non-streaming and we have messages result shape
|
||||
|
||||
Reference in New Issue
Block a user