Cerebras

通过 Mastra 的模型路由器访问 3 个 Cerebras 模型。系统会自动使用 CEREBRAS_API_KEY 环境变量完成身份验证。

更多信息请参阅Cerebras 文档。

.env

CEREBRAS_API_KEY=your-api-key

src/mastra/agents/my-agent.ts

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)
}

模型

3 available 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
cerebras/zai-glm-4.7	131K						$2	$3

高级配置

自定义请求头

src/mastra/agents/my-agent.ts

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',
    },
  },
})

动态选择模型

src/mastra/agents/my-agent.ts

const agent = new Agent({
  id: 'dynamic-agent',
  name: 'Dynamic Agent',
  model: ({ requestContext }) => {
    const useAdvanced = requestContext.task === 'complex'
    return useAdvanced ? 'cerebras/zai-glm-4.7' : 'cerebras/gemma-4-31b'
  },
})

直接安装 Provider

你也可以直接将此 Provider 安装为独立 package，用它代替 Mastra 模型路由字符串。详情请参阅package 文档。

npm

npm install @ai-sdk/cerebras

pnpm

pnpm add @ai-sdk/cerebras

Yarn

yarn add @ai-sdk/cerebras

Bun

bun add @ai-sdk/cerebras

有关此 Provider 的详细说明，请参阅 AI SDK Cerebras Provider 文档。