AI SDK - WaveSpeedAI Provider

The WaveSpeedAI provider for the AI SDK enables integration with the WaveSpeedAI API for image generation.

Setup

The WaveSpeedAI provider is available in the wavespeedai-ai-provider package. You can install it with

npm install wavespeedai-ai-provider@latest

Usage

To use the WaveSpeedAI provider, you'll need to set up your API token:

WAVESPEEDAI_API_TOKEN="your-api-token-here"

Then import and use the provider for image generation:

import { wavespeedai } from "wavespeedai-ai-provider";
import { experimental_generateImage as generateImage } from "ai";

const { image } = await generateImage({
  model: wavespeedai.image("model-id"), // Replace with actual WaveSpeedAI model ID
  prompt: "A detailed description of the image you want to generate",
});

// Save the generated image
const filename = `image-${Date.now()}.png`;
fs.writeFileSync(filename, image.uint8Array);
console.log(`Image saved to ${filename}`);

If you need to pass additional inputs to the model besides the prompt, use the providerOptions.wavespeedai property:

const { image } = await generateImage({
  model: wavespeedai.image("google/nano-banana-pro/text-to-image"),
  prompt: "A detailed cat working as a café barista",
  providerOptions: {
    wavespeedai: {
      customInput: "this model needs this extra field",
    },
  },
});

Configuration

You can also create a custom provider instance with specific settings:

import { createWaveSpeedAI } from "wavespeedai-ai-provider";

const wavespeedaiProvider = createWaveSpeedAI({
  apiToken: process.env.WAVESPEEDAI_API_TOKEN, // Optional if using environment variables
  baseURL: "https://api.wavespeed.ai/api/v3", // Optional, uses default if not provided
  headers: {}, // Optional additional headers
  fetch: undefined, // Optional custom fetch implementation
});

// Use the custom provider instance
const { image } = await generateImage({
  model: wavespeedaiProvider.image("model-id"),
  prompt: "Your prompt here",
});