mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
d774b64df0
* added inline yaml sample * fixed some typos and added intro comment * added description params and pass through to client * add azure assistants * fix tests * observabiltiy mypy fix * for some reason mypy doesn't accept a subclass --------- Co-authored-by: Evan Mattson <35585003+moonbox3@users.noreply.github.com>
41 lines
1.4 KiB
Python
41 lines
1.4 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.azure import AzureOpenAIResponsesClient
|
|
from agent_framework.declarative import AgentFactory
|
|
from azure.identity import AzureCliCredential
|
|
|
|
|
|
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 / "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(
|
|
chat_client=AzureOpenAIResponsesClient(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())
|