OpenAI

export OPENAI_API_KEY="sk-..."

from ai_query import generate_text
from ai_query.providers import openai

result = await generate_text(
    model=openai("gpt-4o"),
    prompt="Explain machine learning in simple terms."
)

result = await generate_text(
    model=openai("gpt-4o"),
    prompt="Write a creative story.",
    provider_options={
        "openai": {
            "temperature": 0.9,
            "max_tokens": 1000,
            "top_p": 0.95,
            "frequency_penalty": 0.5,
            "presence_penalty": 0.5
        }
    }
)

result = await generate_text(
    model=openai("gpt-5.4"),
    prompt="Think this through carefully.",
    reasoning={"effort": "high"},
    max_tokens=800,
)

from ai_query import generate_text
from ai_query.providers import openai, tool, Field

@tool(description="Search the web")
async def search(query: str = Field(description="Search query")) -> str:
    return f"Results for: {query}"

result = await generate_text(
    model=openai("gpt-4o"),
    prompt="Search for the latest Python news",
    tools={"search": search}
)

from ai_query import stream_text
from ai_query.providers import openai

result = stream_text(
    model=openai("gpt-4o"),
    prompt="Write a poem about coding."
)

async for chunk in result.text_stream:
    print(chunk, end="", flush=True)