mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Tao Chen
e064f943ae
* Remove duplicate samples * Correct paths * Update readme * Update readme * Fix ruff --------- Co-authored-by: Evan Mattson <35585003+moonbox3@users.noreply.github.com>
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
from typing import Annotated, Any
|
|
|
|
import anyio
|
|
from agent_framework import tool
|
|
from agent_framework.openai import OpenAIResponsesClient
|
|
|
|
"""
|
|
This sample demonstrates how to expose an Agent as an MCP server.
|
|
|
|
To run this sample, set up your MCP host (like Claude Desktop or VSCode GitHub Copilot Agents)
|
|
with the following configuration:
|
|
```json
|
|
{
|
|
"servers": {
|
|
"agent-framework": {
|
|
"command": "uv",
|
|
"args": [
|
|
"--directory=<path to project>/agent-framework/python/samples/02-agents/mcp",
|
|
"run",
|
|
"agent_as_mcp_server.py"
|
|
],
|
|
"env": {
|
|
"OPENAI_API_KEY": "<OpenAI API key>",
|
|
"OPENAI_RESPONSES_MODEL_ID": "<OpenAI Responses model ID>",
|
|
}
|
|
}
|
|
}
|
|
}
|
|
```
|
|
"""
|
|
|
|
|
|
# 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_specials() -> Annotated[str, "Returns the specials from the menu."]:
|
|
return """
|
|
Special Soup: Clam Chowder
|
|
Special Salad: Cobb Salad
|
|
Special Drink: Chai Tea
|
|
"""
|
|
|
|
|
|
@tool(approval_mode="never_require")
|
|
def get_item_price(
|
|
menu_item: Annotated[str, "The name of the menu item."],
|
|
) -> Annotated[str, "Returns the price of the menu item."]:
|
|
return "$9.99"
|
|
|
|
|
|
async def run() -> None:
|
|
# Define an agent
|
|
# Agent's name and description provide better context for AI model
|
|
agent = OpenAIResponsesClient().as_agent(
|
|
name="RestaurantAgent",
|
|
description="Answer questions about the menu.",
|
|
tools=[get_specials, get_item_price],
|
|
)
|
|
|
|
# Expose the agent as an MCP server
|
|
server = agent.as_mcp_server()
|
|
|
|
# Run server
|
|
from mcp.server.stdio import stdio_server
|
|
|
|
async def handle_stdin(stdin: Any | None = None, stdout: Any | None = None) -> None:
|
|
async with stdio_server() as (read_stream, write_stream):
|
|
await server.run(read_stream, write_stream, server.create_initialization_options())
|
|
|
|
await handle_stdin()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
anyio.run(run)
|