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

import os
from agent_framework.openai import OpenAIChatClient

# Set OPENAI_API_KEY environment variable
# The client will automatically read from the environment
client = OpenAIChatClient()

from agent_framework.openai import OpenAIChatClient

client = OpenAIChatClient(api_key="sk-proj-...")

import asyncio
from agent_framework.openai import OpenAIChatClient

async def main():
    # Create agent with OpenAI Chat Client
    agent = OpenAIChatClient().as_agent(
        name="Assistant",
        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.openai import OpenAIChatClient

async def main():
    agent = OpenAIChatClient().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.openai import OpenAIChatClient

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

async def main():
    agent = OpenAIChatClient().as_agent(
        name="WeatherAgent",
        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.openai import OpenAIResponsesClient

async def main():
    agent = OpenAIResponsesClient().as_agent(
        instructions="You are a helpful assistant.",
    )
    
    result = await agent.run("Explain quantum computing in simple terms.")
    print(result)

asyncio.run(main())

import asyncio
import os
from agent_framework.openai import OpenAIAssistantProvider
from openai import AsyncOpenAI

async def main():
    client = AsyncOpenAI()
    provider = OpenAIAssistantProvider(client)
    
    # Create a new assistant
    agent = await provider.create_agent(
        name="MathTutor",
        model=os.environ.get("OPENAI_CHAT_MODEL_ID", "gpt-4o"),
        instructions="You are a helpful math tutor.",
        tools=[...],
    )
    
    try:
        result = await agent.run("What is 25 * 17?")
        print(result)
    finally:
        # Clean up the assistant
        await client.beta.assistants.delete(agent.id)

asyncio.run(main())

import asyncio
from agent_framework.openai import OpenAIAssistantProvider
from openai import AsyncOpenAI

async def main():
    client = AsyncOpenAI()
    provider = OpenAIAssistantProvider(client)
    
    # Get existing assistant by ID
    agent = await provider.get_agent(assistant_id="asst_...")
    
    result = await agent.run("Hello!")
    print(result)

asyncio.run(main())

from agent_framework.openai import code_interpreter_tool

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

from agent_framework.openai import file_search_tool

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

import asyncio
from agent_framework.openai import OpenAIResponsesClient

async def main():
    agent = OpenAIResponsesClient(
        model_id="o3-mini"  # or "o1-preview", "o1-mini"
    ).as_agent(
        instructions="You are a reasoning assistant.",
    )
    
    result = await agent.run(
        "Solve this logic puzzle: Three people are in a room. "
        "Alice says Bob is lying. Bob says Charlie is lying. "
        "Charlie says both Alice and Bob are lying. Who is telling the truth?"
    )
    print(result)

asyncio.run(main())

import asyncio
from agent_framework import Message
from agent_framework.openai import OpenAIChatClient

async def main():
    agent = OpenAIChatClient(model_id="gpt-4o").as_agent(
        instructions="You analyze images.",
    )
    
    message = Message(
        role="user",
        text="What's in this image?",
        images=["https://example.com/image.jpg"],
    )
    
    result = await agent.run(message)
    print(result)

asyncio.run(main())

import asyncio
from pydantic import BaseModel
from agent_framework.openai import OpenAIChatClient

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

async def main():
    agent = OpenAIChatClient().as_agent(
        instructions="You are a weather assistant.",
        response_format=WeatherResponse,
    )
    
    result = await agent.run("What's the weather in Paris?")
    # result is parsed as WeatherResponse
    print(f"Location: {result.location}")
    print(f"Temperature: {result.temperature}°F")
    print(f"Condition: {result.condition}")

asyncio.run(main())