XAI

AISDK provides first-class support for XAI with fully typed model APIs. Model capabilities are enforced at compile time using Rust's type system. This prevents model capability mismatches and guarantees the selected model is valid for the task (e.g. tool calling).

Installation

Enable the XAI provider feature:

cargo add aisdk --features xai

This installs AISDK with the XAI provider enabled. Once you have enabled the XAI provider, you can use all aisdk features with it.

Create a Provider Instance

To create a provider instance, call XAI::model_name(), where model_name is the XAI model you want to use. Model names are exposed as snake-case methods.

use aisdk::providers::XAI;

let xai = XAI::grok_3();

This initializes the provider with:

• Model: "grok-3"
• API key from environment (if set with XAI_API_KEY)
• XAI's default base URL (https://api.x.ai/)

Basic Text Generation

Example using LanguageModelRequest for text generation.

use aisdk::{
    core::LanguageModelRequest,
    providers::XAI,
};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {

    let xai = XAI::grok_3();

    let response = LanguageModelRequest::builder()
        .model(xai)
        .prompt("Write a short poem about Rust.")
        .build()
        .generate_text()
        .await?;

    println!("Response text: {:?}", response.text());
    Ok(())
}

Provider Settings

You can customize provider configuration using XAI::builder()

API Key

let xai = XAI::<Grok3>::builder()
    .api_key("your-api-key")
    .build()?;

If not specified, AISDK uses the XAI_API_KEY environment variable.

Base URL

Useful when routing through a proxy, gateway, or self-hosted compatible endpoint.

let xai = XAI::<Grok3>::builder()
    .base_url("https://api.x.ai/")
    .build()?;

Provider Name

For logging, analytics, and observability.

let xai = XAI::<Grok3>::builder()
    .provider_name("XAI")
    .build()?;

Full Custom Configuration Example

let xai = XAI::<Grok3>::builder()
    .api_key("your-api-key")
    .base_url("https://api.x.ai/")
    .provider_name("XAI")
    .build()?;

Dynamic Model Selection

For runtime model selection (e.g., loading models from config files), use DynamicModel:

Using model_name() Method with Default Settings

use aisdk::providers::XAI;

// Specify model as a string at runtime
let xai = XAI::model_name("grok-3");

Using Builder Pattern with Custom Settings

use aisdk::{
    core::DynamicModel,
    providers::XAI,
};

let xai = XAI::<DynamicModel>::builder()
    .model_name("grok-3")
    .api_key("your-api-key")
		.base_url("https://api.x.ai/")
    .build()?;

Next Steps

• Take a deeper look at text generation features Generating Text / Streaming Text
• Explore Structured Output for reliable agent data.
• Learn how to create Custom Tools.
• Learn more about Agents.