mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Eduard van Valkenburg
df9d85d1f0
* openai updates * rebuild of openai structure * updated responses structure * renamed sample * added file id support to code interpreter * added hosted file ids to code interpretor * mypy fixes * removed default az cred from codebase * updated agent name setup * added kwargs to entra methods * and further kwargs * extra comment * updated all samples * readded custom get methods for responses * updated int tests with ad credential * missed one
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import asyncio
|
|
from random import randint
|
|
from typing import Annotated
|
|
|
|
from agent_framework.foundry import FoundryChatClient
|
|
from azure.identity.aio import DefaultAzureCredential
|
|
from pydantic import Field
|
|
|
|
|
|
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."
|
|
|
|
|
|
async def non_streaming_example() -> None:
|
|
"""Example of non-streaming response (get the complete result at once)."""
|
|
print("=== Non-streaming Response Example ===")
|
|
|
|
# Since no Agent ID is provided, the agent will be automatically created
|
|
# and deleted after getting a response
|
|
async with FoundryChatClient(async_ad_credential=DefaultAzureCredential()).create_agent(
|
|
name="WeatherAgent",
|
|
instructions="You are a helpful weather agent.",
|
|
tools=get_weather,
|
|
) as agent:
|
|
query = "What's the weather like in Seattle?"
|
|
print(f"User: {query}")
|
|
result = await agent.run(query)
|
|
print(f"Agent: {result}\n")
|
|
|
|
|
|
async def streaming_example() -> None:
|
|
"""Example of streaming response (get results as they are generated)."""
|
|
print("=== Streaming Response Example ===")
|
|
|
|
# Since no Agent ID is provided, the agent will be automatically created
|
|
# and deleted after getting a response
|
|
async with FoundryChatClient(async_ad_credential=DefaultAzureCredential()).create_agent(
|
|
name="WeatherAgent",
|
|
instructions="You are a helpful weather agent.",
|
|
tools=get_weather,
|
|
) as agent:
|
|
query = "What's the weather like in Portland?"
|
|
print(f"User: {query}")
|
|
print("Agent: ", end="", flush=True)
|
|
async for chunk in agent.run_streaming(query):
|
|
if chunk.text:
|
|
print(chunk.text, end="", flush=True)
|
|
print("\n")
|
|
|
|
|
|
async def main() -> None:
|
|
print("=== Basic Foundry Chat Client Agent Example ===")
|
|
|
|
await non_streaming_example()
|
|
await streaming_example()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|