diff --git a/python/samples/getting_started/devui/weather_agent_azure/agent.py b/python/samples/getting_started/devui/weather_agent_azure/agent.py
index ab2d9169a6..40310cb640 100644
--- a/python/samples/getting_started/devui/weather_agent_azure/agent.py
+++ b/python/samples/getting_started/devui/weather_agent_azure/agent.py
@@ -3,7 +3,7 @@
 
 import logging
 import os
-from collections.abc import Awaitable, Callable
+from collections.abc import AsyncIterable, Awaitable, Callable
 from typing import Annotated
 
 from agent_framework import (
@@ -11,8 +11,10 @@ from agent_framework import (
     ChatContext,
     ChatMessage,
     ChatResponse,
+    ChatResponseUpdate,
     FunctionInvocationContext,
     Role,
+    TextContent,
     chat_middleware,
     function_middleware,
     ai_function
@@ -37,28 +39,42 @@ async def security_filter_middleware(
     next: Callable[[ChatContext], Awaitable[None]],
 ) -> None:
     """Chat middleware that blocks requests containing sensitive information."""
-    # Block requests with sensitive information
     blocked_terms = ["password", "secret", "api_key", "token"]
 
-    for message in context.messages:
-        if message.text:
-            message_lower = message.text.lower()
-            for term in blocked_terms:
-                if term in message_lower:
-                    # Override the response without calling the LLM
+    # Check only the last message (most recent user input)
+    last_message = context.messages[-1] if context.messages else None
+    if last_message and last_message.role == Role.USER and last_message.text:
+        message_lower = last_message.text.lower()
+        for term in blocked_terms:
+            if term in message_lower:
+                error_message = (
+                    "I cannot process requests containing sensitive information. "
+                    "Please rephrase your question without including passwords, secrets, "
+                    "or other sensitive data."
+                )
+
+                if context.is_streaming:
+                    # Streaming mode: return async generator
+                    async def blocked_stream() -> AsyncIterable[ChatResponseUpdate]:
+                        yield ChatResponseUpdate(
+                            contents=[TextContent(text=error_message)],
+                            role=Role.ASSISTANT,
+                        )
+
+                    context.result = blocked_stream()
+                else:
+                    # Non-streaming mode: return complete response
                     context.result = ChatResponse(
                         messages=[
                             ChatMessage(
                                 role=Role.ASSISTANT,
-                                text=(
-                                    "I cannot process requests containing sensitive information. "
-                                    "Please rephrase your question without including passwords, secrets, "
-                                    "or other sensitive data."
-                                ),
+                                text=error_message,
                             )
                         ]
                     )
-                    return
+
+                context.terminate = True
+                return
 
     await next(context)