mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Tao Chen
213491da66
* Add support for function approval flow in Foundry hosted agent * Address comments * Address comments * Address comments
75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import os
|
|
import subprocess
|
|
from random import randint
|
|
from typing import Annotated
|
|
|
|
from agent_framework import Agent, tool
|
|
from agent_framework.foundry import FoundryChatClient
|
|
from agent_framework_foundry_hosting import ResponsesHostServer
|
|
from azure.identity import DefaultAzureCredential
|
|
from dotenv import load_dotenv
|
|
from pydantic import Field
|
|
|
|
# Load environment variables from .env file
|
|
load_dotenv()
|
|
|
|
|
|
@tool(approval_mode="never_require")
|
|
def get_weather(
|
|
location: Annotated[str, Field(description="The location to get the weather for.")],
|
|
) -> str:
|
|
"""Get the weather for a given location."""
|
|
conditions = ["sunny", "cloudy", "rainy", "stormy"]
|
|
return f"The weather in {location} is {conditions[randint(0, 3)]} with a high of {randint(10, 30)}°C."
|
|
|
|
|
|
@tool(approval_mode="always_require")
|
|
def run_bash(command: str) -> str:
|
|
"""Execute a shell command locally and return stdout, stderr, and exit code."""
|
|
try:
|
|
result = subprocess.run(
|
|
command,
|
|
shell=True,
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=30,
|
|
)
|
|
parts: list[str] = []
|
|
if result.stdout:
|
|
parts.append(result.stdout)
|
|
if result.stderr:
|
|
parts.append(f"stderr: {result.stderr}")
|
|
parts.append(f"exit_code: {result.returncode}")
|
|
return "\n".join(parts)
|
|
except subprocess.TimeoutExpired:
|
|
return "Command timed out after 30 seconds"
|
|
except Exception as e:
|
|
return f"Error executing command: {e}"
|
|
|
|
|
|
def main():
|
|
client = FoundryChatClient(
|
|
project_endpoint=os.environ["FOUNDRY_PROJECT_ENDPOINT"],
|
|
model=os.environ["AZURE_AI_MODEL_DEPLOYMENT_NAME"],
|
|
credential=DefaultAzureCredential(),
|
|
)
|
|
|
|
agent = Agent(
|
|
client=client,
|
|
instructions="You are a friendly assistant. Keep your answers brief.",
|
|
tools=[get_weather, run_bash],
|
|
# History will be managed by the hosting infrastructure, thus there
|
|
# is no need to store history by the service. Learn more at:
|
|
# https://developers.openai.com/api/reference/resources/responses/methods/create
|
|
default_options={"store": False},
|
|
)
|
|
|
|
server = ResponsesHostServer(agent)
|
|
server.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|