mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Eduard van Valkenburg
136 lines
4.3 KiB
Python
136 lines
4.3 KiB
Python
# 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()
|