pip install agent-framework --pre
# Azure OpenAI support is included in the core package

from azure.identity import AzureCliCredential
from agent_framework.azure import AzureOpenAIResponsesClient

# Run 'az login' first
credential = AzureCliCredential()
client = AzureOpenAIResponsesClient(credential=credential)

from azure.identity import ManagedIdentityCredential
from agent_framework.azure import AzureOpenAIResponsesClient

credential = ManagedIdentityCredential()
client = AzureOpenAIResponsesClient(credential=credential)

from azure.core.credentials import AzureKeyCredential
from agent_framework.azure import AzureOpenAIResponsesClient

credential = AzureKeyCredential(api_key)
client = AzureOpenAIResponsesClient(credential=credential)

import asyncio
from agent_framework.azure import AzureOpenAIResponsesClient
from azure.identity import AzureCliCredential

async def main():
    # Create agent with Azure OpenAI Responses Client
    agent = AzureOpenAIResponsesClient(
        credential=AzureCliCredential()
    ).as_agent(
        instructions="You are a helpful assistant.",
    )
    
    # Non-streaming response
    result = await agent.run("What is the capital of France?")
    print(result)

asyncio.run(main())

import asyncio
from agent_framework.azure import AzureOpenAIResponsesClient
from azure.identity import AzureCliCredential

async def main():
    agent = AzureOpenAIResponsesClient(
        credential=AzureCliCredential()
    ).as_agent(
        instructions="You are a helpful assistant.",
    )
    
    # Streaming response
    query = "Write a short story about AI agents."
    print("Agent: ", end="", flush=True)
    async for chunk in agent.run(query, stream=True):
        if chunk.text:
            print(chunk.text, end="", flush=True)
    print()

asyncio.run(main())

import asyncio
from typing import Annotated
from agent_framework import tool
from agent_framework.azure import AzureOpenAIResponsesClient
from azure.identity import AzureCliCredential
from pydantic import Field

@tool(approval_mode="never_require")  # Use "always_require" in production
def get_weather(
    location: Annotated[str, Field(description="City name")],
) -> str:
    """Get the weather for a location."""
    return f"Weather in {location}: Sunny, 72°F"

async def main():
    agent = AzureOpenAIResponsesClient(
        credential=AzureCliCredential()
    ).as_agent(
        instructions="You are a weather assistant.",
        tools=[get_weather],
    )
    
    result = await agent.run("What's the weather in Seattle?")
    print(result)

asyncio.run(main())

import asyncio
from agent_framework.azure import AzureOpenAIChatClient
from azure.identity import AzureCliCredential

async def main():
    agent = AzureOpenAIChatClient(
        credential=AzureCliCredential()
    ).as_agent(
        instructions="You are a helpful assistant.",
    )
    
    result = await agent.run("Hello, how are you?")
    print(result)

asyncio.run(main())

import asyncio
from agent_framework.azure import AzureAIProjectAgentProvider
from azure.identity.aio import AzureCliCredential

async def main():
    async with (
        AzureCliCredential() as credential,
        AzureAIProjectAgentProvider(credential=credential) as provider
    ):
        # Create a persistent agent
        agent = await provider.create_agent(
            name="WeatherAgent",
            instructions="You are a weather assistant.",
            tools=[...],
        )
        
        # Use the agent
        result = await agent.run("What's the weather?")
        print(result)

asyncio.run(main())

import asyncio
from agent_framework.azure import AzureAIProjectAgentProvider
from azure.identity.aio import AzureCliCredential

async def main():
    async with (
        AzureCliCredential() as credential,
        AzureAIProjectAgentProvider(credential=credential) as provider
    ):
        # Get an existing agent by ID
        agent = await provider.get_agent(agent_id="your-agent-id")
        
        # Use the agent
        result = await agent.run("Hello!")
        print(result)

asyncio.run(main())

from agent_framework.azure import code_interpreter_tool

agent = await provider.create_agent(
    name="DataAnalyst",
    instructions="You are a data analyst.",
    tools=[code_interpreter_tool()],
)

from agent_framework.azure import file_search_tool

agent = await provider.create_agent(
    name="DocumentAssistant",
    instructions="You help with documents.",
    tools=[file_search_tool()],
)

from pydantic import BaseModel

class WeatherResponse(BaseModel):
    location: str
    temperature: float
    condition: str

agent = client.as_agent(
    instructions="You are a weather assistant.",
    response_format=WeatherResponse,
)