Providers

CloudflareWorkersAi

AISDK provides first-class support for CloudflareWorkersAi 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 CloudflareWorkersAi provider feature:

cargo add aisdk --features cloudflare-workers-ai

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

Create a Provider Instance

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

use aisdk::providers::CloudflareWorkersAi;

let cloudflare-workers-ai = CloudflareWorkersAi::ai4bharat_indictrans2_en_indic_1b();

This initializes the provider with:

Basic Text Generation

Example using LanguageModelRequest for text generation.

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

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

    let cloudflare-workers-ai = CloudflareWorkersAi::ai4bharat_indictrans2_en_indic_1b();

    let response = LanguageModelRequest::builder()
        .model(cloudflare-workers-ai)
        .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 CloudflareWorkersAi::builder()

API Key

let cloudflare-workers-ai = CloudflareWorkersAi::<Ai4bharatIndictrans2EnIndic1b>::builder()
    .api_key("your-api-key")
    .build()?;

If not specified, AISDK uses the CLOUDFLARE_ACCOUNT_ID environment variable.

Base URL

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

let cloudflare-workers-ai = CloudflareWorkersAi::<Ai4bharatIndictrans2EnIndic1b>::builder()
    .base_url("https://api.cloudflare.com/client/v4/accounts/${CLOUDFLARE_ACCOUNT_ID}/ai/v1")
    .build()?;

Path (Full URL Override)

Use .path(...) to override the full request URL instead of only the base URL.

let cloudflare-workers-ai = CloudflareWorkersAi::<Ai4bharatIndictrans2EnIndic1b>::builder()
    .path("https://full-url.example/v1/chat/completions")
    .build()?;

Provider Name

For logging, analytics, and observability.

let cloudflare-workers-ai = CloudflareWorkersAi::<Ai4bharatIndictrans2EnIndic1b>::builder()
    .provider_name("CloudflareWorkersAi")
    .build()?;

Full Custom Configuration Example

let cloudflare-workers-ai = CloudflareWorkersAi::<Ai4bharatIndictrans2EnIndic1b>::builder()
    .api_key("your-api-key")
    .base_url("https://api.cloudflare.com/client/v4/accounts/${CLOUDFLARE_ACCOUNT_ID}/ai/v1")
    .path("https://full-url.example/v1/chat/completions")
    .provider_name("CloudflareWorkersAi")
    .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::CloudflareWorkersAi;

// Specify model as a string at runtime
let cloudflare-workers-ai = CloudflareWorkersAi::model_name("@cf/ai4bharat/indictrans2-en-indic-1B");

Using Builder Pattern with Custom Settings

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

let cloudflare-workers-ai = CloudflareWorkersAi::<DynamicModel>::builder()
	.model_name("@cf/ai4bharat/indictrans2-en-indic-1B")
	.api_key("your-api-key")
	.base_url("https://api.cloudflare.com/client/v4/accounts/${CLOUDFLARE_ACCOUNT_ID}/ai/v1")
	.path("https://full-url.example/v1/chat/completions")
	.provider_name("CloudflareWorkersAi")
	.build()?;

Warning: When using DynamicModel, model capabilities are not validated at compile time. This means there's no guarantee the model supports requested features (e.g., tool calls, structured output). For compile-time safety, use the typed methods like CloudflareWorkersAi::ai4bharat_indictrans2_en_indic_1b().

Next Steps