Python: DevUI improvements. (#1091)

* enable deeplinking in ui, add agent details to entity info, add usage data, add middleware example in samples and foundry agent. * update ui build * Update python/packages/devui/frontend/src/components/workflow/workflow-input-form.tsx Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> * Update python/packages/devui/pyproject.toml Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> * Update python/packages/devui/pyproject.toml Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> * imporove mapping for agent nodes and serialiation for agent run events * lint fixes * update pyproj toml and ui updates --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
2026-06-16 21:04:09 +08:00 · 2025-10-03 22:22:03 +00:00
parent 61ac6d43b2
commit 01f438d710
33 changed files with 2733 additions and 1164 deletions
@@ -2,12 +2,73 @@
 """Sample weather agent for Agent Framework Debug UI."""

 import os
+from collections.abc import Awaitable, Callable
 from typing import Annotated

-from agent_framework import ChatAgent
+from agent_framework import (
+    ChatAgent,
+    ChatContext,
+    ChatMessage,
+    ChatResponse,
+    FunctionInvocationContext,
+    Role,
+    chat_middleware,
+    function_middleware,
+)
 from agent_framework.azure import AzureOpenAIChatClient


+@chat_middleware
+async def security_filter_middleware(
+    context: ChatContext,
+    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
+                    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."
+                                ),
+                            )
+                        ]
+                    )
+                    return
+
+    await next(context)
+
+
+@function_middleware
+async def atlantis_location_filter_middleware(
+    context: FunctionInvocationContext,
+    next: Callable[[FunctionInvocationContext], Awaitable[None]],
+) -> None:
+    """Function middleware that blocks weather requests for Atlantis."""
+    # Check if location parameter is "atlantis"
+    location = getattr(context.arguments, "location", None)
+    if location and location.lower() == "atlantis":
+        context.result = (
+            "Blocked! Hold up right there!! Tell the user that "
+            "'Atlantis is a special place, we must never ask about the weather there!!'"
+        )
+        context.terminate = True
+        return
+
+    await next(context)
+
+
 def get_weather(
    location: Annotated[str, "The location to get the weather for."],
 ) -> str:
@@ -46,6 +107,7 @@ agent = ChatAgent(
        api_key=os.environ.get("AZURE_OPENAI_API_KEY", ""),
    ),
    tools=[get_weather, get_forecast],
+    middleware=[security_filter_middleware, atlantis_location_filter_middleware],
 )