Weights & Biases

WANDB_API_KEY=your-api-key

import { Agent } from "@mastra/core";
 
const agent = new Agent({
  name: "my-agent",
  instructions: "You are a helpful assistant",
  model: "wandb/Qwen/Qwen3-235B-A22B-Instruct-2507"
});
 
// Generate a response
const response = await agent.generate("Hello!");
 
// Stream a response
const stream = await agent.stream("Tell me a story");
for await (const chunk of stream) {
  console.log(chunk);
}

const agent = new Agent({
  name: "custom-agent",
  model: {
    url: "https://api.inference.wandb.ai/v1",
    id: "wandb/Qwen/Qwen3-235B-A22B-Instruct-2507",
    apiKey: process.env.WANDB_API_KEY,
    headers: {
      "X-Custom-Header": "value"
    }
  }
});

const agent = new Agent({
  name: "dynamic-agent",
  model: ({ runtimeContext }) => {
    const useAdvanced = runtimeContext.task === "complex";
    return useAdvanced 
      ? "wandb/moonshotai/Kimi-K2-Instruct"
      : "wandb/Qwen/Qwen3-235B-A22B-Instruct-2507";
  }
});