renamed all (#3207)

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
@@ -8,8 +8,8 @@ from typing import Annotated
from agent_framework import (
AgentMiddleware,
AgentResponse,
AgentRunContext,
AgentRunResponse,
FunctionInvocationContext,
)
from agent_framework.azure import AzureAIAgentClient
@@ -121,7 +121,7 @@ class CachingMiddleware(AgentMiddleware):
"""Run-level caching middleware for expensive operations."""
def __init__(self) -> None:
self.cache: dict[str, AgentRunResponse] = {}
self.cache: dict[str, AgentResponse] = {}
async def process(self, context: AgentRunContext, next: Callable[[AgentRunContext], Awaitable[None]]) -> None:
# Create a simple cache key from the last message
@@ -8,8 +8,8 @@ from typing import Annotated
from agent_framework import (
AgentMiddleware,
AgentResponse,
AgentRunContext,
AgentRunResponse,
ChatMessage,
FunctionInvocationContext,
FunctionMiddleware,
@@ -58,7 +58,7 @@ class SecurityAgentMiddleware(AgentMiddleware):
if "password" in query.lower() or "secret" in query.lower():
print("[SecurityAgentMiddleware] Security Warning: Detected sensitive information, blocking request.")
# Override the result with warning message
context.result = AgentRunResponse(
context.result = AgentResponse(
messages=[
ChatMessage(role=Role.ASSISTANT, text="Detected sensitive information, the request is blocked.")
]
@@ -7,8 +7,8 @@ from typing import Annotated
from agent_framework import (
AgentMiddleware,
AgentResponse,
AgentRunContext,
AgentRunResponse,
ChatMessage,
Role,
)
@@ -57,7 +57,7 @@ class PreTerminationMiddleware(AgentMiddleware):
print(f"[PreTerminationMiddleware] Blocked word '{blocked_word}' detected. Terminating request.")
# Set a custom response
context.result = AgentRunResponse(
context.result = AgentResponse(
messages=[
ChatMessage(
role=Role.ASSISTANT,
@@ -6,9 +6,9 @@ from random import randint
from typing import Annotated
from agent_framework import (
AgentResponse,
AgentResponseUpdate,
AgentRunContext,
AgentRunResponse,
AgentRunResponseUpdate,
ChatMessage,
Role,
TextContent,
@@ -64,15 +64,15 @@ async def weather_override_middleware(
if context.is_streaming:
# For streaming: create an async generator that yields chunks
async def override_stream() -> AsyncIterable[AgentRunResponseUpdate]:
async def override_stream() -> AsyncIterable[AgentResponseUpdate]:
for chunk in chunks:
yield AgentRunResponseUpdate(contents=[TextContent(text=chunk)])
yield AgentResponseUpdate(contents=[TextContent(text=chunk)])
context.result = override_stream()
else:
# For non-streaming: just replace with the string message
custom_message = "".join(chunks)
context.result = AgentRunResponse(messages=[ChatMessage(role=Role.ASSISTANT, text=custom_message)])
context.result = AgentResponse(messages=[ChatMessage(role=Role.ASSISTANT, text=custom_message)])
async def main() -> None: