xAI

export XAI_API_KEY="xai-..."

from definable.model import xAI
from definable.model.message import Message

model = xAI(id="grok-3")
response = model.invoke(
    messages=[Message(role="user", content="Hello!")],
    assistant_message=Message(role="assistant", content=""),
)
print(response.content)

from definable.model.message import Message

model = xAI(
    id="grok-3",
    search_parameters={"mode": "auto"},
)

response = model.invoke(
    messages=[Message(role="user", content="What happened in tech news today?")],
    assistant_message=Message(role="assistant", content=""),
)
print(response.content)

# Access citations if available
if response.citations:
    for url in response.citations.urls:
        print(f"  Source: {url.url} - {url.title}")

from definable.model.message import Message

for chunk in model.invoke_stream(
    messages=[Message(role="user", content="Explain the latest AI developments.")],
    assistant_message=Message(role="assistant", content=""),
):
    if chunk.content:
        print(chunk.content, end="", flush=True)

from definable.model.message import Message

response = await model.ainvoke(
    messages=[Message(role="user", content="Hello!")],
    assistant_message=Message(role="assistant", content=""),
)