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 安裝為獨立套件，用它取代 Mastra 模型路由字串。詳情請參閱套件文件。

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 文件。