wehub-resource-sync
db620d33df
CodeQL / Analyze (csharp) (push) Waiting to run
CodeQL / Analyze (python) (push) Waiting to run
dotnet-build-and-test / dotnet-test-functions (push) Has been cancelled
dotnet-build-and-test / paths-filter (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Debug, windows-latest, net9.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, ubuntu-latest, net10.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, ubuntu-latest, net8.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, windows-latest, net472) (push) Has been cancelled
dotnet-build-and-test / dotnet-test (Release, integration, true, ubuntu-latest, net10.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-test (Release, integration, true, windows-latest, net472) (push) Has been cancelled
dotnet-build-and-test / dotnet-foundry-hosted-it (push) Has been cancelled
dotnet-build-and-test / dotnet-build-and-test-check (push) Has been cancelled
dotnet-build-and-test / Integration Test Report (push) Has been cancelled
91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import asyncio
|
|
import os
|
|
from random import randint
|
|
from typing import Annotated
|
|
|
|
from agent_framework import Agent, tool
|
|
from agent_framework.openai import OpenAIChatClient
|
|
from azure.identity import AzureCliCredential
|
|
from dotenv import load_dotenv
|
|
from pydantic import Field
|
|
|
|
# Load environment variables from .env file
|
|
load_dotenv()
|
|
|
|
"""
|
|
Azure OpenAI Chat Client Basic Example
|
|
|
|
This sample demonstrates basic usage of OpenAIChatClient with explicit Azure
|
|
settings and a credential, showing both streaming and non-streaming responses.
|
|
"""
|
|
|
|
|
|
# NOTE: approval_mode="never_require" is for sample brevity. Use "always_require" in production.
|
|
@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."
|
|
|
|
|
|
async def non_streaming_example() -> None:
|
|
"""Example of non-streaming response (get the complete result at once)."""
|
|
print("=== Non-streaming Response Example ===")
|
|
|
|
agent = Agent(
|
|
client=OpenAIChatClient(
|
|
model=os.getenv("AZURE_OPENAI_MODEL"),
|
|
azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
|
|
api_version=os.getenv("AZURE_OPENAI_API_VERSION"),
|
|
credential=AzureCliCredential(),
|
|
),
|
|
name="WeatherAgent",
|
|
instructions="You are a helpful weather agent.",
|
|
tools=get_weather,
|
|
)
|
|
|
|
query = "What's the weather in Seattle?"
|
|
print(f"User: {query}")
|
|
result = await agent.run(query)
|
|
print(f"Result: {result}\n")
|
|
|
|
|
|
async def streaming_example() -> None:
|
|
"""Example of streaming response (get results as they are generated)."""
|
|
print("=== Streaming Response Example ===")
|
|
|
|
agent = Agent(
|
|
client=OpenAIChatClient(
|
|
model=os.getenv("AZURE_OPENAI_MODEL"),
|
|
azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
|
|
api_version=os.getenv("AZURE_OPENAI_API_VERSION"),
|
|
credential=AzureCliCredential(),
|
|
),
|
|
name="WeatherAgent",
|
|
instructions="You are a helpful weather agent.",
|
|
tools=get_weather,
|
|
)
|
|
|
|
query = "What's the weather in Portland?"
|
|
print(f"User: {query}")
|
|
print("Agent: ", end="", flush=True)
|
|
async for chunk in agent.run(query, stream=True):
|
|
if chunk.text:
|
|
print(chunk.text, end="", flush=True)
|
|
print("\n")
|
|
|
|
|
|
async def main() -> None:
|
|
print("=== Basic Azure OpenAI Chat Client Agent Example ===")
|
|
|
|
await non_streaming_example()
|
|
await streaming_example()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|