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 のドキュメントを参照してください。