> Discover all available pages from the documentation index: https://mastra.ai/llms.txt

# ![Cerebras logo](https://models.dev/logos/cerebras.svg)Cerebras

Access 2 Cerebras models through Mastra's model router. Authentication is handled automatically using the `CEREBRAS_API_KEY` environment variable.

Learn more in the [Cerebras documentation](https://inference-docs.cerebras.ai/models/overview).

```bash
CEREBRAS_API_KEY=your-api-key
```

```typescript
import { Agent } from "@mastra/core/agent";

const agent = new Agent({
  id: "my-agent",
  name: "My Agent",
  instructions: "You are a helpful assistant",
  model: "cerebras/gemma-4-31b"
});

// 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);
}
```

## Models

| Model                   | Context | Tools | Reasoning | Image | Audio | Video | Input $/1M | Output $/1M |
| ----------------------- | ------- | ----- | --------- | ----- | ----- | ----- | ---------- | ----------- |
| `cerebras/gemma-4-31b`  | 131K    |       |           |       |       |       | $0.99      | $1          |
| `cerebras/gpt-oss-120b` | 131K    |       |           |       |       |       | $0.35      | $0.75       |

## Advanced configuration

### Custom headers

```typescript
const agent = new Agent({
  id: "custom-agent",
  name: "custom-agent",
  model: {
    id: "cerebras/gemma-4-31b",
    apiKey: process.env.CEREBRAS_API_KEY,
    headers: {
      "X-Custom-Header": "value"
    }
  }
});
```

### Dynamic model selection

```typescript
const agent = new Agent({
  id: "dynamic-agent",
  name: "Dynamic Agent",
  model: ({ requestContext }) => {
    const useAdvanced = requestContext.task === "complex";
    return useAdvanced
      ? "cerebras/gpt-oss-120b"
      : "cerebras/gemma-4-31b";
  }
});
```

## Direct provider installation

This provider can also be installed directly as a standalone package, which can be used instead of the Mastra model router string. View the [package documentation](https://www.npmjs.com/package/@ai-sdk/cerebras) for more details.

**npm**:

```bash
npm install @ai-sdk/cerebras
```

**pnpm**:

```bash
pnpm add @ai-sdk/cerebras
```

**Yarn**:

```bash
yarn add @ai-sdk/cerebras
```

**Bun**:

```bash
bun add @ai-sdk/cerebras
```

For detailed provider-specific documentation, see the [AI SDK Cerebras provider docs](https://ai-sdk.dev/providers/ai-sdk-providers/cerebras).