chore: import upstream snapshot with attribution

python/samples/02-agents/devui/in_memory_mode.py

@@ -0,0 +1,135 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+"""Example of using Agent Framework DevUI with in-memory entity registration.
+
+This demonstrates the simplest way to serve agents and workflows as OpenAI-compatible API endpoints.
+Includes both agents and a basic workflow to showcase different entity types.
+"""
+
+import logging
+import os
+from typing import Annotated
+
+from agent_framework import (
+    Agent,
+    Executor,
+    WorkflowBuilder,
+    WorkflowContext,
+    handler,
+    tool,
+)
+from agent_framework.devui import serve
+from agent_framework.foundry import FoundryChatClient
+from azure.identity.aio import AzureCliCredential
+from dotenv import load_dotenv
+from typing_extensions import Never
+
+# Load environment variables from .env file
+load_dotenv()
+
+
+# NOTE: approval_mode="never_require" is for sample brevity. Use "always_require" in production;
+# see samples/02-agents/tools/function_tool_with_approval.py
+# and samples/02-agents/tools/function_tool_with_approval_and_sessions.py.
+@tool(approval_mode="never_require")
+def get_weather(
+    location: Annotated[str, "The location to get the weather for."],
+) -> str:
+    """Get the weather for a given location."""
+    conditions = ["sunny", "cloudy", "rainy", "stormy"]
+    temperature = 53
+    return f"The weather in {location} is {conditions[0]} with a high of {temperature}°C."
+
+
+@tool(approval_mode="never_require")
+def get_time(
+    timezone: Annotated[str, "The timezone to get time for."] = "UTC",
+) -> str:
+    """Get current time for a timezone."""
+    from datetime import datetime
+
+    # Simplified for example
+    return f"Current time in {timezone}: {datetime.now().strftime('%H:%M:%S')}"
+
+
+# Basic workflow executors
+class UpperCase(Executor):
+    """Convert text to uppercase."""
+
+    @handler
+    async def to_upper(self, text: str, ctx: WorkflowContext[str]) -> None:
+        """Convert input to uppercase and forward to next executor."""
+        result = text.upper()
+        await ctx.send_message(result)
+
+
+class AddExclamation(Executor):
+    """Add exclamation mark to text."""
+
+    @handler
+    async def add_exclamation(self, text: str, ctx: WorkflowContext[Never, str]) -> None:
+        """Add exclamation and yield as workflow output."""
+        result = f"{text}!"
+        await ctx.yield_output(result)
+
+
+def main():
+    """Main function demonstrating in-memory entity registration."""
+    # Setup logging
+    logging.basicConfig(level=logging.INFO, format="%(message)s")
+    logger = logging.getLogger(__name__)
+
+    # Create Azure OpenAI chat client
+    client = FoundryChatClient(
+        model=os.environ["FOUNDRY_MODEL"],
+        project_endpoint=os.environ.get("FOUNDRY_PROJECT_ENDPOINT"),
+        credential=AzureCliCredential(),
+    )
+
+    # Create agents
+    weather_assistant = Agent(
+        name="weather-assistant",
+        description="Provides weather information and time",
+        instructions=(
+            "You are a helpful weather and time assistant. Use the available tools to "
+            "provide accurate weather information and current time for any location."
+        ),
+        client=client,
+        tools=[get_weather, get_time],
+    )
+
+    simple_agent = Agent(
+        name="general-assistant",
+        description="A simple conversational agent",
+        instructions="You are a helpful assistant.",
+        client=client,
+    )
+
+    # Create a basic workflow: Input -> UpperCase -> AddExclamation -> Output
+    upper_executor = UpperCase(id="upper_case")
+    exclaim_executor = AddExclamation(id="add_exclamation")
+
+    basic_workflow = (
+        WorkflowBuilder(
+            name="Text Transformer",
+            description="Simple 2-step workflow that converts text to uppercase and adds exclamation",
+            start_executor=upper_executor,
+        )
+        .add_edge(upper_executor, exclaim_executor)
+        .build()
+    )
+
+    # Collect entities for serving
+    entities = [weather_assistant, simple_agent, basic_workflow]
+
+    logger.info("Starting DevUI on http://localhost:8090")
+    logger.info("Entities available:")
+    logger.info("  - Agents: weather-assistant, general-assistant")
+    logger.info("  - Workflow: basic text transformer (uppercase + exclamation)")
+
+    # Launch server with auto-generated entity IDs
+    serve(entities=entities, port=8090, auto_open=True)
+
+
+if __name__ == "__main__":
+    main()