mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
3446eb8d5d
* updates to final deprecated pieces and versions * fix mypy * fix readme links
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
import asyncio
|
|
from pathlib import Path
|
|
from random import randint
|
|
from typing import Literal
|
|
|
|
from agent_framework.declarative import AgentFactory
|
|
from agent_framework.foundry import FoundryChatClient
|
|
from azure.identity import AzureCliCredential
|
|
from dotenv import load_dotenv
|
|
|
|
# Load environment variables from .env file
|
|
load_dotenv()
|
|
|
|
|
|
def get_weather(location: str, unit: Literal["celsius", "fahrenheit"] = "celsius") -> str:
|
|
"""A simple function tool to get weather information."""
|
|
return f"The weather in {location} is {randint(-10, 30) if unit == 'celsius' else randint(30, 100)} degrees {unit}."
|
|
|
|
|
|
async def main():
|
|
"""Create an agent from a declarative yaml specification and run it."""
|
|
# get the path
|
|
current_path = Path(__file__).parent
|
|
yaml_path = (
|
|
current_path.parent.parent.parent.parent
|
|
/ "declarative-agents"
|
|
/ "agent-samples"
|
|
/ "chatclient"
|
|
/ "GetWeather.yaml"
|
|
)
|
|
# load the yaml from the path
|
|
with yaml_path.open("r") as f:
|
|
yaml_str = f.read()
|
|
# create the AgentFactory with a chat client and bindings
|
|
agent_factory = AgentFactory(
|
|
client=FoundryChatClient(credential=AzureCliCredential()),
|
|
bindings={"get_weather": get_weather},
|
|
)
|
|
# create the agent from the yaml
|
|
agent = agent_factory.create_agent_from_yaml(yaml_str)
|
|
# use the agent
|
|
response = await agent.run("What's the weather in Amsterdam, in celsius?")
|
|
print("Agent response:", response.text)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|