xAI

export XAI_API_KEY="xai-..."

from ai_query import generate_text
from ai_query.providers import xai

result = await generate_text(
    model=xai("grok-2-1212"),
    prompt="Tell me a joke."
)

result = await generate_text(
    model=xai("grok-2-1212"),
    prompt="Write a story.",
    provider_options={
        "xai": {
            "temperature": 0.9,
            "max_tokens": 1000
        }
    }
)

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

@tool(description="Get weather")
async def get_weather(location: str = Field(description="City")) -> str:
    return f"Weather in {location}: 72°F, sunny"

result = await generate_text(
    model=xai("grok-2-1212"),
    prompt="What is the weather in SF?",
    tools={"get_weather": get_weather}
)

from ai_query import stream_text
from ai_query.providers import xai

result = stream_text(
    model=xai("grok-2-1212"),
    prompt="Write a poem."
)

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